Searched refs:UniformIntDistribution (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.h | 77 RngOutput UniformIntDistribution(XlaOp key, XlaOp initial_state,
|
| D | prng.cc | 592 RngOutput UniformIntDistribution(XlaOp key, XlaOp initial_state, in UniformIntDistribution() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | rng_expander.cc | 105 result = UniformIntDistribution(key, state, generator, a_or_mean, in GetComputationForRng()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateless_random_ops.cc | 98 return UniformIntDistribution( in StatelessRngUniform()
|
| D | stateful_random_ops.cc | 80 return UniformIntDistribution(key, initial_state, BitGen(alg), minval, in StatefulRngUniform()
|
| D | stateless_random_ops_v2.cc | 130 return UniformIntDistribution(key, counter, generator, minval, maxval, in StatelessRngUniformV2()
|