Searched refs:seed_bytes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | rng.cc | 23 bool RngSupport::CheckSeed(const uint8 *seed, uint64 seed_bytes) { in CheckSeed() argument
26 if (seed_bytes < kMinSeedBytes) { in CheckSeed()
27 LOG(INFO) << "Insufficient RNG seed data specified: " << seed_bytes in CheckSeed()
33 if (seed_bytes > kMaxSeedBytes) { in CheckSeed()
34 LOG(INFO) << "Too much RNG seed data specified: " << seed_bytes in CheckSeed()
|
| D | rng.h | 84 uint64 seed_bytes) = 0;
87 static bool CheckSeed(const uint8 *seed, uint64 seed_bytes);
|
| D | stream.cc | 4662 Stream &Stream::ThenSetRngSeed(const uint8 *seed, uint64 seed_bytes) { in ThenSetRngSeed() argument
4663 VLOG_CALL(PARAM(seed), PARAM(seed_bytes)); in ThenSetRngSeed()
4667 CheckError(rng->SetSeed(this, seed, seed_bytes)); in ThenSetRngSeed()
4675 << "; bytes: " << seed_bytes; in ThenSetRngSeed()
|
| D | stream.h | 1660 Stream &ThenSetRngSeed(const uint8 *seed, uint64 seed_bytes);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_rng.cc | 199 bool GpuRng::SetSeed(Stream* stream, const uint8* seed, uint64 seed_bytes) { in SetSeed() argument
203 if (!CheckSeed(seed, seed_bytes)) { in SetSeed()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_rng.cc | 254 bool GpuRng::SetSeed(Stream* stream, const uint8* seed, uint64 seed_bytes) { in SetSeed() argument
258 if (!CheckSeed(seed, seed_bytes)) { in SetSeed()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_rng.h | 68 bool SetSeed(Stream* stream, const uint8* seed, uint64 seed_bytes) override;
|