Minor fix conv3d

apache · Mar 13, 2020 · 785964b · 785964b
1 parent 08945d9
commit 785964b
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
@@ -413,7 +413,7 @@ def _impl(inputs, attr, params, mod):
         inputs_data = inputs[0] if opname != 'conv_transpose' else inputs[2]
 
         # NCDHW Layout require weights transpose
-        weights_shape = _infer_shape([inputs[1]], mod)
+        weights_shape = _infer_shape(inputs[1], mod)
         if attr['data_format'] == 'NCDHW':
             tmp_shape = weights_shape
             tmp_shape = [tmp_shape[ii] for ii in (4, 3, 0, 1, 2)]