Searched refs:new_while_init (Results 1 – 3 of 3) sorted by relevance
118 HloInstruction* new_while_init = in MakeInstructionsLiveIn() local123 new_while_body, new_while_init)); in MakeInstructionsLiveIn()
386 HloInstruction* new_while_init = while_parent->AddInstruction( in CreateNewWhileInit() local388 return WhileInitContext{new_while_init, tuple_index_to_old_buffer}; in CreateNewWhileInit()
156 auto* new_while_init = computation->AddInstruction( in RemoveDeadTupleIndices() local164 new_while_init)); in RemoveDeadTupleIndices()