Searched refs:temp_new_while_body (Results 1 – 1 of 1) sorted by relevance
1157 HloComputation* temp_new_while_body = in TryMergeInductionVariables() local1165 temp_new_while_body->CloneWithReplacementPairs({ in TryMergeInductionVariables()1166 temp_new_while_body->root_instruction(), in TryMergeInductionVariables()1168 add_new_instr(temp_new_while_body->root_instruction()->Clone()), in TryMergeInductionVariables()1170 temp_new_while_body->parameter_instruction(0))), in TryMergeInductionVariables()1172 TF_RETURN_IF_ERROR(module->RemoveEmbeddedComputation(temp_new_while_body)); in TryMergeInductionVariables()