Searched defs:branch_computation (Results 1 – 5 of 5) sorted by relevance
267 auto branch_computation = in ReplaceRootWithEmptyTupleIfNoUsers() local
997 for (auto branch_computation : instruction->branch_computations()) { in CheckLayouts() local
2221 HloComputation* HloInstruction::branch_computation(int b) const { in branch_computation() function in xla::HloInstruction
1706 for (auto branch_computation : conditional->branch_computations()) { in CheckConditionalBuffersShareAllocation() local1779 HloComputation* branch_computation = hlo->branch_computation(j); in BuildConditionalThunk() local
1892 for (const XlaComputation* branch_computation : branch_computations) { in ConditionalImpl() local