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

Update Batch_MatMul op version during conversion.

Browse Source 
PiperOrigin-RevId: 347062719
Change-Id: Ic98dafe6601a65f48b61961032413060381aebe9
This commit is contained in:
Yunlu Li 2020-12-11 13:23:23 -08:00 committed by TensorFlower Gardener
parent 627a032dc2
commit 5f77cc8ebd
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/tools/optimize/quantize_weights.cc
View File

@ -356,6 +356,8 @@ void UpdateInt8OperatorVersions(ModelT* model, bool use_updated_hybrid_scheme) {
model->operator_codes[i]->version = use_updated_hybrid_scheme ? 5 : 2;
} else if (op_code == BuiltinOperator_FULLY_CONNECTED) {
model->operator_codes[i]->version = use_updated_hybrid_scheme ? 9 : 3;
} else if (op_code == BuiltinOperator_BATCH_MATMUL) {
model->operator_codes[i]->version = use_updated_hybrid_scheme ? 4 : 1;
} else if (op_code == BuiltinOperator_SVDF) {
model->operator_codes[i]->version = use_updated_hybrid_scheme ? 4 : 2;
} else if (op_code == BuiltinOperator_DEPTHWISE_CONV_2D) {