Searched refs:UnigramSampler (Results 1 – 4 of 4) sorted by relevance
200 UnigramSampler::UnigramSampler(int64 range) in UnigramSampler() function in tensorflow::UnigramSampler205 int64 UnigramSampler::Sample(random::SimplePhilox* rnd) const { in Sample()210 float UnigramSampler::Probability(int64 value) const { in Probability()216 void UnigramSampler::SampleBatchGetExpectedCountAvoid( in SampleBatchGetExpectedCountAvoid()227 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>);