Home
last modified time | relevance | path

Searched refs:num_skips (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/lib/random/
Drandom_distributions.cc22 void SingleSampleAdapter<PhiloxRandom>::SkipFromGenerator(uint64 num_skips) { in SkipFromGenerator() argument
24 generator_->Skip(num_skips); in SkipFromGenerator()
Drandom_distributions.h330 void Skip(uint64 num_skips) { in Skip() argument
331 if (!num_skips) { in Skip()
335 if (num_skips <= num_unused_results) { in Skip()
336 used_result_index_ += num_skips; in Skip()
339 num_skips -= num_unused_results; in Skip()
341 SkipFromGenerator(num_skips / kNativeElementCount); in Skip()
342 num_skips = num_skips % kNativeElementCount; in Skip()
343 if (num_skips) { in Skip()
345 used_result_index_ = num_skips; in Skip()
354 void SkipFromGenerator(uint64 num_skips) { in SkipFromGenerator() argument
[all …]