/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
H A D | loopnest.h | 201 static void splitWithMask(const ForPtr& f, int factor, ForPtr* inner); 204 static void splitWithMask(const ForPtr& f, int factor);
|
H A D | kernel.cpp | 868 LoopNest::splitWithMask(flattened, blockSize, &inner); in transformLoops() 879 LoopNest::splitWithMask(flattened, blockCount * blockSize, &inner); in transformLoops() 880 LoopNest::splitWithMask(inner, blockSize, &inner1); in transformLoops() 910 LoopNest::splitWithMask(flattened, blockSize, &inner); in transformLoops()
|
H A D | loopnest_randomization.cpp | 277 l.splitWithMask(loop, factor); in loopnestRandomization()
|
H A D | loopnest.cpp | 1613 void LoopNest::splitWithMask(const ForPtr& f, int factor) { in splitWithMask() function in torch::jit::tensorexpr::LoopNest 1615 splitWithMask(f, factor, &inner); in splitWithMask() 1618 void LoopNest::splitWithMask(const ForPtr& f, int factor, ForPtr* inner) { in splitWithMask() function in torch::jit::tensorexpr::LoopNest
|
H A D | tensorexpr_init.cpp | 544 LoopNest::splitWithMask(f, factor, &inner); in initTensorExprBindings()
|
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
H A D | test_reductions.cpp | 935 LoopNest::splitWithMask(loops[i], 8); in TEST() 965 LoopNest::splitWithMask(loops[i], 5); in TEST() 996 LoopNest::splitWithMask(loops[i], 16); in TEST() 1536 LoopNest::splitWithMask(l.getLoopStmtsFor(e)[0], 4); in TEST() 1581 LoopNest::splitWithMask(l.getLoopStmtsFor(d)[0], 4, &inner); in TEST() 1584 LoopNest::splitWithMask(l.getLoopStmtsFor(e)[0], 4, &inner); in TEST() 1635 LoopNest::splitWithMask(l.getLoopStmtsFor(e)[0], 4, &inner); in TEST()
|
H A D | test_loopnest.cpp | 589 LoopNest::splitWithMask(loops[1], 4); in TEST() 622 LoopNest::splitWithMask(loops[0], 4); in TEST() 623 LoopNest::splitWithMask(loops[0], 4); in TEST() 889 LoopNest::splitWithMask(loops[0], 4, &inner); in TEST() 1405 LoopNest::splitWithMask(loops[0], 4); in TEST() 1417 LoopNest::splitWithMask(loops[0], 3); in TEST() 1441 LoopNest::splitWithMask(loops[0], 4, &i_inner); in TEST() 1442 LoopNest::splitWithMask(i_inner, 2); in TEST() 1456 LoopNest::splitWithMask(loops.back(), 3); in TEST() 1476 LoopNest::splitWithMask(loops.back(), 2); in TEST() [all …]
|
H A D | test_llvm.cpp | 1649 loop.splitWithMask(m, 16); in TEST() 1654 loop.splitWithMask(n, 16); in TEST()
|
H A D | test_cuda.cpp | 179 l.splitWithMask(loops[0], block_size, &n_inner); in testCudaTestVectorAdd02_impl() 398 l.splitWithMask(loops[0], 1024, &inner); in TEST()
|
H A D | test_memdependency.cpp | 3077 loop.splitWithMask(m, 4); in TEST() 3082 loop.splitWithMask(n, 16); in TEST()
|