Searched defs:branch_computation (Results 1 – 7 of 7) sorted by relevance
301 auto branch_computation = in ReplaceRootWithEmptyTupleIfNoUsers() local
1218 HloComputation* branch_computation = inst->branch_computation(i); in BoundariesToMoveInOrOut() local
998 for (auto branch_computation : instruction->branch_computations()) { in CheckLayouts() local
1306 HloComputation* branch_computation = hlo->branch_computation(branch_index); in HandleConditional() local
2522 HloComputation* HloInstruction::branch_computation(int b) const { in branch_computation() function in xla::HloInstruction
3812 for (auto branch_computation : conditional->branch_computations()) { in CheckConditionalBuffersShareAllocation() local3912 HloComputation* branch_computation = hlo->branch_computation(j); in BuildConditionalThunk() local
2401 for (const XlaComputation* branch_computation : branch_computations) { in ConditionalImpl() local