tflite:add empty checking

For calling 'GetInputTensorBuffers', 'SetInputLayerProperty' should be called.

Change-Id: I6e8695cf8be42c2cd04a43b31684323ec9ed8f85
Signed-off-by: Hyunsoo Park <hance.park@samsung.com>

diff --git a/packaging/inference-engine-tflite.spec b/packaging/inference-engine-tflite.spec
index 0d47911eddfa0217eefb82b83da26c109b6e2833..971f8c6a43333a5f86deb9ac3c28d8fb44d2fa6b 100644 (file)
--- a/packaging/inference-engine-tflite.spec
+++ b/packaging/inference-engine-tflite.spec
@@ -1,7 +1,7 @@
 Name:       inference-engine-tflite
 Summary:    Tensorflow-Lite based implementation of inference-engine-interface
 Version:    0.0.1
-Release:    8
+Release:    9
 Group:      Multimedia/Libraries
 License:    Apache-2.0
 Source0:    %{name}-%{version}.tar.gz

diff --git a/src/inference_engine_tflite.cpp b/src/inference_engine_tflite.cpp
index ae643af3e4285e27e4338055f9ccb137faea9742..83a5f7a28a17a5c9b28d0ce70bfec73ff8305784 100644 (file)
--- a/src/inference_engine_tflite.cpp
+++ b/src/inference_engine_tflite.cpp
@@ -146,6 +146,11 @@ int InferenceTFLite::GetInputTensorBuffers(std::vector<inference_engine_tensor_b
 {
     LOGI("ENTER");
 
+    if (mInputTensorInfo.empty()) {
+        LOGE("InputTensorInfo is empty. Do SetInputLayerProperty first.");
+        return INFERENCE_ENGINE_ERROR_INVALID_OPERATION;
+    }
+
     mInputData.clear();
 
     void *pBuff = NULL;