Home
last modified time | relevance | path

Searched refs:set_parallel (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/jit/tensorexpr/
Dstmt.h659 void set_parallel() { in set_parallel() function
783 void set_parallel() { in set_parallel() function
784 loop_options_.set_parallel(); in set_parallel()
Dtensorexpr_init.cpp413 .def("set_parallel", &For::set_parallel) in initTensorExprBindings()
Dkernel.cpp784 flattened->set_parallel(); in parallelizeOuterLoops()
/external/pytorch/test/cpp/tensorexpr/
Dtest_llvm.cpp1532 m->set_parallel(); in testSimpleParallel()
1535 n->set_parallel(); in testSimpleParallel()
1608 loop_list[i]->set_parallel(); in TEST()