Lines Matching refs:affine_map
28 [affine_map<(d0, d1, d2, d3, d4) -> (d0, d1)>,
29 affine_map<(d0, d1, d2, d3, d4) -> (d2)>,
30 affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>] :
33 [affine_map<(d0, d1, d2) -> (d0, d1)>,
34 affine_map<(d0, d1, d2) -> (d2)>] :
38 // CHECK-DAG: #[[$MAP0:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d0, d1, d2)>
39 // CHECK-DAG: #[[$MAP1:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>
50 %0 = linalg.tensor_reshape %arg0 [affine_map<(d0, d1, d2) -> (d0, d1, d2)>] :
63 %0 = linalg.reshape %arg0 [affine_map<(d0, d1, d2) -> (d0, d1, d2)>] :
77 [affine_map<(d0, d1, d2) -> (d0, d1)>,
78 affine_map<(d0, d1, d2) -> (d2)>] :
81 [affine_map<(d0, d1, d2, d3, d4) -> (d0, d1)>,
82 affine_map<(d0, d1, d2, d3, d4) -> (d2)>,
83 affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>] :
87 // CHECK-DAG: #[[$MAP0:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d0, d1, d2)>
88 // CHECK-DAG: #[[$MAP1:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>
98 [affine_map<(d0, d1, d2, d3, d4) -> (d0, d1)>,
99 affine_map<(d0, d1, d2, d3, d4) -> (d2)>,
100 affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>] :
103 [affine_map<(d0, d1, d2) -> (d0, d1)>,
104 affine_map<(d0, d1, d2) -> (d2)>] :
108 // CHECK-DAG: #[[$MAP0:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d0, d1, d2)>
109 // CHECK-DAG: #[[$MAP1:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>
119 [affine_map<(d0, d1, d2) -> (d0, d1)>,
120 affine_map<(d0, d1, d2) -> (d2)>] :
123 [affine_map<(d0, d1, d2, d3, d4) -> (d0, d1)>,
124 affine_map<(d0, d1, d2, d3, d4) -> (d2)>,
125 affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>] :
129 // CHECK-DAG: #[[$MAP0:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d0, d1, d2)>
130 // CHECK-DAG: #[[$MAP1:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>
140 %1 = linalg.tensor_reshape %0 [affine_map<(d0, d1, d2) -> (d0, d1, d2)>] :
154 [affine_map<(d0, d1, d2) -> (d0, d1, d2)>] :
167 [affine_map<(d0, d1, d2) -> (d0, d1)>,
168 affine_map<(d0, d1, d2) -> (d2)>] :
171 [affine_map<(d0, d1, d2) -> (d0, d1)>,
172 affine_map<(d0, d1, d2) -> (d2)>] :
184 [affine_map<(d0, d1, d2) -> (d0, d1)>,
185 affine_map<(d0, d1, d2) -> (d2)>] :
188 [affine_map<(d0, d1, d2) -> (d0, d1)>,
189 affine_map<(d0, d1, d2) -> (d2)>] :
202 [affine_map<(d0, d1, d2) -> (d0, d1)>,
203 affine_map<(d0, d1, d2) -> (d2)>] :
206 [affine_map<(d0, d1, d2) -> (d0, d1)>,
207 affine_map<(d0, d1, d2) -> (d2)>] :
219 [affine_map<(d0, d1, d2) -> (d0, d1)>,
220 affine_map<(d0, d1, d2) -> (d2)>] :
223 [affine_map<(d0, d1, d2) -> (d0, d1)>,
224 affine_map<(d0, d1, d2) -> (d2)>] :
235 affine_map<(i) -> (i)>,
236 affine_map<(i) -> (i)>
266 [affine_map<(d0, d1, d2) -> (d0)>,
267 affine_map<(d0, d1, d2) -> (d1, d2)>]
280 [affine_map<(d0, d1, d2) -> (d0)>,
281 affine_map<(d0, d1, d2) -> (d1, d2)>]
294 [affine_map<(d0, d1, d2) -> (d0)>,
295 affine_map<(d0, d1, d2) -> (d1, d2)>]
308 [affine_map<(d0, d1, d2) -> (d0)>,
309 affine_map<(d0, d1, d2) -> (d1, d2)>]