Searched refs:UniformSampler (Results 1 – 7 of 7) sorted by relevance
135 class UniformSampler : public RangeSampler {137 explicit UniformSampler(int64 range);139 ~UniformSampler() override {} in ~UniformSampler()
148 UniformSampler::UniformSampler(int64 range) in UniformSampler() function in tensorflow::UniformSampler151 int64 UniformSampler::Sample(random::SimplePhilox* rnd) const { in Sample()155 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>);
214 type Sampler: UniformSampler<X = Self>;227 pub trait UniformSampler: Sized { trait278 let uniform: Self = UniformSampler::new(low, high); in sample_single()295 let uniform: Self = UniformSampler::new_inclusive(low, high); in sample_single_inclusive()431 impl UniformSampler for UniformInt<$ty> {449 UniformSampler::new_inclusive(low, high - 1)585 impl UniformSampler for UniformInt<$ty> {597 UniformSampler::new_inclusive(low, high - 1)747 impl UniformSampler for UniformChar {820 impl UniformSampler for UniformFloat<$ty> {[all …]
11 use crate::distributions::uniform::{SampleBorrow, SampleUniform, UniformSampler};
63 - Add `UniformSampler::sample_single_inclusive` (#1003)