Fix infer shape bug (#148)

392eb60d · reminisce · Tianqi Chen · 35f36fa4 · 392eb60d
Commit 392eb60d authored Sep 20, 2017 by reminisce Committed by Tianqi Chen May 29, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

nnvm/src/pass/infer_shape_type.cc
+1 -1

No files found.
--- a/nnvm/src/pass/infer_shape_type.cc
+++ b/nnvm/src/pass/infer_shape_type.cc
@@ -108,7 +108,7 @@ Graph InferAttr(Graph &&ret,
          uint32_t eid = idx.entry_id(nid, igrad[i].index);
          if (fis_none(rshape[eid])) {
            rshape[eid] = rshape[idx.entry_id(fnode.inputs[i])];
-          } else {
+          } else if (!fis_none(rshape[idx.entry_id(fnode.inputs[i])])) {
            CHECK_EQ(rshape[eid], rshape[idx.entry_id(fnode.inputs[i])])
                << "Backward shape inconsistent with the forward shape";
          }