Home
last modified time | relevance | path

Searched refs:splitWithMask (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.h201 static void splitWithMask(const ForPtr& f, int factor, ForPtr* inner);
204 static void splitWithMask(const ForPtr& f, int factor);
H A Dkernel.cpp868 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 Dloopnest_randomization.cpp277 l.splitWithMask(loop, factor); in loopnestRandomization()
H A Dloopnest.cpp1613 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 Dtensorexpr_init.cpp544 LoopNest::splitWithMask(f, factor, &inner); in initTensorExprBindings()
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_reductions.cpp935 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 Dtest_loopnest.cpp589 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 Dtest_llvm.cpp1649 loop.splitWithMask(m, 16); in TEST()
1654 loop.splitWithMask(n, 16); in TEST()
H A Dtest_cuda.cpp179 l.splitWithMask(loops[0], block_size, &n_inner); in testCudaTestVectorAdd02_impl()
398 l.splitWithMask(loops[0], 1024, &inner); in TEST()
H A Dtest_memdependency.cpp3077 loop.splitWithMask(m, 4); in TEST()
3082 loop.splitWithMask(n, 16); in TEST()