Searched refs:UnigramSampler (Results 1 – 4 of 4) sorted by relevance
203 UnigramSampler::UnigramSampler(int64 range) in UnigramSampler() function in tensorflow::UnigramSampler208 int64 UnigramSampler::Sample(random::SimplePhilox* rnd) const { in Sample()213 float UnigramSampler::Probability(int64 value) const { in Probability()219 void UnigramSampler::SampleBatchGetExpectedCountAvoid( in SampleBatchGetExpectedCountAvoid()230 void UnigramSampler::Update(ArraySlice<int64> values) { in Update()
181 class UnigramSampler : public RangeSampler {183 explicit UnigramSampler(int64 range);184 ~UnigramSampler() override {} in ~UnigramSampler()
118 sampler_.reset(new UnigramSampler(10)); in TEST_F()128 sampler_.reset(new UnigramSampler(10)); in TEST_F()135 sampler_.reset(new UnigramSampler(10)); in TEST_F()141 sampler_.reset(new UnigramSampler(10)); in TEST_F()
131 SimpleCandidateSamplerOp<UnigramSampler>);