Add --disable_per_channel_quantize option

pfnet-research · Aug 27, 2019 · eeb6612 · eeb6612
1 parent 9fa16b6
commit eeb6612
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/compiler/passes.cc b/compiler/passes.cc
@@ -118,6 +118,7 @@ void RunDefaultPasses(Graph* graph, bool gen_backprop, bool skip_scheduling) {
 
         if (g_quantize) {
             QuantizationOptions q_opts;
+            q_opts.per_channel = !g_disable_per_channel_quantize;
             Recursively([q_opts](Graph* graph) { Quantize(q_opts, graph); }, graph);
         }
 

diff --git a/scripts/generate_flags_code.py b/scripts/generate_flags_code.py
@@ -118,6 +118,10 @@
         'type': 'bool',
         'doc': 'Quantize ONNX model'
     },
+    'disable_per_channel_quantize': {
+        'type': 'bool',
+        'doc': 'Disables per channel quantization'
+    },
 
     'computation_order': {
         'type': 'std::string',