Searched refs:while_body_param (Results 1 – 6 of 6) sorted by relevance
49 auto* while_body_param = while_body_comp->parameter_instruction(0); in RunWhileDCE() local74 while_body_param->shape().tuple_shapes(i), while_body_param, in RunWhileDCE()
53 HloInstruction* while_body_param = while_body->parameter_instruction(0); in CreateLoopInvariantCopy() local65 return old_operand == while_body_param in CreateLoopInvariantCopy()93 next_operand == while_body_param) { in CreateLoopInvariantCopy()
54 auto* while_body_param = while_body_comp->parameter_instruction(0); in WhileBodyHasPassThroughTupleElement() local62 operand->operand(0) == while_body_param) { in WhileBodyHasPassThroughTupleElement()
132 HloInstruction* while_body_param = new_while_body->parameter_instruction(0); in MakeInstructionsLiveIn() local139 while_body_param, i))); in MakeInstructionsLiveIn()
137 auto* while_body_param = while_body->parameter_instruction(0); in GetLoopInductionVarTupleIdx() local139 GetGTEOperandIndex(while_body_inc, while_body_param); in GetLoopInductionVarTupleIdx()
518 auto while_body_param = while_body->parameter_instruction(0); in TryPropagateConstant() local530 m::GetTupleElement(m::Op().Is(while_body_param), i) in TryPropagateConstant()