From 63f52eb2cf542dcef30102dd1120e78c632bc126 Mon Sep 17 00:00:00 2001 From: Haichen Shen Date: Tue, 16 Jun 2020 22:47:54 +0000 Subject: [PATCH] add a todo --- python/tvm/relay/frontend/nnvm_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/nnvm_common.py b/python/tvm/relay/frontend/nnvm_common.py index b613cc26ecc6..a2eea94b06f0 100644 --- a/python/tvm/relay/frontend/nnvm_common.py +++ b/python/tvm/relay/frontend/nnvm_common.py @@ -57,7 +57,7 @@ def _impl(inputs, attrs): def _softmax_op(new_op): """softmax/log_softmax""" def _impl(inputs, attrs, _dtype='float32'): - # currently ignore the 2nd input to softmax in mxnet 1.6 + # TODO(@icemelon9): currently ignore the 2nd input to softmax for mxnet 1.6 # assert len(inputs) == 1 axis = attrs.get_int("axis", -1) return new_op(inputs[0], axis=axis)