Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Djit_trace.cpp110 LoopView lv(n); in traceLoopCounter()
123 LoopView lv(n); in traceLoopCounter()
145 LoopView lv(n); in traceLoop()
153 LoopView lv(n); in traceLoop()
173 insertTracingNodes(LoopView(n).bodyBlock(), pr, td); in traceLoop()
181 LoopView(n).bodyBlock()->appendNode(opt_pn); in traceLoop()
192 LoopView lv(n); in traceLoop()
H A Dsymbolic_shape_registry.cpp128 LoopView loop(node); in checkForWhileLoop()
129 if (loop.loopType() != LoopView::For) { in checkForWhileLoop()
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dcanonicalize_modified_loop.cpp16 LoopView loop(n); in canonicalizeModifiedLoop()
17 if (loop.loopType() != LoopView::ModifiedLoop) { in canonicalizeModifiedLoop()
H A Dexit_transforms.cpp170 LoopView loop(node); in transformLoop()
/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/
H A Dir_views.h47 struct LoopView { struct
48 explicit LoopView(Node* node) : node_(node) { in LoopView() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dloop_unrolling.cpp246 LoopView loop_view(loop); in addCondAsOutput()
322 auto orig_loop = LoopView(n); in PeelLoop()
336 LoopView new_lv(peeled_copy); in PeelLoop()
H A Dliveness.cpp45 LoopView lv(it); in insertExplicitUsesOfLoopCounters()
126 LoopView lv(it); in processBlock()
H A Dbailout_graph.cpp121 LoopView lv(outer_node); in buildBailOutLoop()
171 LoopView new_lv(new_loop); in buildBailOutLoop()
H A Dcanonicalize.cpp198 LoopView(n).permuteLoopCarried(new_indices); in CanonicalizeLoopOutputs()
H A Ddead_code_elimination.cpp117 auto loop = LoopView(outerNode); in markReturnNode()
H A Dshape_analysis.cpp83 LoopView loop(node); in processLoop()
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpython_print.cpp575 void printLoop(LoopView stmt) { in printLoop()
581 if (loop_type == LoopView::ModifiedLoop) { in printLoop()
589 bool emit_as_for_loop = loop_type == LoopView::For; in printLoop()
797 printLoop(LoopView(node)); in printNode()