Addressed NN Basic PR comment

apache · Feb 4, 2020 · 07d8238 · 07d8238
1 parent 689cff0
commit 07d8238
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/benchmark/opperf/nd_operations/nn_activation_operators.py b/benchmark/opperf/nd_operations/nn_activation_operators.py
@@ -53,6 +53,8 @@ def run_activation_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler=
         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