Searched refs:getLoopBodyFor (Results 1 – 5 of 5) sorted by relevance
49 StmtPtr getLoopBodyFor(const Tensor&) const;50 StmtPtr getLoopBodyFor(BufPtr) const;
473 return self.getLoopBodyFor(t); in initTensorExprBindings()479 return self.getLoopBodyFor(b.node()); in initTensorExprBindings()
2598 StmtPtr cur_stmt = getLoopBodyFor(t); in getLoopStmtsFor()2603 StmtPtr cur_stmt = getLoopBodyFor(buf); in getLoopStmtsFor()2620 StmtPtr LoopNest::getLoopBodyFor(const Tensor& t) const { in getLoopBodyFor() function in torch::jit::tensorexpr::LoopNest2621 return getLoopBodyFor(t.buf()); in getLoopBodyFor()2624 StmtPtr LoopNest::getLoopBodyFor(BufPtr buf) const { in getLoopBodyFor() function in torch::jit::tensorexpr::LoopNest
1816 l.computeInline(l.getLoopBodyFor(e)); in TEST()1817 l.computeInline(l.getLoopBodyFor(f)); in TEST()1878 LoopNest::computeAt(l.getLoopBodyFor(A), loops[0]); in TEST()1942 LoopNest::computeAt(l.getLoopBodyFor(p), loops[0]); in TEST()1968 LoopNest::computeAt(l.getLoopBodyFor(p), loops[1]); in TEST()2034 LoopNest::computeAt(l.getLoopBodyFor(A), loops[0]); in TEST()2065 LoopNest::computeAt(l.getLoopBodyFor(A), loops[1]); in TEST()2139 LoopNest::computeAt(l.getLoopBodyFor(p), loops[0]); in TEST()2175 LoopNest::computeAt(l.getLoopBodyFor(p), loops[1]); in TEST()2246 LoopNest::computeAt(l.getLoopBodyFor(A), loops[0]); in TEST()[all …]
134 StmtPtr body = l.getLoopBodyFor(c); in TEST()264 StmtPtr body = l.getLoopBodyFor(c); in TEST()