Searched defs:branch_computation (Results 1 – 4 of 4) sorted by relevance
942 for (auto branch_computation : instruction->branch_computations()) { in CheckLayouts() local
2080 HloComputation* HloInstruction::branch_computation(int b) const { in branch_computation() function in xla::HloInstruction
2089 for (auto branch_computation : conditional->branch_computations()) { in CheckConditionalBuffersShareAllocation() local2162 HloComputation* branch_computation = hlo->branch_computation(j); in BuildConditionalThunk() local
1925 for (const XlaComputation* branch_computation : branch_computations) { in Conditional() local