From a2af85153ba4af38a342f3d2ba3ae2f3edc4f5d8 Mon Sep 17 00:00:00 2001 From: Hyo Jong Kim Date: Tue, 25 Aug 2020 11:29:42 +0900 Subject: [PATCH] Support two more tensor type Support INT64 and UINT64 Change-Id: Iec1445e27bfeb245e1b38bd83679ea8f27e7bcf7 Signed-off-by: Hyo Jong Kim --- src/inference_engine_mlapi.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/inference_engine_mlapi.cpp b/src/inference_engine_mlapi.cpp index b4d7a04..bad1997 100644 --- a/src/inference_engine_mlapi.cpp +++ b/src/inference_engine_mlapi.cpp @@ -487,6 +487,10 @@ namespace MLAPIImpl return INFERENCE_TENSOR_DATA_TYPE_UINT8; case ML_TENSOR_TYPE_UINT16: return INFERENCE_TENSOR_DATA_TYPE_UINT16; + case ML_TENSOR_TYPE_INT64: + return INFERENCE_TENSOR_DATA_TYPE_INT64; + case ML_TENSOR_TYPE_UINT64: + return INFERENCE_TENSOR_DATA_TYPE_UINT64; default: LOGE("Tensor type(%d) is invalid.", tensor_type); return INFERENCE_ENGINE_ERROR_INVALID_PARAMETER; -- 2.7.4