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 <inki.dae@samsung.com>

diff --git a/packaging/inference-engine-tflite.spec b/packaging/inference-engine-tflite.spec
index f80e5cbd98fe9a0079229d5230fbba960d43dd52..17da4029ad934dbe99a70b3db7873f3f08da1d9c 100644 (file)
--- 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 21a0e05792985c52f7ebe94588bce568cac53e76..a4e83899190cd5fa1f4e498d25bd0e099aeed4ce 100644 (file)
--- 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;
                                }
 
                        }