[neurun] Fix Shape4D in LowerInfo order (#3339)

Fix Shape4D order bug (NHWC)

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>

diff --git a/runtimes/neurun/src/graph/operand/Shape4DConvert.h b/runtimes/neurun/src/graph/operand/Shape4DConvert.h
index b840d19..2825740 100644 (file)
--- a/runtimes/neurun/src/graph/operand/Shape4DConvert.h
+++ b/runtimes/neurun/src/graph/operand/Shape4DConvert.h
@@ -34,16 +34,16 @@ inline LowerInfo::Shape4D asShape4D(const Shape &shape)
       return LowerInfo::Shape4D(1, 1, 1, 1);
 
     case 1u:
-      return LowerInfo::Shape4D(1, 1, 1, shape.dim(0));
+      return LowerInfo::Shape4D(shape.dim(0), 1, 1, 1);
 
     case 2u:
-      return LowerInfo::Shape4D(1, 1, shape.dim(1), shape.dim(0));
+      return LowerInfo::Shape4D(shape.dim(0), shape.dim(1), 1, 1);
 
     case 3u:
-      return LowerInfo::Shape4D(1, shape.dim(2), shape.dim(1), shape.dim(0));
+      return LowerInfo::Shape4D(shape.dim(0), shape.dim(1), shape.dim(2), 1);
 
     case 4u:
-      return LowerInfo::Shape4D(shape.dim(3), shape.dim(2), shape.dim(1), shape.dim(0));
+      return LowerInfo::Shape4D(shape.dim(0), shape.dim(1), shape.dim(2), shape.dim(3));
 
     default:
       throw "Unsupported rank > 4";