Fix comment

apache · Jun 11, 2020 · d5848b6 · d5848b6
1 parent f20f0a4
commit d5848b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
@@ -1353,7 +1353,7 @@ def _impl(inputs, attr, params, mod):
         # try to compute its shape.
         if output_shape is None or -1 in output_shape:
             try:
-                output_shape = _infer_value(inputs[0], params).asnumpy().reshape([-1]).tolist()
+                output_shape = _infer_value(inputs[0], params, mod).asnumpy().reshape([-1]).tolist()
             except (IndexError, KeyError, AttributeError):
                 output_shape = inputs[0]