connorgoggins commented on a change in pull request #16253: [OpPerf] Add 
Indexing ops
URL: https://github.com/apache/incubator-mxnet/pull/16253#discussion_r374295736
 
 

 ##########
 File path: benchmark/opperf/utils/benchmark_utils.py
 ##########
 @@ -24,10 +24,9 @@
 from .common_utils import merge_map_list
 from .op_registry_utils import prepare_op_inputs
 from benchmark.opperf.rules.default_params import PARAMS_OF_TYPE_NDARRAY
-from .profiler_utils import cpp_profile,python_profile
+from .profiler_utils import cpp_profile, python_profile
 
-
-no_backward = ['softmax_cross_entropy']
+no_backward = ['gather_nd', 'softmax_cross_entropy']
 
 Review comment:
   Nice, this will be very useful.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
[email protected]


With regards,
Apache Git Services

Reply via email to