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