Searched refs:new_while_body (Results 1 – 3 of 3) sorted by relevance
112 HloComputation* new_while_body; in MakeInstructionsLiveIn() local115 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()
245 std::unique_ptr<HloComputation> new_while_body = in TryRemoveDeadWhileParams() local266 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveDeadWhileParams()418 std::unique_ptr<HloComputation> new_while_body = in TryRemoveConstantParams() local441 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveConstantParams()713 std::unique_ptr<HloComputation> new_while_body = in TryFlattenNestedTuples() local733 module->AddEmbeddedComputation(std::move(new_while_body)), in TryFlattenNestedTuples()952 std::unique_ptr<HloComputation> new_while_body = in TryMergeInductionVariables() local968 module->AddEmbeddedComputation(std::move(new_while_body)), in TryMergeInductionVariables()
283 HloComputation* new_while_body = in TryHoistingInvariantInstructionsFromWhileBody() local290 TF_RETURN_IF_ERROR(new_while_body->ReplaceInstruction( in TryHoistingInvariantInstructionsFromWhileBody()