From 518a93cc48ebca7e01dda434cfae63d2abf413df Mon Sep 17 00:00:00 2001 From: Inki Dae Date: Wed, 6 Nov 2024 13:46:23 +0900 Subject: [PATCH] add quantization type [Version] : 0.0.7 [Issue type] : new feature Add quantization type to tensor info for quantization tensor. Change-Id: I65f06daa99af4740cf3ac62884077286c5aa6e38 Signed-off-by: Inki Dae --- packaging/inference-engine-tflite.spec | 2 +- src/inference_engine_tflite.cpp | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/packaging/inference-engine-tflite.spec b/packaging/inference-engine-tflite.spec index f80e5cb..17da402 100644 --- a/packaging/inference-engine-tflite.spec +++ b/packaging/inference-engine-tflite.spec @@ -1,6 +1,6 @@ Name: inference-engine-tflite Summary: Tensorflow-Lite based implementation of inference-engine-interface -Version: 0.0.6 +Version: 0.0.7 Release: 0 Group: Multimedia/Libraries License: Apache-2.0 diff --git a/src/inference_engine_tflite.cpp b/src/inference_engine_tflite.cpp index 21a0e05..a4e8389 100644 --- a/src/inference_engine_tflite.cpp +++ b/src/inference_engine_tflite.cpp @@ -495,8 +495,8 @@ namespace TFLiteImpl tensor_info.scale = quant_parms->scale->data[0]; tensor_info.zero_point = quant_parms->zero_point->data[0]; - LOGD("scale of %s : %f", tensor->name, tensor_info.scale); - LOGD("zero_point of %s : %d", tensor->name, tensor_info.zero_point); + if (tensor->quantization.type == kTfLiteAffineQuantization) + tensor_info.quantization_type = INFERENCE_TENSOR_QUANTIZATION_AFFINE; } } -- 2.34.1