diff options
| -rw-r--r-- | src/inference_engine_mlapi.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/inference_engine_mlapi.cpp b/src/inference_engine_mlapi.cpp index 7570359..29913e5 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; |