Support two more tensor type

Support INT64 and UINT64

Change-Id: Iec1445e27bfeb245e1b38bd83679ea8f27e7bcf7
Signed-off-by: Hyo Jong Kim <hue.kim@samsung.com>

diff --git a/src/inference_engine_mlapi.cpp b/src/inference_engine_mlapi.cpp
index b4d7a04..bad1997 100644 (file)
--- 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;