| 3530 | } |
| 3531 | |
| 3532 | XlaOp Conditional(const XlaOp predicate, const XlaOp true_operand, |
| 3533 | const XlaComputation& true_computation, |
| 3534 | const XlaOp false_operand, |
| 3535 | const XlaComputation& false_computation) { |
| 3536 | return predicate.builder()->Conditional(predicate, true_operand, |
| 3537 | true_computation, false_operand, |
| 3538 | false_computation); |
| 3539 | } |
| 3540 | |
| 3541 | XlaOp Conditional(const XlaOp branch_index, |
| 3542 | absl::Span<const XlaComputation* const> branch_computations, |