| 1728 | } |
| 1729 | |
| 1730 | XlaOp XlaBuilder::RngNormal(XlaOp mu, XlaOp sigma, const Shape& shape) { |
| 1731 | return RngOp(RandomDistribution::RNG_NORMAL, {mu, sigma}, shape); |
| 1732 | } |
| 1733 | |
| 1734 | XlaOp XlaBuilder::RngUniform(XlaOp a, XlaOp b, const Shape& shape) { |
| 1735 | return RngOp(RandomDistribution::RNG_UNIFORM, {a, b}, shape); |