Lines Matching full:24

4 // CHECK-SAME: %[[A:.*]]: tensor<24x12xf32>
6 // CHECK-SAME: %[[C:.*]]: tensor<24x25xf32>
7 func.func @vectorize_matmul(%arg0: tensor<24x12xf32>,
9                             %arg2: tensor<24x25xf32>) -> tensor<24x25xf32> {
15 …inalg.matmul ins(%arg0, %arg1 : tensor<24x12xf32>, tensor<12x25xf32>) outs(%arg2 : tensor<24x25xf3…
16   func.return %0 : tensor<24x25xf32>
31 // CHECK-SAME: %[[A:.*]]: memref<24x12xf32>
33 // CHECK-SAME: %[[C:.*]]: memref<24x25xf32>
34 func.func @vectorize_matmul_memref(%arg0: memref<24x12xf32>,
36                                    %arg2: memref<24x25xf32>) {
42 …linalg.matmul ins(%arg0, %arg1 : memref<24x12xf32>, memref<12x25xf32>) outs(%arg2 : memref<24x25xf…
83 // CHECK-SAME: %[[C:[a-zA-Z0-9_]+]]: tensor<24x25xf32>
85     %arg0: tensor<24x12xf32>, %arg1: tensor<12x25xf32>,
86     %arg2: tensor<24x25xf32>, %arg3: index, %arg4: index,
87     %arg5: index) -> tensor<24x25xf32> {
91 …%1 = tensor.extract_slice %arg0[%arg3, %arg5] [4, %0] [1, 1] : tensor<24x12xf32> to tensor<4x?xf32>
93   %3 = tensor.extract_slice %arg2[%arg3, %arg4] [4, 5] [1, 1] : tensor<24x25xf32> to tensor<4x5xf32>
112 …sor.insert_slice %8 into %arg2[%arg3, %arg4] [4, 5] [1, 1] : tensor<4x5xf32> into tensor<24x25xf32>
113   return %9 : tensor<24x25xf32>
131 // CHECK-SAME: %[[A:.+]]: tensor<24x12xf32>
133 // CHECK-SAME: %[[C:.+]]: tensor<24x25xf32>
135     %arg0: tensor<24x12xf32>, %arg1: tensor<12x25xf32>,
136     %arg2: tensor<24x25xf32>, %arg3: index, %arg4: index,
137     %arg5: index) -> tensor<24x25xf32> {
143 …%1 = tensor.extract_slice %arg0[%arg3, %arg5] [4, %0] [1, 1] : tensor<24x12xf32> to tensor<4x?xf32>
145   %3 = tensor.extract_slice %arg2[%arg3, %arg4] [4, 5] [1, 1] : tensor<24x25xf32> to tensor<4x5xf32>
162 …sor.insert_slice %8 into %arg2[%arg3, %arg4] [4, 5] [1, 1] : tensor<4x5xf32> into tensor<24x25xf32>
163   return %9 : tensor<24x25xf32>
177 func.func @vectorize(%arg0: tensor<24x12xf32>,
179                      %arg2: tensor<24x25xf32>) -> tensor<24x25xf32> {
181 …inalg.matmul ins(%arg0, %arg1 : tensor<24x12xf32>, tensor<12x25xf32>) outs(%arg2 : tensor<24x25xf3…
182   func.return %0 : tensor<24x25xf32>