return LowerInfo::Shape4D(1, 1, 1, 1);
case 1u:
- return LowerInfo::Shape4D(shape.dim(0), 1, 1, 1);
+ return LowerInfo::Shape4D(1, 1, 1, shape.dim(0));
case 2u:
- return LowerInfo::Shape4D(shape.dim(0), shape.dim(1), 1, 1);
+ return LowerInfo::Shape4D(1, 1, shape.dim(0), shape.dim(1));
case 3u:
- return LowerInfo::Shape4D(shape.dim(0), shape.dim(1), shape.dim(2), 1);
+ return LowerInfo::Shape4D(1, shape.dim(0), shape.dim(1), shape.dim(2));
case 4u:
return LowerInfo::Shape4D(shape.dim(0), shape.dim(1), shape.dim(2), shape.dim(3));