| /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()
|