experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add Mul, Mean and Pad to Quantizer.

Browse Source 
PiperOrigin-RevId: 238738419
This commit is contained in:
Shashi Shekhar 2019-03-15 17:12:21 -07:00 committed by TensorFlower Gardener
parent 216932e897
commit b3991e03ab
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/tools/optimize/subgraph_quantizer.cc
View File

@ -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);