Fix build warnings

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

diff --git a/src/inference_engine_armnn.cpp b/src/inference_engine_armnn.cpp
index 5fee20d3c16354645c88be3b7dcc14109c35a1bf..6707507b27fe1df8d4ae0498966d1562795f9616 100644 (file)
--- a/src/inference_engine_armnn.cpp
+++ b/src/inference_engine_armnn.cpp
@@ -470,8 +470,8 @@ int InferenceARMNN::Run(std::vector<inference_engine_tensor_buffer> &input_buffe
         input_tensors.push_back({inBindingInfo.first, input_tensor});
 
         armnn::TensorShape shape = inputTensorInfo.GetShape();
-        int tensor_size = 1;
-        for (int i = 0; i < inputTensorInfo.GetNumDimensions(); i++)
+        unsigned int tensor_size = 1;
+        for (unsigned int i = 0; i < inputTensorInfo.GetNumDimensions(); i++)
             tensor_size *= shape[i];
 
         LOGI("Input Tensor dimension = %d (size = %d)", inputTensorInfo.GetNumDimensions(), tensor_size);
@@ -490,8 +490,8 @@ int InferenceARMNN::Run(std::vector<inference_engine_tensor_buffer> &input_buffe
         output_tensors.push_back({outBindingInfo.first, output_tensor});
 
         armnn::TensorShape shape = outputTensorInfo.GetShape();
-        int tensor_size = 1;
-        for (int i = 0; i < outputTensorInfo.GetNumDimensions(); i++)
+        unsigned int tensor_size = 1;
+        for (unsigned int i = 0; i < outputTensorInfo.GetNumDimensions(); i++)
             tensor_size *= shape[i];
 
         LOGI("Output Tensor dimension = %d (size = %d)", outputTensorInfo.GetNumDimensions(), tensor_size);