Searched defs:for_j (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
H A D | test_simplify.cpp | 1228 auto for_j = For::make(j, 0, 10, Store::make(a_buf, {i, j}, (i + j * 6) / 6)); in TEST() local 1254 auto for_j = in TEST() local 1281 auto for_j = in TEST() local 1413 auto for_j = For::make(j, 0, 10, Store::make(a_buf, {i, j}, (i + j * 6) % 6)); in TEST() local 1439 auto for_j = in TEST() local 1466 auto for_j = in TEST() local
|
H A D | test_loopnest.cpp | 679 auto for_j = For::make(j, 0, 100, Block::make({store1, for_k1, for_k2})); in TEST() local
|