| func.func @reverse_dim0() { |
| %input = util.unfoldable_constant dense<[[1.0, 2.0, 3.0], |
| [4.0, 5.0, 6.0]]> : tensor<2x3xf32> |
| |
| %init = tensor.empty() : tensor<2x3xf32> |
| %0 = iree_linalg_ext.reverse |
| dimensions(dense<0> : tensor<1xi64>) |
| ins(%input : tensor<2x3xf32>) |
| outs(%init : tensor<2x3xf32>) : tensor<2x3xf32> |
| |
| check.expect_almost_eq_const( |
| %0, |
| dense<[[4.0, 5.0, 6.0], [1.0, 2.0, 3.0]]> : tensor<2x3xf32> |
| ) : tensor<2x3xf32> |
| |
| return |
| } |
| |
| func.func @reverse_dim1() { |
| %input = util.unfoldable_constant dense<[[1, 2, 3], |
| [4, 5, 6]]> : tensor<2x3xi32> |
| |
| %init = tensor.empty() : tensor<2x3xi32> |
| %0 = iree_linalg_ext.reverse |
| dimensions(dense<1> : tensor<1xi64>) |
| ins(%input : tensor<2x3xi32>) |
| outs(%init : tensor<2x3xi32>) : tensor<2x3xi32> |
| |
| check.expect_eq_const( |
| %0, |
| dense<[[3, 2, 1], [6, 5, 4]]> : tensor<2x3xi32> |
| ) : tensor<2x3xi32> |
| |
| return |
| } |
| |
| func.func @reverse_multi_dims() { |
| %input = util.unfoldable_constant dense<[[1, 2, 3], |
| [4, 5, 6]]> : tensor<2x3xi32> |
| |
| %init = tensor.empty() : tensor<2x3xi32> |
| %0 = iree_linalg_ext.reverse |
| dimensions(dense<[0, 1]> : tensor<2xi64>) |
| ins(%input : tensor<2x3xi32>) |
| outs(%init : tensor<2x3xi32>) : tensor<2x3xi32> |
| |
| check.expect_eq_const( |
| %0, |
| dense<[[6, 5, 4], [3, 2, 1]]> : tensor<2x3xi32> |
| ) : tensor<2x3xi32> |
| |
| return |
| } |
