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

All the inputs of the BATCH_MATMUL should be quantized by the type of the activation

Browse Source
This commit is contained in:
Thibaut Goetghebuer-Planchon 2020-09-23 17:05:53 +01:00
parent 0f082f6b40
commit dff42b15e2
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/tools/optimize/operator_property.cc
View File

@ -97,6 +97,7 @@ OperatorProperty GetOperatorProperty(const ModelT* model, int subgraph_index,
property.inputs = {{0, {}}, {1, {}}}; property.inputs = {{0, {}}, {1, {}}};
property.outputs = {{0, {}}}; property.outputs = {{0, {}}};
property.version = 2; property.version = 2;
property.quantize_input_as_activations = true;
break; break;
} }
case BuiltinOperator_BATCH_TO_SPACE_ND: case BuiltinOperator_BATCH_TO_SPACE_ND: