diff options
| author | Inki Dae <inki.dae@samsung.com> | 2020-08-28 11:51:43 +0900 |
|---|---|---|
| committer | Inki Dae <inki.dae@samsung.com> | 2020-08-28 11:51:43 +0900 |
| commit | 863a079d4304ffbcff47fd1c77ad88a649e94453 (patch) | |
| tree | 835eb6e08acb7e8bd2913b4a43d098feca36d936 | |
| parent | 2fc9aa5a36a9047f4161e0261d265a8d1c436420 (diff) | |
| download | inference-engine-tflite-863a079d4304ffbcff47fd1c77ad88a649e94453.tar.gz inference-engine-tflite-863a079d4304ffbcff47fd1c77ad88a649e94453.tar.bz2 inference-engine-tflite-863a079d4304ffbcff47fd1c77ad88a649e94453.zip | |
Add int64 tensor type supporttizen_6.0.m2_releasesubmit/tizen_6.0_hotfix/20201103.114803submit/tizen_6.0_hotfix/20201102.192503submit/tizen_6.0/20201029.205103submit/tizen/20200828.100528submit/tizen/20200828.025650accepted/tizen/unified/20200831.002554accepted/tizen/6.0/unified/hotfix/20201103.052008accepted/tizen/6.0/unified/20201030.121127tizen_6.0_hotfixtizen_6.0accepted/tizen_6.0_unified_hotfixaccepted/tizen_6.0_unified
Change-Id: I57251340b8321940c66e4e8bd3d04a0373726088
Signed-off-by: Inki Dae <inki.dae@samsung.com>
| -rw-r--r-- | src/inference_engine_tflite.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/inference_engine_tflite.cpp b/src/inference_engine_tflite.cpp index 8490b4d..d1a7687 100644 --- 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; |