Searched refs:UniformSampler (Results 1 – 4 of 4) sorted by relevance
135 class UniformSampler : public RangeSampler {137 explicit UniformSampler(int64 range);139 ~UniformSampler() override {} in ~UniformSampler()
147 UniformSampler::UniformSampler(int64 range) in UniformSampler() function in tensorflow::UniformSampler150 int64 UniformSampler::Sample(random::SimplePhilox* rnd) const { in Sample()154 float UniformSampler::Probability(int64 value) const { return inv_range_; } in Probability()
82 sampler_.reset(new UniformSampler(10)); in TEST_F()89 sampler_.reset(new UniformSampler(10)); in TEST_F()94 sampler_.reset(new UniformSampler(10)); in TEST_F()
124 SimpleCandidateSamplerOp<UniformSampler>);