Home
last modified time | relevance | path

Searched refs:computeInline (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/torch/csrc/jit/tensorexpr/
Dloopnest.h121 bool computeInline(const StmtPtr& s);
122 bool computeInline(const BufPtr& b);
Dtensorexpr_init.cpp660 [](LoopNest& self, const StmtPtr& s) { self.computeInline(s); }, in initTensorExprBindings()
665 self.computeInline(b.node()); in initTensorExprBindings()
Dloopnest_randomization.cpp230 l.computeInline(bufs[buf_number]); in loopnestRandomization()
Dloopnest.cpp925 bool LoopNest::computeInline(const BufPtr& b) { in computeInline() function in torch::jit::tensorexpr::LoopNest
941 bool LoopNest::computeInline(const StmtPtr& s) { in computeInline() function in torch::jit::tensorexpr::LoopNest
947 return computeInline(s_store->buf()); in computeInline()
1010 computeInline(b); in inlineIntermediateBufs()
/external/pytorch/test/cpp/tensorexpr/
Dtest_loopnest.cpp1028 l2.computeInline(x.buf()); in TEST()
1115 l.computeInline(x.buf()); in InlineFunc01Helper()
1117 l.computeInline(y.buf()); in InlineFunc01Helper()
1221 l1.computeInline(x.buf()); in TEST()
1257 l1.computeInline(x.buf()); in TEST()
1289 l1.computeInline(x.buf()); in TEST()
1341 l2.computeInline(x.buf()); in TEST()
1384 l1.computeInline(x.buf()); in TEST()
1406 ASSERT_FALSE(l.computeInline(a.buf())); in TEST()
1418 l.computeInline(a.buf()); in TEST()
[all …]
Dtutorial.cpp284 nest.computeInline(C.buf()); in main()
Dtest_reductions.cpp1139 ASSERT_FALSE(l1.computeInline(x.buf())); in TEST()
1172 l2.computeInline(x.buf()); in TEST()
1229 l2.computeInline(x.buf()); in TEST()
Dtest_memdependency.cpp2895 l.computeInline(c.buf()); in TEST()