Add barrier enumeration to inference_tensor_data_type_e

Change-Id: I7b5d659f749bf20273e485ca0a5378137a752cdc
Signed-off-by: Inki Dae <inki.dae@samsung.com>

diff --git a/include/inference_engine_type.h b/include/inference_engine_type.h
index b79a549..aaad014 100644 (file)
--- 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 (executable)
--- a/src/inference_engine_common_impl.cpp
+++ b/src/inference_engine_common_impl.cpp
@@ -113,7 +113,7 @@ int InferenceEngineCommon::CheckTensorBuffers(std::vector<inference_engine_tenso
                        return INFERENCE_ENGINE_ERROR_INVALID_PARAMETER;
                }
 
-               if (tensor_buffer.data_type < TENSOR_DATA_TYPE_FLOAT16 || tensor_buffer.data_type > 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;
                }