diff --git a/imperative/src/impl/proxy_graph/mini_graph.h b/imperative/src/impl/proxy_graph/mini_graph.h
index a3866b4f..52aa5567 100644
--- a/imperative/src/impl/proxy_graph/mini_graph.h
+++ b/imperative/src/impl/proxy_graph/mini_graph.h
@@ -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;
             }