Searched refs:NormalFloatingPointDistribution (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | prng.h | 84 RngOutput NormalFloatingPointDistribution(XlaOp key, XlaOp initial_state,
|
D | prng.cc | 611 RngOutput NormalFloatingPointDistribution(XlaOp key, XlaOp initial_state, in NormalFloatingPointDistribution() function
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | rng_expander.cc | 93 NormalFloatingPointDistribution(key, state, generator, result_shape) in GetComputationForRng()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | stateless_random_ops.cc | 309 xla::NormalFloatingPointDistribution( in Compile()
|
D | stateful_random_ops.cc | 274 xla::RngOutput value_state = xla::NormalFloatingPointDistribution( in Compile()
|
D | stateless_random_ops_v2.cc | 392 auto result = xla::NormalFloatingPointDistribution(key, counter, generator, in Compile()
|