diff --git a/nnvm/python/nnvm/frontend/tensorflow.py b/nnvm/python/nnvm/frontend/tensorflow.py index b01d489fb0423fef179fff299b634bfd0e580845..b0b546a32b3d5daf9a29a4cae02a496d4d6cada4 100644 --- a/nnvm/python/nnvm/frontend/tensorflow.py +++ b/nnvm/python/nnvm/frontend/tensorflow.py @@ -342,7 +342,7 @@ def _matmul(): def _impl(inputs, attr, params): channels = _infer_channels(inputs[1], params, not attr['transpose_b']) if attr['transpose_a']: - inputs[0] = _sym.transpose(inputs[0], axes(1, 0)) + inputs[0] = _sym.transpose(inputs[0], axes=(1, 0)) if not attr['transpose_b']: inputs[1] = _sym.transpose(inputs[1], axes=(1, 0)) return AttrCvt(op_name="dense",