Searched refs:while_body_param (Results 1 – 7 of 7) sorted by relevance
52 auto* while_body_param = while_body_comp->parameter_instruction(0); in RunWhileDCE() local78 while_body_param->shape().tuple_shapes(i), while_body_param, in RunWhileDCE()
101 auto* while_body_param = while_body->parameter_instruction(0); in GetAuxiliaryLoopInductionVars() local103 VLOG(2) << "the parameter instr:" << while_body_param->ToShortString(); in GetAuxiliaryLoopInductionVars()104 VLOG(2) << "the parameter user count:" << while_body_param->users().size(); in GetAuxiliaryLoopInductionVars()105 if (while_body_param == nullptr) return aux_ind_gte; in GetAuxiliaryLoopInductionVars()115 for (const HloInstruction* indx_instr : while_body_param->users()) { in GetAuxiliaryLoopInductionVars()279 auto* while_body_param = while_body->parameter_instruction(0); in GetLoopInductionVarTupleIdx() local281 GetGTEOperandIndex(while_body_inc, while_body_param); in GetLoopInductionVarTupleIdx()
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()
67 HloInstruction* while_body_param = while_body->parameter_instruction(0); in CreateLoopInvariantCopy() local81 return old_operand == while_body_param in CreateLoopInvariantCopy()104 if (next_operand == while_body_param || in CreateLoopInvariantCopy()
53 auto* while_body_param = while_body_comp->parameter_instruction(0); in WhileBodyHasPassThroughTupleElement() local61 operand->operand(0) == while_body_param) { in WhileBodyHasPassThroughTupleElement()
133 HloInstruction* while_body_param = new_while_body->parameter_instruction(0); in MakeInstructionsLiveIn() local140 while_body_param, i))); in MakeInstructionsLiveIn()
730 auto while_body_param = while_body->parameter_instruction(0); in TryPropagateConstant() local742 m::GetTupleElement(m::Op().Is(while_body_param), i) in TryPropagateConstant()