From: Inki Dae Date: Fri, 3 Apr 2020 02:03:00 +0000 (+0900) Subject: Add barrier enumeration to inference_tensor_data_type_e X-Git-Tag: submit/tizen/20200423.063253~20 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=b2877b95e521c5dfed5a9fe8bc29efe2fe0f986d;p=platform%2Fcore%2Fmultimedia%2Finference-engine-interface.git Add barrier enumeration to inference_tensor_data_type_e Change-Id: I7b5d659f749bf20273e485ca0a5378137a752cdc Signed-off-by: Inki Dae --- diff --git a/include/inference_engine_type.h b/include/inference_engine_type.h index b79a549..aaad014 100644 --- a/include/inference_engine_type.h +++ b/include/inference_engine_type.h @@ -91,11 +91,13 @@ typedef enum { * */ typedef enum { - TENSOR_DATA_TYPE_FLOAT16 = 0, + TENSOR_DATA_TYPE_NONE = 0, + TENSOR_DATA_TYPE_FLOAT16, TENSOR_DATA_TYPE_FLOAT32, TENSOR_DATA_TYPE_UINT8, TENSOR_DATA_TYPE_UINT16, - TENSOR_DATA_TYPE_UINT32 + TENSOR_DATA_TYPE_UINT32, + TENSOR_DATA_TYPE_MAX } inference_tensor_data_type_e; #define INFERENCE_TARGET_MASK (INFERENCE_TARGET_CPU | INFERENCE_TARGET_GPU | INFERENCE_TARGET_CUSTOM) diff --git a/src/inference_engine_common_impl.cpp b/src/inference_engine_common_impl.cpp index d29ba67..6b2923f 100755 --- a/src/inference_engine_common_impl.cpp +++ b/src/inference_engine_common_impl.cpp @@ -113,7 +113,7 @@ int InferenceEngineCommon::CheckTensorBuffers(std::vector TENSOR_DATA_TYPE_UINT32) { + if (tensor_buffer.data_type <= TENSOR_DATA_TYPE_NONE || tensor_buffer.data_type >= TENSOR_DATA_TYPE_MAX) { LOGE("tensor data type is invalid."); return INFERENCE_ENGINE_ERROR_INVALID_PARAMETER; }