Searched refs:UniformFloatingPointDistribution (Results 1 – 6 of 6) sorted by relevance
70 RngOutput UniformFloatingPointDistribution(XlaOp key, XlaOp initial_state,
581 RngOutput UniformFloatingPointDistribution(XlaOp key, XlaOp initial_state, in UniformFloatingPointDistribution() function619 RngOutput bits_state = UniformFloatingPointDistribution( in NormalFloatingPointDistribution()
101 result = UniformFloatingPointDistribution( in GetComputationForRng()
91 return xla::UniformFloatingPointDistribution( in StatelessRngUniform()
74 return xla::UniformFloatingPointDistribution( in StatefulRngUniform()
124 return xla::UniformFloatingPointDistribution(key, counter, generator, in StatelessRngUniformV2()