From 3c3bb6720948621de71b7af36718f35195e0b926 Mon Sep 17 00:00:00 2001 From: Haichen Shen Date: Fri, 29 Mar 2019 22:35:33 -0700 Subject: [PATCH] Fix mxnet converter for take --- python/tvm/relay/frontend/mxnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/mxnet.py b/python/tvm/relay/frontend/mxnet.py index eafa5c3cab826..3eabb016c10e2 100644 --- a/python/tvm/relay/frontend/mxnet.py +++ b/python/tvm/relay/frontend/mxnet.py @@ -450,7 +450,7 @@ def _mx_take(inputs, attrs): if mode == "raise": raise RuntimeError("take doesn't support raise mode") axis = attrs.get_int("axis", 0) - return _op.take(inputs[0], inputs[1].astype("int32"), axis, mode.upper()) + return _op.take(inputs[0], inputs[1].astype("int32"), axis, mode) def _mx_reverse(inputs, attrs):