| 3525 | } |
| 3526 | |
| 3527 | XlaOp While(const XlaComputation& condition, const XlaComputation& body, |
| 3528 | const XlaOp init) { |
| 3529 | return init.builder()->While(condition, body, init); |
| 3530 | } |
| 3531 | |
| 3532 | XlaOp Conditional(const XlaOp predicate, const XlaOp true_operand, |
| 3533 | const XlaComputation& true_computation, |