Add int64 tensor type support

Change-Id: I57251340b8321940c66e4e8bd3d04a0373726088
Signed-off-by: Inki Dae <inki.dae@samsung.com>

diff --git a/src/inference_engine_tflite.cpp b/src/inference_engine_tflite.cpp
index 8490b4dcaca65a4bb4053d6a1051f1381366f211..d1a7687b523390092d94cd239b462ca7950ef245 100644 (file)
--- a/src/inference_engine_tflite.cpp
+++ b/src/inference_engine_tflite.cpp
@@ -225,6 +225,10 @@ namespace TFLiteImpl
                                pBuff = (void *) mInterpreter->typed_tensor<uint8_t>(
                                                mOutputLayerId[idx]);
                                buffer = { pBuff, INFERENCE_TENSOR_DATA_TYPE_UINT8, size, 1 };
+                       } else if (mInterpreter->tensor(mOutputLayerId[idx])->type == kTfLiteInt64) {
+                               LOGI("type is kTfLiteInt64");
+                               pBuff = (void*)mInterpreter->typed_tensor<int64_t>(mOutputLayerId[idx]);
+                               buffer = {pBuff, INFERENCE_TENSOR_DATA_TYPE_INT64, size * 8, 1};
                        } else if (mInterpreter->tensor(mOutputLayerId[idx])->type ==
                                           kTfLiteFloat32) {
                                LOGI("type is kTfLiteFloat32");
@@ -295,6 +299,9 @@ namespace TFLiteImpl
                        if (mInterpreter->tensor((*iter))->type == kTfLiteUInt8) {
                                LOGI("type is kTfLiteUInt8");
                                tensor_info.data_type = INFERENCE_TENSOR_DATA_TYPE_UINT8;
+                       } else if (mInterpreter->tensor((*iter))->type == kTfLiteInt64) {
+                               LOGI("type is kTfLiteInt64");
+                               tensor_info.data_type = INFERENCE_TENSOR_DATA_TYPE_INT64;
                        } else if (mInterpreter->tensor((*iter))->type == kTfLiteFloat32) {
                                LOGI("type is kTfLiteFloat32");
                                tensor_info.data_type = INFERENCE_TENSOR_DATA_TYPE_FLOAT32;