diff options
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/tflitefile_tool/select_operator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py index c5d311d59..892341029 100644 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -141,7 +141,7 @@ def GenerateQuantization(new_builder, selected_quantization): tflite.QuantizationParameters.QuantizationParametersStartScaleVector( new_builder, zeropoint_num) for zeropoint_idx in reversed(range(zeropoint_num)): - new_builder.PrependFloat32(selected_quantization.ZeroPoint(zeropoint_idx)) + new_builder.PrependInt64(selected_quantization.ZeroPoint(zeropoint_idx)) new_zeropoint = new_builder.EndVector(zeropoint_num) # Create quantization |