| 54 | } |
| 55 | |
| 56 | void RandomSeedGenerator::Reset() { |
| 57 | mutex_lock l(mu_); |
| 58 | // Reset the generators based on the current seeds. |
| 59 | parent_generator_ = random::PhiloxRandom(seed_, seed2_); |
| 60 | generator_ = |
| 61 | random::SingleSampleAdapter<random::PhiloxRandom>(&parent_generator_); |
| 62 | generator_.Skip(num_random_samples_); |
| 63 | } |
| 64 | |
| 65 | void RandomSeedGenerator::Serialize(OpKernelContext* ctx) { |
| 66 | mutex_lock l(mu_); |
nothing calls this directly
no test coverage detected