Searched refs:new_branch_computations (Results 1 – 1 of 1) sorted by relevance
1429 std::vector<HloComputation*> new_branch_computations; in HandleConditional() local1515 new_branch_computations.push_back(new_computation); in HandleConditional()1532 for (int64_t j = 0; j < new_branch_computations.size(); ++j) { in HandleConditional()1534 new_branch_computations[j]->root_instruction(), index, i); in HandleConditional()1558 new_branch_computations[branch_index]->root_instruction(), in HandleConditional()1564 new_branch_computations[branch_index]->AddInstruction( in HandleConditional()1578 new_branch_computations[branch_index]->root_instruction(), in HandleConditional()1580 new_branch_computations[branch_index]->set_root_instruction( in HandleConditional()1592 new_branch_computations[0]->root_instruction()->shape(), in HandleConditional()1593 hlo->mutable_operand(0), new_branch_computations, new_operands)); in HandleConditional()