[Bug] fix layout checking when calculating tensor size

This patch fixes layout checking when calculating tensor size.

Signed-off-by: Dongju Chae <dongju.chae@samsung.com>

diff --git a/src/core/ne-model.cc b/src/core/ne-model.cc
index 8815a9b6605c9f436f8b531a4a0a95ce2c7e98b4..63a534e9666ae480524769d93c84a0f83462467d 100644 (file)
--- a/src/core/ne-model.cc
+++ b/src/core/ne-model.cc
@@ -263,7 +263,7 @@ Metadata_v3::getInputTensorSize (uint32_t idx, data_layout layout) const {
     image = (depth == 1 || depth == 3);
 
     /** special handling for TRIV2 */
-    if (layout == DATA_LAYOUT_TRIV2 && depth % granularity != 0 && !image) {
+    if (depth % granularity != 0 && !image) {
       uint32_t new_depth;
 
       new_depth = (depth + granularity - 1) >> shift;
@@ -308,7 +308,7 @@ Metadata_v3::getOutputTensorSize (uint32_t idx, data_layout layout) const {
     granularity /= div;
 
     /** special handling for TRIV2 */
-    if (layout == DATA_LAYOUT_TRIV2 && depth % granularity != 0) {
+    if (depth % granularity != 0) {
       uint32_t new_depth;
 
       new_depth = (depth + granularity - 1) >> shift;