Searched defs:branch_computation (Results 1 – 7 of 7) sorted by relevance
165 auto branch_computation = in ReplaceRootWithEmptyTupleIfNoUsers() local
1240 HloComputation* branch_computation = inst->branch_computation(i); in BoundariesToMoveInOrOut() local
1086 for (auto branch_computation : instruction->branch_computations()) { in CheckLayouts() local
1343 HloComputation* branch_computation = hlo->branch_computation(branch_index); in HandleConditional() local
2722 HloComputation* HloInstruction::branch_computation(int b) const { in branch_computation() function in xla::HloInstruction
658 mlir::Region* branch_computation = &conditional.branches()[j]; in EmitConditional() local
2429 for (const XlaComputation* branch_computation : branch_computations) { in ConditionalImpl() local