Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Ddead_code_elimination.cpp366 auto loop_input_offset = 2; // offset of loop carried deps in input list in removeDeadLoopOutputs() local
374 logDeadLoopOutputs(node, i, loop_input_offset, loop_body_offset); in removeDeadLoopOutputs()
376 node->removeInput(loop_input_offset + i); in removeDeadLoopOutputs()
386 size_t loop_input_offset, in logDeadLoopOutputs() argument
391 loop_input_offset + i, in logDeadLoopOutputs()
H A Dconstant_propagation.cpp211 auto loop_input_offset = 2; // offset of loop carried deps in input list in removeLoopNode() local
214 n->inputs().at(i + loop_input_offset)); in removeLoopNode()
314 auto loop_input_offset = 2; // offset of loop carried deps in input list in removeExtraLoopOutputs() local
322 auto node_input = node->inputs().at(loop_input_offset + i); in removeExtraLoopOutputs()
328 node->removeInput(loop_input_offset + i); in removeExtraLoopOutputs()