Lines Matching refs:affine_map
11 func @slice_number_of_indexings(%arg0: memref<?x?xf32, affine_map<(i, j)[off, M]->(off + M * i + j)…
14 ….slice %arg0[%c0] : memref<?x?xf32, affine_map<(i, j)[off, M]->(off + M * i + j)>>, index, memref<…
19 func @slice_rank_vs_range_indices(%arg0: memref<?x?xf32, affine_map<(i, j)[off, M]->(off + M * i + …
22 …arg0[%c0, %c0] : memref<?x?xf32, affine_map<(i, j)[off, M]->(off + M * i + j)>>, index, index, mem…
36 func @yield_parent(%arg0: memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
38 linalg.yield %arg0: memref<?xf32, affine_map<(i)[off]->(off + i)>>
46 indexing_maps = [ affine_map<() -> (0)> ],
56 indexing_maps = [ affine_map<() -> ()> ],
69 indexing_maps = [ affine_map<() -> (0)> ],
79 func @generic_one_d_view(%arg0: memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
80 …op expected indexing_map #0 results to match view rank: 'memref<?xf32, affine_map<(d0)[s0] -> (d0 …
82 indexing_maps = [ affine_map<() -> (0, 0)> ],
84 outs(%arg0 : memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
92 func @generic_result_0_element_type(%arg0: memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
95 indexing_maps = [ affine_map<(i) -> (i)> ],
97 outs(%arg0 : memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
106 …@generic_singular_maps(%arg0: memref<?xf32, affine_map<(i)[off]->(off + i)>>, %arg1: memref<?xf32,…
110 affine_map<(i, j) -> (i + j)>,
111 affine_map<(i, j) -> (i + j)>
114 ins(%arg0 : memref<?xf32, affine_map<(i)[off]->(off + i)>>)
115 outs(%arg1 : memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
131 indexing_maps = [ affine_map<() -> (0)> ],
148 indexing_maps = [ affine_map<() -> (0)> ],
160 indexing_maps = [ affine_map<() -> (0)> ],
173 indexing_maps = [ affine_map<() -> (0)> ],
186 indexing_maps = [ affine_map<(d0) -> (d0)> ],
199 indexing_maps = [ affine_map<(d0) -> (d0)> ],
212 indexing_maps = [ affine_map<(d0) -> (d0)> ],
225 indexing_maps = [ affine_map<()[] -> ()> ],
240 indexing_maps = [ affine_map<(i) -> (i)> ]}
252 indexing_maps = [ affine_map<(d0) -> (d0)> ],
262 func @generic_result_0_element_type(%arg0: memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
265 indexing_maps = [ affine_map<(i) -> (i)> ],
267 outs(%arg0 : memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
276 func @generic_result_tensor_type(%arg0: memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
279 indexing_maps = [ affine_map<(i) -> (i)> ],
281 ins(%arg0 : memref<?xf32, affine_map<(i)[off]->(off + i)>>) {
293 indexing_maps = [ affine_map<(i) -> (i)> ],
323 %0 = linalg.reshape %arg0 [affine_map<()->(0)>] : memref<f32> into memref<f32>
330 %0 = linalg.reshape %arg0 [affine_map<(i)->(i)>] : memref<?xf32> into memref<?xf32>
337 %0 = linalg.reshape %arg0 [affine_map<(i, j, k) -> (i, j)>] :
345 %0 = linalg.reshape %arg0 [affine_map<(i) -> (i)>, affine_map<(i, j, k) -> (k)>] :
353 %0 = linalg.reshape %arg0 [affine_map<(i, j, k) -> (i, j)>, affine_map<(i, j, k) -> (k, j)>] :
360 …ected collapsed type to be 'memref<?x?xf32>', but got 'memref<?x?xf32, affine_map<(d0, d1)[s0] -> …
361 %0 = linalg.reshape %arg0 [affine_map<(i, j, k) -> (i, j)>, affine_map<(i, j, k) -> (k)>] :
362 memref<?x?x?xf32> into memref<?x?xf32, affine_map<(d0, d1)[s0] -> (d0 * s0 + d1)>>