add a todo

apache · Jun 16, 2020 · 63f52eb · 63f52eb
1 parent 87ddf70
commit 63f52eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)