| func.func public @extract_slice_i32_offset1_size2_stride1() { | |
| %1 = arith.constant dense<[1, 2, 3, 4]> : tensor<4xi32> | |
| %2 = util.optimization_barrier %1 : tensor<4xi32> | |
| %extracted_slice = tensor.extract_slice %2[1] [2] [1] : tensor<4xi32> to tensor<2xi32> | |
| check.expect_eq_const(%extracted_slice, dense<[2, 3]> : tensor<2xi32>) : tensor<2xi32> | |
| return | |
| } | |
| func.func public @extract_slice_i64_offset1_size2_stride1() { | |
| %1 = arith.constant dense<[1, 2, 3, 4]> : tensor<4xi64> | |
| %2 = util.optimization_barrier %1 : tensor<4xi64> | |
| %extracted_slice = tensor.extract_slice %2[1] [2] [1] : tensor<4xi64> to tensor<2xi64> | |
| check.expect_eq_const(%extracted_slice, dense<[2, 3]> : tensor<2xi64>) : tensor<2xi64> | |
| return | |
| } | |
| func.func public @extract_slice_i32_offset1_size2_stride2() { | |
| %1 = arith.constant dense<[1, 2, 3, 4]> : tensor<4xi32> | |
| %2 = util.optimization_barrier %1 : tensor<4xi32> | |
| %extracted_slice = tensor.extract_slice %2[1] [2] [2] : tensor<4xi32> to tensor<2xi32> | |
| check.expect_eq_const(%extracted_slice, dense<[2, 4]> : tensor<2xi32>) : tensor<2xi32> | |
| return | |
| } |