From 0edb9443dc1e725037c8c9a47ed906c61073ae70 Mon Sep 17 00:00:00 2001
From: Alexey Romanov <alexey.v.romanov@gmail.com>
Date: Wed, 21 Nov 2018 20:58:19 +0300
Subject: [PATCH] [FRONTEND][TENSORFLOW] Fix a typo in _matmul (#2152)
---
nnvm/python/nnvm/frontend/tensorflow.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/nnvm/python/nnvm/frontend/tensorflow.py b/nnvm/python/nnvm/frontend/tensorflow.py
index b01d489fb..b0b546a32 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",
--
GitLab