Searched refs:temp_new_while_body (Results 1 – 1 of 1) sorted by relevance
945 HloComputation* temp_new_while_body = in TryMergeInductionVariables() local953 temp_new_while_body->CloneWithReplacementPairs({ in TryMergeInductionVariables()954 temp_new_while_body->root_instruction(), in TryMergeInductionVariables()956 add_new_instr(temp_new_while_body->root_instruction()->Clone()), in TryMergeInductionVariables()958 temp_new_while_body->parameter_instruction(0))), in TryMergeInductionVariables()960 TF_RETURN_IF_ERROR(module->RemoveEmbeddedComputation(temp_new_while_body)); in TryMergeInductionVariables()