Searched refs:RangeSampler (Results 1 – 4 of 4) sorted by relevance
37 class RangeSampler {39 explicit RangeSampler(int64 range) : range_(range) { CHECK_GT(range_, 0); } in RangeSampler() function40 virtual ~RangeSampler();110 class AllSampler : public RangeSampler {135 class UniformSampler : public RangeSampler {149 class LogUniformSampler : public RangeSampler {164 class ThreadUnsafeUnigramSampler : public RangeSampler {181 class UnigramSampler : public RangeSampler {211 class FixedUnigramSampler : public RangeSampler {
36 RangeSampler::~RangeSampler() {} in ~RangeSampler()38 void RangeSampler::SampleBatch(random::SimplePhilox* rnd, bool unique, in SampleBatch()45 void RangeSampler::SampleBatchGetExpectedCount( in SampleBatchGetExpectedCount()78 void RangeSampler::SampleBatchGetExpectedCountAvoid( in SampleBatchGetExpectedCountAvoid()123 AllSampler::AllSampler(int64 range) : RangeSampler(range) {} in AllSampler()149 : RangeSampler(range), inv_range_(1.0 / range) {} in UniformSampler()158 : RangeSampler(range), log_range_(log1p(range)) {} in LogUniformSampler()181 : RangeSampler(range), picker_(range) { in ThreadUnsafeUnigramSampler()204 : RangeSampler(range), unsafe_sampler_(range) { in UnigramSampler()240 : RangeSampler(range), in FixedUnigramSampler()[all …]
102 void set_sampler(RangeSampler* sampler) { sampler_.reset(sampler); } in set_sampler()108 std::unique_ptr<RangeSampler> sampler_;
78 std::unique_ptr<RangeSampler> sampler_;