Lines Matching full:d2

56         # CHECK: (d0, d1, d2) -> (d2, d0, d1)
109         # CHECK: (d0, d1, d2, d3, d4) -> (d1, d2, d3)
113         # CHECK: (d0, d1, d2, d3, d4) -> (d0, d1)
117         # CHECK: (d0, d1, d2, d3, d4) -> (d3, d4)
128         d2 = AffineDimExpr.get(2)
129         map1 = AffineMap.get(3, 0, [d2, d0])
130         map2 = AffineMap.get(3, 0, [d2, d0, d1])
131         map3 = AffineMap.get(3, 1, [d2, d0, d1])
152         d2 = AffineDimExpr.get(2)
153         map3 = AffineMap.get(3, 1, [d2, d0, d1])
166             # CHECK: d2
173             # CHECK: d2
175         assert list(map3.results) == [d2, d0, d1]
182         d0, d1, d2 = (AffineDimExpr.get(0), AffineDimExpr.get(1), AffineDimExpr.get(2))
189             AffineMap.get(3, 3, [d2, d0, d1]),
190             AffineMap.get(3, 3, [d2, d0 + s2, d1]),
191             AffineMap.get(3, 3, [d1, d2, d0]),
196         #      CHECK: AffineMap((d0, d1, d2)[s0, s1, s2] -> (d2, d0, d1))
197         # CHECK-SAME: AffineMap((d0, d1, d2)[s0, s1, s2] -> (d2, d0 + s2, d1))
198         # CHECK-SAME: AffineMap((d0, d1, d2)[s0, s1, s2] -> (d1, d2, d0))
201         #      CHECK: AffineMap((d0, d1, d2)[s0] -> (d2, d0, d1))
202         # CHECK-SAME: AffineMap((d0, d1, d2)[s0] -> (d2, d0 + s0, d1))
203         # CHECK-SAME: AffineMap((d0, d1, d2)[s0] -> (d1, d2, d0))
211         d0, d1, d2 = (AffineDimExpr.get(0), AffineDimExpr.get(1), AffineDimExpr.get(2))
217         map1 = AffineMap.get(3, 3, [d2, d0 + s1 + s2, d1 + s0])
223         # CHECK: (d0, d1, d2)[s0, s1, s2] -> (d2, d0 + s1 + s2, d1 + 42)
226         # CHECK: (d0, d1, d2)[s0, s1, s2] -> (d2, d0 + s2 + 42, d1 + s0)
229         # CHECK: (d0, d1, d2)[s0, s1] -> (d2, d0 + s1 + 42, d1 + s0)