From b3991e03abad123095e43b05c200e1c01bc7ad49 Mon Sep 17 00:00:00 2001 From: Shashi Shekhar Date: Fri, 15 Mar 2019 17:12:21 -0700 Subject: [PATCH] Add Mul, Mean and Pad to Quantizer. PiperOrigin-RevId: 238738419 --- tensorflow/lite/tools/optimize/subgraph_quantizer.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/lite/tools/optimize/subgraph_quantizer.cc b/tensorflow/lite/tools/optimize/subgraph_quantizer.cc index 7eadbd5983e..a444b10ae01 100644 --- a/tensorflow/lite/tools/optimize/subgraph_quantizer.cc +++ b/tensorflow/lite/tools/optimize/subgraph_quantizer.cc @@ -369,6 +369,9 @@ TfLiteStatus SubgraphQuantizer::QuantizeOperator(int op_idx) { case BuiltinOperator_SQUEEZE: case BuiltinOperator_RESHAPE: case BuiltinOperator_ADD: + case BuiltinOperator_MUL: + case BuiltinOperator_PAD: + case BuiltinOperator_MEAN: return QuantizeInputsAndOutputs(op_code, op); case BuiltinOperator_SOFTMAX: return AsymmetricQuantizeSoftmax(op_code, op);