diff options
-rw-r--r-- | src/inference_engine_common_impl.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/inference_engine_common_impl.cpp b/src/inference_engine_common_impl.cpp index 367ff7c..d27be80 100644 --- a/src/inference_engine_common_impl.cpp +++ b/src/inference_engine_common_impl.cpp @@ -110,8 +110,8 @@ namespace Common return INFERENCE_ENGINE_ERROR_INVALID_PARAMETER; } - if (tensor_info.data_type < INFERENCE_TENSOR_DATA_TYPE_FLOAT16 || - tensor_info.data_type > INFERENCE_TENSOR_DATA_TYPE_UINT32) { + if (tensor_info.data_type <= INFERENCE_TENSOR_DATA_TYPE_NONE || + tensor_info.data_type >= INFERENCE_TENSOR_DATA_TYPE_MAX) { LOGE("tensor data type is invalid."); return INFERENCE_ENGINE_ERROR_INVALID_PARAMETER; } |