Skip to content

T
tic
Commit 0edb9443 by Alexey Romanov Committed by Tianqi Chen
Options

[FRONTEND][TENSORFLOW] Fix a typo in _matmul (#2152)

parent c1157ecf
Showing with 1 additions and 1 deletions
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",
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment