Searched refs:ReserveSamples128 (Results 1 – 3 of 3) sorted by relevance
58 random::PhiloxRandom ReserveSamples128(int64 samples);62 return ReserveSamples128((samples + 3) / 4); in ReserveSamples32()69 return ReserveSamples128(conservative_sample_count); in ReserveRandomOutputs()
54 random::PhiloxRandom GuardedPhiloxRandom::ReserveSamples128(int64 samples) { in ReserveSamples128() function in tensorflow::GuardedPhiloxRandom
425 generator_.ReserveSamples128(num_batches * 2 * functor::kMaxIterations * in Compute()