Clean up

apache · Aug 8, 2019 · e122e21 · e122e21
1 parent d0a4cd4
commit e122e21
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
@@ -474,7 +474,7 @@ def _impl(inputs, attr, params):
         # reshape result back to n-dimensional
         if len(orig_shape_x) > 3:
             final_shape = attr['_output_shapes'][0]
-            ret = _op.reshape(ret, final_shape)
+            ret = _op.reshape(ret, newshape=final_shape)
 
         return ret
     return _impl