Home
last modified time | relevance | path

Searched refs:UnigramSampler (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drange_sampler.cc200 UnigramSampler::UnigramSampler(int64 range) in UnigramSampler() function in tensorflow::UnigramSampler
205 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()
Drange_sampler.h181 class UnigramSampler : public RangeSampler {
183 explicit UnigramSampler(int64 range);
184 ~UnigramSampler() override {} in ~UnigramSampler()
Drange_sampler_test.cc118 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()
Dcandidate_sampler_ops.cc131 SimpleCandidateSamplerOp<UnigramSampler>);