Searched refs:PhiloxRandom (Results 1 – 5 of 5) sorted by relevance
51 void Init(random::PhiloxRandom::ResultType counter, random::PhiloxRandom::Key key);56 random::PhiloxRandom ReserveSamples128(int64 samples);59 random::PhiloxRandom ReserveSamples32(int64 samples) { in ReserveSamples32()64 random::PhiloxRandom ReserveRandomOutputs(int64 output_count, int multiplier) { in ReserveRandomOutputs()71 random::PhiloxRandom generator_; // GUARDED_BY(mu_);
29 generator_ = random::PhiloxRandom(seed, seed2); in Init()33 void GuardedPhiloxRandom::Init(random::PhiloxRandom::ResultType counter, in Init()34 random::PhiloxRandom::Key key) { in Init()37 generator_ = random::PhiloxRandom(counter, key); in Init()41 random::PhiloxRandom GuardedPhiloxRandom::ReserveSamples128(int64 samples) { in ReserveSamples128()
105 class PhiloxRandom {118 PhiloxRandom() {} in PhiloxRandom() function121 explicit PhiloxRandom(uint64 seed) { in PhiloxRandom() function127 explicit PhiloxRandom(uint64 seed_lo, uint64 seed_hi) { in PhiloxRandom() function135 PhiloxRandom(ResultType counter, Key key) : counter_(counter), key_(key) {} in PhiloxRandom() function
34 explicit SimplePhilox(PhiloxRandom* gen) : single_(gen) {} in SimplePhilox()60 SingleSampleAdapter<PhiloxRandom> single_;
65 tensorflow::random::PhiloxRandom rng(kFixedRandomSeed1); in Invoke()