Browse Source

fix(imperative/mini_graph): fix no_err value in shape infer

GitOrigin-RevId: 087d15db66
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
267d612723
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/src/impl/proxy_graph/mini_graph.h

+ 1
- 0
imperative/src/impl/proxy_graph/mini_graph.h View File

@@ -606,6 +606,7 @@ public:
desc.comp_node = minigraph.output_var(i)->comp_node();
if (auto* shape = sess.infer_shape(i, false)) {
desc.layout.init_contiguous_stride(*shape);
noerr = true;
} else {
noerr = false;
}


Loading…
Cancel
Save