| func.func @reshape_1D_2D() { |
| %input = util.unfoldable_constant dense<[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]> : tensor<12xi32> |
| %result = "stablehlo.reshape"(%input) : (tensor<12xi32>) -> tensor<3x4xi32> |
| check.expect_eq_const(%result, dense<[ |
| [1, 2, 3, 4], |
| [5, 6, 7, 8], |
| [9, 10, 11, 12]]> : tensor<3x4xi32>) : tensor<3x4xi32> |
| return |
| } |
| |
| // func.func @reshape_1D_3D() { |
| // %input = util.unfoldable_constant dense<[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]> : tensor<12xi32> |
| // %result = "stablehlo.reshape"(%input) : (tensor<12xi32>) -> tensor<2x2x3xi32> |
| // check.expect_eq_const(%result, dense<[ |
| // [[1, 2, 3], [4, 5, 6]], |
| // [[7, 8, 9], [10, 11, 12]]]> : tensor<2x2x3xi32>) : tensor<2x2x3xi32> |
| // return |
| // } |
| |
| // func.func @reshape_2D_3D() { |
| // %input = util.unfoldable_constant dense<[[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12]]> : tensor<2x6xi32> |
| // %result = "stablehlo.reshape"(%input) : (tensor<2x6xi32>) -> tensor<2x1x6xi32> |
| // check.expect_eq_const(%result, dense<[[[1, 2, 3, 4, 5, 6]], [[7, 8, 9, 10, 11, 12]]]> : tensor<2x1x6xi32>) : tensor<2x1x6xi32> |
| // return |
| // } |
| |
| // func.func @reshape_3D_1D() { |
| // %input = util.unfoldable_constant dense<[[[1, 2, 3, 4, 5, 6]], [[7, 8, 9, 10, 11, 12]]]> : tensor<2x1x6xi32> |
| // %result = "stablehlo.reshape"(%input) : (tensor<2x1x6xi32>) -> tensor<2x6xi32> |
| // check.expect_eq_const(%result, dense<[[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12]]> : tensor<2x6xi32>) : tensor<2x6xi32> |
| // return |
| // } |
