Searched refs:new_while_body (Results 1 – 4 of 4) 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()133 HloInstruction* while_body_param = new_while_body->parameter_instruction(0); in MakeInstructionsLiveIn()138 new_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in MakeInstructionsLiveIn()
142 std::unique_ptr<HloComputation> new_while_body = in RemoveDeadTupleIndices() local163 module->AddEmbeddedComputation(std::move(new_while_body)), in RemoveDeadTupleIndices()599 std::unique_ptr<HloComputation> new_while_body = in TryRemoveConstantParams() local622 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveConstantParams()925 std::unique_ptr<HloComputation> new_while_body = in TryFlattenNestedTuples() local945 module->AddEmbeddedComputation(std::move(new_while_body)), in TryFlattenNestedTuples()1164 std::unique_ptr<HloComputation> new_while_body = in TryMergeInductionVariables() local1180 module->AddEmbeddedComputation(std::move(new_while_body)), in TryMergeInductionVariables()
284 HloComputation* new_while_body = in TryHoistingInvariantInstructionsFromWhileBody() local291 TF_RETURN_IF_ERROR(new_while_body->ReplaceInstruction( in TryHoistingInvariantInstructionsFromWhileBody()
316 HloComputation* new_while_body = in TryHoistingInvariantInstructionsFromWhileBody() local323 TF_RETURN_IF_ERROR(new_while_body->ReplaceInstruction( in TryHoistingInvariantInstructionsFromWhileBody()