Searched refs:bit_generator (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.h | 71 BitGeneratorTy bit_generator,
78 BitGeneratorTy bit_generator, XlaOp minval,
85 BitGeneratorTy bit_generator,
|
| D | prng.cc | 582 BitGeneratorTy bit_generator, in UniformFloatingPointDistribution() argument
585 RngOutput bits_state = bit_generator(key, initial_state, shape); in UniformFloatingPointDistribution()
593 BitGeneratorTy bit_generator, XlaOp minval, in UniformIntDistribution() argument
595 RngOutput bits_state = bit_generator(key, initial_state, shape); in UniformIntDistribution()
612 BitGeneratorTy bit_generator, in NormalFloatingPointDistribution() argument
620 key, initial_state, bit_generator, in NormalFloatingPointDistribution()
|