Lines Matching refs:map0
102 #map0 = affine_map<(d0) -> (d0 * 2 + 1)>
104 func @example(%arg0: memref<?xf32>, %arg1: memref<?xvector<4xf32>, #map0>) {
110 %2 = load %arg1[%arg2] : memref<?xvector<4xf32>, #map0>
112 store %3, %arg1[%arg2] : memref<?xvector<4xf32>, #map0>
182 #map0 = affine_map<(d0, d1) -> (d0 * 2 + d1 * 2)>
184 func @example(%arg0: memref<8x?xf32, #map0>, %arg1: memref<?xvector<4xf32>>) {
188 %0 = dim %arg0, %c1 : memref<8x?xf32, #map0>
191 %1 = load %arg0[%arg3, %arg2] : memref<8x?xf32, #map0>
374 #map0 = affine_map<(d0, d1)[s0, s1, s2] -> (d0 * s1 + s0 + d1 * s2)>
377 %0 = memref_cast %arg0 : memref<?x?xf32> to memref<?x?xf32, #map0>
378 %1 = memref_cast %arg1 : memref<?x?xf32> to memref<?x?xf32, #map0>
379 %2 = memref_cast %arg2 : memref<?x?xf32> to memref<?x?xf32, #map0>
380 …ll @pointwise_add(%0, %1, %2) : (memref<?x?xf32, #map0>, memref<?x?xf32, #map0>, memref<?x?xf32, #…
383 func @pointwise_add(memref<?x?xf32, #map0>, memref<?x?xf32, #map0>, memref<?x?xf32, #map0>) attribu…