| 3511 | } |
| 3512 | |
| 3513 | XlaOp RngNormal(const XlaOp mu, const XlaOp sigma, const Shape& shape) { |
| 3514 | return mu.builder()->RngNormal(mu, sigma, shape); |
| 3515 | } |
| 3516 | |
| 3517 | XlaOp RngUniform(const XlaOp a, const XlaOp b, const Shape& shape) { |
| 3518 | return a.builder()->RngUniform(a, b, shape); |