Searched refs:getLoopStmtsFor (Results 1 – 12 of 12) sorted by relevance
595 std::vector<ForPtr> loops = l.getLoopStmtsFor(tensor); in TEST()623 std::vector<ForPtr> loops = l.getLoopStmtsFor(tensor); in TEST()661 auto loops = l.getLoopStmtsFor(tensor); in TEST()705 std::vector<ForPtr> loops = loop.getLoopStmtsFor(c); in TEST()738 std::vector<ForPtr> loops = loop.getLoopStmtsFor(c); in TEST()771 std::vector<ForPtr> loops = loop.getLoopStmtsFor(c); in TEST()810 reduce, loop.getLoopStmtsFor(tmp_buf).at(idx), &tmp_buf)); in TEST()841 std::vector<ForPtr> loops = loop.getLoopStmtsFor(c); in TEST()871 std::vector<ForPtr> loops = loop.getLoopStmtsFor(c); in TEST()903 std::vector<ForPtr> loops = loop.getLoopStmtsFor(c); in TEST()[all …]
48 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in testCudaTestVectorAdd01_impl()116 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in TEST()178 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in testCudaTestVectorAdd02_impl()345 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in TEST()397 std::vector<ForPtr> loops = l.getLoopStmtsFor(b); in TEST()1182 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in TEST()1184 loops = l.getLoopStmtsFor(d); in TEST()1279 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in TEST()1281 loops = l.getLoopStmtsFor(d); in TEST()1378 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in TEST()[all …]
133 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in TEST()209 std::vector<ForPtr> loops = l.getLoopStmtsFor(b); in TEST()263 std::vector<ForPtr> loops = l.getLoopStmtsFor(c); in TEST()505 StmtPtr j_loop = l.getLoopStmtsFor(B)[1]; in TEST()644 ForPtr loopRootA = l.getLoopStmtsFor(A)[0]; in TEST()645 ForPtr loopRootB = l.getLoopStmtsFor(B)[0]; in TEST()669 ForPtr loopRootA = l.getLoopStmtsFor(A)[0]; in TEST()670 ForPtr loopRootB = l.getLoopStmtsFor(B)[0]; in TEST()688 ForPtr outer = l.getLoopStmtsFor(A)[0]; in TEST()
1437 std::vector<ForPtr> loops = loop.getLoopStmtsFor(b); in TEST()1442 loops = loop.getLoopStmtsFor(b); in TEST()1480 std::vector<ForPtr> loops = loopnest.getLoopStmtsFor(b); in TEST()1528 auto const& loops = loop_nest.getLoopStmtsFor(f); in testSimpleParallel()1588 auto const& loops = loop_nest.getLoopStmtsFor(t1); in TEST()1592 auto const& loops = loop_nest.getLoopStmtsFor(t2); in TEST()1596 auto const& loops = loop_nest.getLoopStmtsFor(t3); in TEST()1601 auto const& loops = loop_nest.getLoopStmtsFor(t4); in TEST()1647 auto const& loops = loop.getLoopStmtsFor(CT); in TEST()1652 auto const& loops = loop.getLoopStmtsFor(CT); in TEST()[all …]
15 auto loops = ln->getLoopStmtsFor(target); in vectorize()
2860 auto c_loop = l.getLoopStmtsFor(c)[0]; in TEST()2861 auto d_loop = l.getLoopStmtsFor(d)[0]; in TEST()2928 l.splitWithTail(l.getLoopStmtsFor(c)[0], 2); in TEST()2974 auto loops = l.getLoopStmtsFor(c); in TEST()3046 auto c_loop = l.getLoopStmtsFor(c)[0]; in TEST()3047 auto d_loop = l.getLoopStmtsFor(d)[0]; in TEST()3075 auto const& loops = loop.getLoopStmtsFor(CT); in TEST()3080 auto const& loops = loop.getLoopStmtsFor(CT); in TEST()3087 auto const& loops = loop.getLoopStmtsFor(CT); in TEST()3095 auto const& loops = loop.getLoopStmtsFor(CT); in TEST()[all …]
46 std::vector<ForPtr> getLoopStmtsFor(const Tensor&) const;47 std::vector<ForPtr> getLoopStmtsFor(const BufPtr&) const;48 std::vector<ForPtr> getLoopStmtsFor(StmtPtr) const;
1323 auto fors = getLoopStmtsFor(store); in optimizeConditionals()2597 std::vector<ForPtr> LoopNest::getLoopStmtsFor(const Tensor& t) const { in getLoopStmtsFor() function in torch::jit::tensorexpr::LoopNest2599 return getLoopStmtsFor(cur_stmt); in getLoopStmtsFor()2602 std::vector<ForPtr> LoopNest::getLoopStmtsFor(const BufPtr& buf) const { in getLoopStmtsFor() function in torch::jit::tensorexpr::LoopNest2604 return getLoopStmtsFor(cur_stmt); in getLoopStmtsFor()2607 std::vector<ForPtr> LoopNest::getLoopStmtsFor(StmtPtr s) const { in getLoopStmtsFor() function in torch::jit::tensorexpr::LoopNest
759 auto loops = l.getLoopStmtsFor(buf); in parallelizeOuterLoops()847 std::vector<ForPtr> loops = l.getLoopStmtsFor(buf); in transformLoops()899 std::vector<ForPtr> loops = l.getLoopStmtsFor(buf); in transformLoops()
485 return self.getLoopStmtsFor(t); in initTensorExprBindings()
80 auto loops = nest.getLoopStmtsFor(conv); in conv2d_depthwise_static()82 loops = nest.getLoopStmtsFor(conv); in conv2d_depthwise_static()
34 std::vector<ForPtr> loops = ln->getLoopStmtsFor(target); in optimizePointwise()