Searched defs:branch_computation (Results 1 – 7 of 7) sorted by relevance
167 auto branch_computation = in ReplaceRootWithEmptyTupleIfNoUsers() local
1335 HloComputation* branch_computation = inst->branch_computation(i); in BoundariesToMoveInOrOut() local
1212 for (const auto branch_computation : branch_computations) { in CheckLayouts() local
1481 HloComputation* branch_computation = hlo->branch_computation(branch_index); in HandleConditional() local
2883 HloComputation* HloInstruction::branch_computation(int b) const { in branch_computation() function in xla::HloInstruction
649 mlir::Region* branch_computation = &conditional.getBranches()[j]; in EmitConditional() local
2556 for (const XlaComputation* branch_computation : branch_computations) { in ConditionalImpl() local