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

Add dynamic range quantization for TransposeConv2d filters

Browse Source 
PiperOrigin-RevId: 358327761
Change-Id: I1b4c22a0b4b6f01e70f359c2207f925ea77448d6
This commit is contained in:
David Rim 2021-02-18 20:56:04 -08:00 committed by TensorFlower Gardener
parent 2abdad6ab1
commit be58584469
1 changed files with 2 additions and 1 deletions

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

@ -91,7 +91,8 @@ std::vector<int32_t> GetWeightInputIndices(const OperatorCodeT* op_code,
builtin_op_code == BuiltinOperator_DEPTHWISE_CONV_2D ||
builtin_op_code == BuiltinOperator_FULLY_CONNECTED ||
builtin_op_code == BuiltinOperator_BATCH_MATMUL ||
builtin_op_code == BuiltinOperator_EMBEDDING_LOOKUP) {
builtin_op_code == BuiltinOperator_EMBEDDING_LOOKUP ||
builtin_op_code == BuiltinOperator_TRANSPOSE_CONV) {
return {1};
} else if (builtin_op_code == BuiltinOperator_SVDF) {
// https://www.tensorflow.org/code/tensorflow/lite/kernels/svdf.cc