Lines Matching refs:affine_map
18 // CHECK-DAG: #[[$strided1D:.*]] = affine_map<(d0)[s0] -> (d0 + s0)>
79 // CHECK-DAG: #[[$strided1D:.*]] = affine_map<(d0)[s0] -> (d0 + s0)>
80 // CHECK-DAG: #[[$strided2D:.*]] = affine_map<(d0, d1)[s0, s1] -> (d0 * s1 + s0 + d1)>
113 // CHECK-DAG: #[[$strided1D:.*]] = affine_map<(d0)[s0] -> (d0 + s0)>
125 // CHECK-DAG: #[[$strided3D:.*]] = affine_map<(d0, d1, d2)[s0, s1, s2] -> (d0 * s1 + s0 + d1 * s2 +…
126 // CHECK-DAG: #[[$strided3DT:.*]] = affine_map<(d0, d1, d2)[s0, s1, s2] -> (d2 * s1 + s0 + d1 * s2 …
129 … memref<?x?x?xf32, offset: ?, strides: [?, ?, 1]> to memref<?x?x?xf32, affine_map<(d0, d1, d2)[s0,…
138 // CHECK-DAG: #[[$strided3D:.*]] = affine_map<(d0, d1, d2)[s0, s1, s2] -> (d0 * s1 + s0 + d1 * s2 +…
150 // CHECK-DAG: #[[$strided1D:.*]] = affine_map<(d0)[s0] -> (d0 + s0)>
164 // CHECK-DAG: #[[$strided3D:.*]] = affine_map<(d0, d1, d2)[s0, s1, s2] -> (d0 * s1 + s0 + d1 * s2 +…
165 // CHECK-DAG: #[[$map0:.*]] = affine_map<(d0, d1, d2) -> (d0, d2, d1)>
166 // CHECK-DAG: #[[$map1:.*]] = affine_map<(d0, d1, d2) -> (d2, d1, d0)>
170 linalg.copy(%arg0, %arg1) {inputPermutation = affine_map<(i, j, k) -> (i, k, j)>,
171 outputPermutation = affine_map<(i, j, k) -> (k, j, i)>} :
185 // CHECK-DAG: #[[$strided3D:.*]] = affine_map<(d0, d1, d2)[s0, s1, s2] -> (d0 * s1 + s0 + d1 * s2 +…
203 // CHECK-DAG: #[[$strided6D:.*]] = affine_map<(d0, d1, d2, d3, d4, d5)[s0, s1, s2, s3, s4, s5] -> (…
287 // CHECK-DAG: #[[$strided2D:.*]] = affine_map<(d0, d1)[s0, s1] -> (d0 * s1 + s0 + d1)>
288 // CHECK-DAG: #[[$strided3D:.*]] = affine_map<(d0, d1, d2)[s0, s1, s2] -> (d0 * s1 + s0 + d1 * s2 +…
291 affine_map<(i, j, k) -> (j, i)>,
292 affine_map<(i, j, k) -> (i, k, i + j)>
344 #map0 = affine_map<(d0, d1, d2) -> (d0, d1, d2)>
363 affine_map<(i, j, k) -> (j, i)>,
364 affine_map<(i, j, k) -> (i, k, i + j)>,
365 affine_map<(i, j, k) -> (i, k, i + j)>
398 affine_map<(i, j, k) -> (j, i)>,
399 affine_map<(i, j, k) -> (i, k, i + j)>,
400 affine_map<(i, j, k) -> (i, k, i + j)>
433 affine_map<(i, j) -> ()>,
434 affine_map<(i, j) -> (i, j)>
465 // CHECK-DAG: #[[$strided2D:.*]] = affine_map<(d0, d1)[s0, s1] -> (d0 * s1 + s0 + d1)>
466 // CHECK-DAG: #[[$strided3D:.*]] = affine_map<(d0, d1, d2)[s0, s1, s2] -> (d0 * s1 + s0 + d1 * s2 +…
469 affine_map<(i, j, k) -> (j, i)>,
470 affine_map<(i, j, k) -> (i, k, i + j)>
526 // CHECK-DAG: #[[$reshapeD01:.*]] = affine_map<(d0, d1, d2) -> (d0, d1)>
527 // CHECK-DAG: #[[$reshapeD2:.*]] = affine_map<(d0, d1, d2) -> (d2)>
528 // CHECK-DAG: #[[$reshapeD0:.*]] = affine_map<(d0, d1, d2) -> (d0)>
529 // CHECK-DAG: #[[$reshapeD12:.*]] = affine_map<(d0, d1, d2) -> (d1, d2)>
530 // CHECK-DAG: #[[$reshapeD012:.*]] = affine_map<(d0, d1, d2) -> (d0, d1, d2)>
531 // CHECK-DAG: #[[$reshape5D01:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d0, d1)>
532 // CHECK-DAG: #[[$reshape5D2:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d2)>
533 // CHECK-DAG: #[[$reshape5D34:.*]] = affine_map<(d0, d1, d2, d3, d4) -> (d3, d4)>
537 %0 = linalg.reshape %arg0 [affine_map<(i, j, k) -> (i, j)>,
538 affine_map<(i, j, k) -> (k)>] :
540 %r0 = linalg.reshape %0 [affine_map<(i, j, k) -> (i, j)>,
541 affine_map<(i, j, k) -> (k)>] :
543 %1 = linalg.reshape %arg0 [affine_map<(i, j, k) -> (i)>,
544 affine_map<(i, j, k) -> (j, k)>] :
546 %r1 = linalg.reshape %1 [affine_map<(i, j, k) -> (i)>,
547 affine_map<(i, j, k) -> (j, k)>] :
549 %2 = linalg.reshape %arg0 [affine_map<(i, j, k) -> (i, j, k)>] :
551 %r2 = linalg.reshape %2 [affine_map<(i, j, k) -> (i, j, k)>] :
554 %3 = linalg.reshape %arg0 [affine_map<(i, j, k, l, m) -> (i, j)>,
555 affine_map<(i, j, k, l, m) -> (k)>,
556 affine_map<(i, j, k, l, m) -> (l, m)>] :
558 %r3 = linalg.reshape %3 [affine_map<(i, j, k, l, m) -> (i, j)>,
559 affine_map<(i, j, k, l, m) -> (k)>,
560 affine_map<(i, j, k, l, m) -> (l, m)>] :
563 %t0 = linalg.tensor_reshape %arg1 [affine_map<(i, j, k, l, m) -> (i, j)>,
564 affine_map<(i, j, k, l, m) -> (k)>,
565 affine_map<(i, j, k, l, m) -> (l, m)>] :
567 %rt0 = linalg.tensor_reshape %t0 [affine_map<(i, j, k, l, m) -> (i, j)>,
568 affine_map<(i, j, k, l, m) -> (k)>,
569 affine_map<(i, j, k, l, m) -> (l, m)>] :
571 %t1 = linalg.tensor_reshape %arg2 [affine_map<(i, j, k, l, m) -> (i, j)>,
572 affine_map<(i, j, k, l, m) -> (k)>,
573 affine_map<(i, j, k, l, m) -> (l, m)>] :
575 %rt1 = linalg.tensor_reshape %t1 [affine_map<(i, j, k, l, m) -> (i)>,
576 affine_map<(i, j, k, l, m) -> (j, k)>,
577 affine_map<(i, j, k, l, m) -> (l, m)>] :
606 // CHECK-DAG: #[[$strided2D:.*]] = affine_map<(d0, d1)[s0, s1] -> (d0 * s1 + s0 + d1)>
607 // CHECK-DAG: #[[$strided2DOFF0:.*]] = affine_map<(d0, d1)[s0] -> (d0 * s0 + d1)>
608 // CHECK-DAG: #[[$strided3D:.*]] = affine_map<(d0, d1, d2)[s0, s1, s2] -> (d0 * s1 + s0 + d1 * s2 +…
609 // CHECK-DAG: #[[$strided3DOFF0:.*]] = affine_map<(d0, d1, d2)[s0, s1] -> (d0 * s0 + d1 * s1 + d2)>
614 %0 = linalg.reshape %arg0 [affine_map<(i, j, k) -> (i, j)>,
615 affine_map<(i, j, k) -> (k)>] :
617 %r0 = linalg.reshape %0 [affine_map<(i, j, k) -> (i, j)>,
618 affine_map<(i, j, k) -> (k)>] :
620 %1 = linalg.reshape %arg1 [affine_map<(i, j, k) -> (i, j)>,
621 affine_map<(i, j, k) -> (k)>] :
624 %r1 = linalg.reshape %1 [affine_map<(i, j, k) -> (i, j)>,
625 affine_map<(i, j, k) -> (k)>] :
628 %2 = linalg.reshape %arg2 [affine_map<(i, j, k) -> (i, j)>,
629 affine_map<(i, j, k) -> (k)>] :
632 %r2 = linalg.reshape %2 [affine_map<(i, j, k) -> (i, j)>,
633 affine_map<(i, j, k) -> (k)>] :
639 // CHECK-DAG: #[[$reshapeD01:.*]] = affine_map<(d0, d1, d2) -> (d0, d1)>
640 // CHECK-DAG: #[[$reshapeD2:.*]] = affine_map<(d0, d1, d2) -> (d2)>