Lines Matching refs:idx2
259 // CHECK-SAME: %[[t1:.*]]: tensor<?x?xf32>, %[[idx1:.*]]: index, %[[idx2:.*]]: index
261 %t1: tensor<?x?xf32>, %idx1: index, %idx2: index) -> tensor<?x10xf32> {
263 …// CHECK: %[[r:.*]] = memref.subview %[[m]][5, %[[idx2]]] [%[[idx1]], 10] [1, 1] : memref<?x?xf32>…
264 %0 = tensor.extract_slice %t1[5, %idx2][%idx1, 10][1, 1]
277 // CHECK-SAME: %[[idx2:.*]]: index
279 %t1: tensor<?x10x?xf32>, %idx1: index, %idx2: index) -> tensor<?x15xf32> {
281 …// CHECK: %[[r:.*]] = memref.subview %[[m1]][5, %[[idx1]], 10] [%[[idx2]], 1, 15] [1, 1, 1] : memr…
282 %0 = tensor.extract_slice %t1[5, %idx1, 10][%idx2, 1, 15][1, 1, 1]
293 // CHECK-SAME: %[[idx1:.*]]: index, %[[idx2:.*]]: index
295 %idx1: index, %idx2: index) -> tensor<?x?xf32> {
304 // CHECK: %[[subview:.*]] = memref.subview %[[alloc]][%[[idx1]], 5] [%[[idx2]], 10] [1, 1]
306 %0 = tensor.insert_slice %t2 into %t1[%idx1, 5][%idx2, 10][1, 1]
320 %t1: tensor<?x?xf32>, %f: tensor<f32>, %idx1: index, %idx2: index)
326 %0 = tensor.insert_slice %f into %t1[%idx1, %idx2][1, 1][1, 1]