| Thomas | f0f64ca | 2022-03-30 10:38:01 -0700 | [diff] [blame] | 1 | func.func @test_transpose() { |
| iree-copybara-bot | df503b4 | 2021-10-13 09:28:25 -0700 | [diff] [blame] | 2 | %0 = arith.constant dense<[[[0, 1, 2], [3, 4, 5]]]> : tensor<1x2x3xi32> |
| 3 | %1 = arith.constant dense<[1, 2, 0]> : tensor<3xi32> |
| Han-Chung Wang | 42e54ab | 2023-08-22 23:10:14 -0700 | [diff] [blame] | 4 | %2 = tosa.transpose %0, %1 : (tensor<1x2x3xi32>, tensor<3xi32>) -> (tensor<2x3x1xi32>) |
| rsuderman | 8baa8e4 | 2021-03-04 17:04:44 -0800 | [diff] [blame] | 5 | check.expect_eq_const(%2, dense<[[[0], [1], [2]], [[3], [4], [5]]]> : tensor<2x3x1xi32>) : tensor<2x3x1xi32> |
| 6 | return |
| 7 | } |
