Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.cpp1780 const std::unordered_set<VarPtr>& outer_loop_vars) { in areIndicesLoopIndependent() argument
1787 if (doesExprContainAnyVar(expr1, outer_loop_vars) || in areIndicesLoopIndependent()
1788 doesExprContainAnyVar(expr2, outer_loop_vars)) { in areIndicesLoopIndependent()
1801 std::unordered_set<VarPtr> outer_loop_vars = {loop->var()}; in hasLoopCarriedDependence() local
1804 outer_loop_vars.insert(l->var()); in hasLoopCarriedDependence()
1854 aStore->indices(), bLoad->indices(), outer_loop_vars)) { in hasLoopCarriedDependence()
1867 bStore->indices(), aLoad->indices(), outer_loop_vars)) { in hasLoopCarriedDependence()
1880 aStore->indices(), bStore->indices(), outer_loop_vars)) { in hasLoopCarriedDependence()