Skip to content
Permalink
Browse files

Add Mul, Mean and Pad to Quantizer.

PiperOrigin-RevId: 238738419
  • Loading branch information...
shashishekhar authored and tensorflower-gardener committed Mar 16, 2019
1 parent 216932e commit b3991e03abad123095e43b05c200e1c01bc7ad49
Showing with 3 additions and 0 deletions.
  1. +3 −0 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);

0 comments on commit b3991e0

Please sign in to comment.
You can’t perform that action at this time.