Searched refs:BitGeneratorTy (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.h | 44 using BitGeneratorTy = std::function<RngOutput(XlaOp key, XlaOp initial_state, variable
71 BitGeneratorTy bit_generator,
78 BitGeneratorTy bit_generator, XlaOp minval,
85 BitGeneratorTy bit_generator,
|
| D | prng.cc | 582 BitGeneratorTy bit_generator, in UniformFloatingPointDistribution()
593 BitGeneratorTy bit_generator, XlaOp minval, in UniformIntDistribution()
612 BitGeneratorTy bit_generator, in NormalFloatingPointDistribution()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateless_random_ops.cc | 39 xla::BitGeneratorTy GetBitGeneratorForDevice( in GetBitGeneratorForDevice()
|
| D | stateful_random_ops.cc | 41 xla::BitGeneratorTy BitGen(Algorithm alg) { in BitGen()
|