Home
last modified time | relevance | path

Searched refs:new_while_init (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_util.cc122 HloInstruction* new_while_init = in MakeInstructionsLiveIn() local
127 new_while_body, new_while_init)); in MakeInstructionsLiveIn()
H A Dwhile_loop_all_reduce_code_motion.cc594 HloInstruction* new_while_init = while_parent->AddInstruction( in CreateNewWhileInit() local
596 return WhileInitContext{new_while_init, tuple_index_to_old_buffer}; in CreateNewWhileInit()
H A Dwhile_loop_simplifier.cc167 auto* new_while_init = computation->AddInstruction( in RemoveDeadTupleIndices() local
175 new_while_init)); in RemoveDeadTupleIndices()