Searched refs:induction_var_init (Results 1 – 2 of 2) sorted by relevance
90 auto induction_var_init = in BuildWhileInstruction() local99 HloInstruction::CreateTuple({induction_var_init, data_init})) in BuildWhileInstruction()101 HloInstruction::CreateTuple({data_init, induction_var_init})); in BuildWhileInstruction()
638 auto induction_var_init = builder.AddInstruction( in BuildWhileInstruction() local649 HloInstruction::CreateTuple({induction_var_init, inner_init})); in BuildWhileInstruction()657 HloInstruction::CreateTuple({induction_var_init, data_init})); in BuildWhileInstruction()758 auto induction_var_init = builder->AddInstruction( in BuildWhileInstructionWithCustomInit() local765 HloInstruction::CreateTuple({induction_var_init, data_init})); in BuildWhileInstructionWithCustomInit()