Rebase with master

apache · Apr 13, 2019 · 1b6132d · 1b6132d
1 parent 6924f96
commit 1b6132d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/rnn.cc b/src/operator/rnn.cc
@@ -50,7 +50,7 @@ static bool RNNShape(const nnvm::NodeAttrs& attrs,
       "Needed input:[data, parameters, state], got in_shape->size(): " << in_shape->size();
   }
   const TShape &dshape = (*in_shape)[rnn_enum::kData];
-  if (dshape.ndim() ==  0) return false;
+  if (!mxnet::ndim_is_known(dshape)) return false;
   CHECK_EQ(dshape.ndim(), 3U) \
       << "Input data should be rank-3 tensor of dim [sequence length, batch size, input size]";
   // data: [sequence len, batch, input dimension]