Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_util.cc112 HloComputation* new_while_body; in MakeInstructionsLiveIn() local
115 std::tie(new_while_body, inlined_instructions_map), in MakeInstructionsLiveIn()
123 new_while_body, new_while_init)); in MakeInstructionsLiveIn()
132 HloInstruction* while_body_param = new_while_body->parameter_instruction(0); in MakeInstructionsLiveIn()
137 new_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in MakeInstructionsLiveIn()
Dwhile_loop_simplifier.cc245 std::unique_ptr<HloComputation> new_while_body = in TryRemoveDeadWhileParams() local
266 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveDeadWhileParams()
418 std::unique_ptr<HloComputation> new_while_body = in TryRemoveConstantParams() local
441 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveConstantParams()
713 std::unique_ptr<HloComputation> new_while_body = in TryFlattenNestedTuples() local
733 module->AddEmbeddedComputation(std::move(new_while_body)), in TryFlattenNestedTuples()
952 std::unique_ptr<HloComputation> new_while_body = in TryMergeInductionVariables() local
968 module->AddEmbeddedComputation(std::move(new_while_body)), in TryMergeInductionVariables()
Dwhile_loop_invariant_code_motion.cc283 HloComputation* new_while_body = in TryHoistingInvariantInstructionsFromWhileBody() local
290 TF_RETURN_IF_ERROR(new_while_body->ReplaceInstruction( in TryHoistingInvariantInstructionsFromWhileBody()