Searched refs:branch_id (Results 1 – 4 of 4) sorted by relevance
265 for (int64 branch_id = 0; branch_id < conditional_op->branch_count(); in ReplaceRootWithEmptyTupleIfNoUsers() local266 ++branch_id) { in ReplaceRootWithEmptyTupleIfNoUsers()269 conditional_op->branch_computation(branch_id)->Clone()); in ReplaceRootWithEmptyTupleIfNoUsers()270 conditional_op->set_branch_computation(branch_id, branch_computation); in ReplaceRootWithEmptyTupleIfNoUsers()379 for (int branch_id = 0; branch_id < conditional_op->branch_count(); in RemoveUnusedTupleElements() local380 ++branch_id) { in RemoveUnusedTupleElements()381 HloComputation* old_branch = conditional_op->branch_computation(branch_id); in RemoveUnusedTupleElements()385 conditional_op->set_branch_computation(branch_id, cloned_branch); in RemoveUnusedTupleElements()
249 uint32_t branch_id = phi->GetSingleWordInOperand(i + 1); in SplitLoopHeader() local250 if (branch_id == latch_block->id()) { in SplitLoopHeader()252 header_phi_ops.push_back({SPV_OPERAND_TYPE_ID, {branch_id}}); in SplitLoopHeader()255 preheader_phi_ops.push_back(branch_id); in SplitLoopHeader()