| 20 | namespace random { |
| 21 | template <> |
| 22 | void SingleSampleAdapter<PhiloxRandom>::SkipFromGenerator(uint64 num_skips) { |
| 23 | // Use the O(1) PhiloxRandom::Skip instead of the default O(N) impl. |
| 24 | generator_->Skip(num_skips); |
| 25 | } |
| 26 | } // namespace random |
| 27 | } // namespace tensorflow |