Searched refs:RngOp (Results 1 – 2 of 2) sorted by relevance
1780 XlaOp XlaBuilder::RngOp(RandomDistribution distribution, in RngOp() function in xla::XlaBuilder1811 return RngOp(RandomDistribution::RNG_NORMAL, {mu, sigma}, shape); in RngNormal()1816 return RngOp(RandomDistribution::RNG_UNIFORM, {a, b}, shape); in RngUniform()
610 XlaOp RngOp(RandomDistribution distribution,