Fix UT

apache · Jun 11, 2019 · 14ea11a · 14ea11a
1 parent 23c2c82
commit 14ea11a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/imperative/cached_op.cc b/src/imperative/cached_op.cc
@@ -290,7 +290,7 @@ bool CachedOp::CheckDynamicShapeExists(const Context& default_ctx,
   CheckAndInferShape(&g, std::move(shape_inputs), true,
                      {0, 0}, {0, 0},
                      &contain_dynamic_shape);
-  if (contain_dynamic_shape && erase_result) {
+  if (!config_.static_shape && erase_result) {
     g.attrs.erase("shape");
     g.attrs.erase("shape_inputs");
   }