Lines Matching refs:ModularIndexing

23     ModularIndexing,
43             + ModularIndexing(i1, 1, 64)
44             + 64 * ModularIndexing(i1 + 64 * r3, 64, 2)
50             i1 + 128 * i2 + 64 * ModularIndexing(r3, 1, 2),
58         expr = ModularIndexing(i1 - 15, 1, 64)
61             ModularIndexing(i1 - 15, 1, 64),
69         expr = ModularIndexing(2 * i2 + r3, 1, 64)
75         self.assertEqual(ModularIndexing(r3 * i0, r3, 10), ModularIndexing(i0, 1, 10))
79             ModularIndexing(i0 + i1 * 10, 1, 10), ModularIndexing(i0, 1, 10)
84             ModularIndexing(i0 + i1 * 20, 2, 10), ModularIndexing(i0, 2, 10)
89             ModularIndexing(i0 + i1 * i2 * r3, i2, r3), ModularIndexing(i0, i2, r3)
94             ModularIndexing(-i0 + i1 * 20, 2, 10), ModularIndexing(-i0 + i1 * 20, 2, 10)
97             ModularIndexing(-15 + i1 * 20, 2, 10), ModularIndexing(-15 + i1 * 20, 2, 10)
101         self.assertEqual(ModularIndexing(i0 * 4, 2, 10), ModularIndexing(i0 * 2, 1, 10))
106         expr = ModularIndexing(ModularIndexing(121 * i1 + i2, 1, 784), 1, 28)
108         expr = ModularIndexing(ModularIndexing(121 * i1 + i2, 1, 784) + 1, 1, 28)
111         expr = ModularIndexing(ModularIndexing(i2, 1, 28), 7, 4)
112         expected = FloorDiv(ModularIndexing(i2, 1, 28), 7)
114         expr = ModularIndexing(ModularIndexing(i2, 1, 28) + 1, 7, 4)
124         expr1 = ModularIndexing(i0, 1, 32) + 32 * ModularIndexing(i0, 32, 4)
126             sizevars.simplify_with_ranges(expr1, {}), ModularIndexing(i0, 1, 128)
132             2 * ModularIndexing(i0, 1, 128),
136         expr2 = ModularIndexing(i0, 3, 32) + 32 * ModularIndexing(i0, 32 * 3, 4)
138         self.assertEqual(simplified, ModularIndexing(i0, 3, 128))
142         expr3 = ModularIndexing(i0, 1, 30) + 32 * ModularIndexing(i0, 32, 4)
146         expr4 = ModularIndexing(i0, 10, i1) + i1 * ModularIndexing(i0, i1 * 10, i2)
151         self.assertEqual(simplified, ModularIndexing(i0, 10, i1 * i2))
156             ModularIndexing(i0, 10, i1 * i2) + 10,
160         expr5 = 197 * FloorDiv(i0, 197) + ModularIndexing(i0, 1, 197)
172         expr6 = 197 * FloorDiv(i0, 197 * 3) + ModularIndexing(i0, 3, 197)
182         expr1 = ModularIndexing(x, 1, a)
183         expr2 = ModularIndexing(expr1, 1, b)
184         expected = ModularIndexing(x, 1, b)
195         expr1 = ModularIndexing(x, 1, a)
196         expr2 = ModularIndexing(expr1, 1, b)
200         self.assertNotEqual(ModularIndexing(x, 1, b), actual)