Addressed PR comments

apache · Feb 4, 2020 · f100fe6 · f100fe6
1 parent 3f1e5d9
commit f100fe6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/benchmark/opperf/nd_operations/nn_basic_operators.py b/benchmark/opperf/nd_operations/nn_basic_operators.py
@@ -55,6 +55,8 @@ def run_nn_basic_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler='n
         Context to run benchmarks
     dtype: str, default 'float32'
         Precision to use for benchmarks
+    profiler: str, default 'native'
+        Module to use for tracking benchmark excecution time
     warmup: int, default 25
         Number of times to run for warmup
     runs: int, default 100