Fix mxnet converter for take

apache · Mar 30, 2019 · 3c3bb67 · 3c3bb67
1 parent daf6211
commit 3c3bb67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):