diff options
author | 장지섭/On-Device Lab(SR)/Engineer/삼성전자 <jiseob.jang@samsung.com> | 2019-08-08 09:49:33 +0900 |
---|---|---|
committer | 오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com> | 2019-08-08 09:49:33 +0900 |
commit | 53e01a476b02369aa6d8d76ac876379545968e6b (patch) | |
tree | ac9618a8a7928549c8b0f64440f48086fb2e013e /tools | |
parent | 9962d0b6551c4204abb778e1e050c425699c7761 (diff) | |
download | nnfw-53e01a476b02369aa6d8d76ac876379545968e6b.tar.gz nnfw-53e01a476b02369aa6d8d76ac876379545968e6b.tar.bz2 nnfw-53e01a476b02369aa6d8d76ac876379545968e6b.zip |
Fix converting type of zero point in select_operator.py (#6353)
This commit fixes converting type of zero point in select_operator.py.
Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>
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 |