Searched refs:SetSeed (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | prng_test.cc | 58 SetSeed(seed); in UniformTest()
156 SetSeed(seed); in UniformChiSquared()
331 SetSeed(42); in XLA_TEST_F()
343 SetSeed(0); in XLA_TEST_F()
|
| D | client_library_test_base.h | 84 void SetSeed(uint64 seed) { execution_options_.set_seed(seed); } in SetSeed() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | rng.h | 83 virtual bool SetSeed(Stream *stream, const uint8 *seed,
|
| D | stream.cc | 4403 CheckError(rng->SetSeed(this, seed, seed_bytes)); in ThenSetRngSeed()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_rng.h | 67 bool SetSeed(Stream* stream, const uint8* seed, uint64 seed_bytes) override;
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_rng.cc | 199 bool GpuRng::SetSeed(Stream* stream, const uint8* seed, uint64 seed_bytes) { in SetSeed() function in stream_executor::gpu::GpuRng
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_rng.cc | 253 bool GpuRng::SetSeed(Stream* stream, const uint8* seed, uint64 seed_bytes) { in SetSeed() function in stream_executor::gpu::GpuRng
|