mv_inference: Caculate tensor size correctly

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

diff --git a/mv_inference/inference/src/Inference.cpp b/mv_inference/inference/src/Inference.cpp
index 75a17039c909740a0244752e2868209815d5900c..359bca34ac64204959dd9fb6fc87e99bb93a6d78 100755 (executable)
--- a/mv_inference/inference/src/Inference.cpp
+++ b/mv_inference/inference/src/Inference.cpp
@@ -663,8 +663,8 @@ namespace inference
                                        tensor_buffer.buffer = new float[tensor_info.size];
                                        tensor_buffer.size = tensor_info.size * 4;
                                } else if (tensor_info.data_type ==
-                                       INFERENCE_TENSOR_DATA_TYPE_UINT64) {
-                                       tensor_buffer.buffer = new unsigned long long[tensor_info.size];
+                                       INFERENCE_TENSOR_DATA_TYPE_INT64) {
+                                       tensor_buffer.buffer = new long long[tensor_info.size];
                                        tensor_buffer.size = tensor_info.size * 8;
                                } else if (tensor_info.data_type ==
                                        INFERENCE_TENSOR_DATA_TYPE_UINT32) {
@@ -677,9 +677,9 @@ namespace inference
                                } else if (tensor_info.data_type ==
                                                   INFERENCE_TENSOR_DATA_TYPE_UINT16) {
                                        tensor_buffer.buffer = new short[tensor_info.size];
-                                       tensor_buffer.size = tensor_info.size;
+                                       tensor_buffer.size = tensor_info.size * 2;
                                } else {
-                                       LOGE("Invalid output tensor data type.");
+                                       LOGE("Invalid output tensor data type.(%d)", tensor_buffer.data_type);
                                        CleanupTensorBuffers();
                                        return MEDIA_VISION_ERROR_INVALID_PARAMETER;
                                }