Home
last modified time | relevance | path

Searched refs:unsafeFuseLoops (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.h341 static bool unsafeFuseLoops(const std::vector<ForPtr>& loops, ForPtr* fused);
H A Dloopnest.cpp1893 bool LoopNest::unsafeFuseLoops( in unsafeFuseLoops() function in torch::jit::tensorexpr::LoopNest
1994 bool ret = unsafeFuseLoops(loops_copy, &fused_copy); in fuseLoops()
2000 return unsafeFuseLoops(loops, fused); in fuseLoops()
H A Dtensorexpr_init.cpp599 LoopNest::unsafeFuseLoops(loops, &fused_loop); in initTensorExprBindings()