Fix infer shape bug (#148)

eric-haibin-lin · Sep 21, 2017 · e75c53e · e75c53e
1 parent d00545a
commit e75c53e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pass/infer_shape_type.cc b/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";
           }