Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Drange_sampler_test.cc160 sampler_.reset(new FixedUnigramSampler(env, 9, fname, 0.8, 0, 1, 0)); in TEST_F()
170 sampler_.reset(new FixedUnigramSampler(env, 9, fname, 0.8, 0, 1, 0)); in TEST_F()
178 sampler_.reset(new FixedUnigramSampler(env, 9, fname, 0.8, 0, 1, 0)); in TEST_F()
185 sampler_.reset(new FixedUnigramSampler(env, 10, fname, 0.8, 1, 1, 0)); in TEST_F()
196 sampler_.reset(new FixedUnigramSampler(env, 11, fname, 0.8, 2, 1, 0)); in TEST_F()
206 sampler_.reset(new FixedUnigramSampler(9, weights, 0.8, 0, 1, 0)); in TEST_F()
214 sampler_.reset(new FixedUnigramSampler(9, weights, 0.8, 0, 1, 0)); in TEST_F()
219 sampler_.reset(new FixedUnigramSampler(9, weights, 0.8, 0, 1, 0)); in TEST_F()
224 sampler_.reset(new FixedUnigramSampler(10, weights, 0.8, 1, 1, 0)); in TEST_F()
233 sampler_.reset(new FixedUnigramSampler(11, weights, 0.8, 2, 1, 0)); in TEST_F()
Drange_sampler.cc232 FixedUnigramSampler::FixedUnigramSampler(Env* env, int64 range, in FixedUnigramSampler() function in tensorflow::FixedUnigramSampler
248 FixedUnigramSampler::FixedUnigramSampler(int64 range, in FixedUnigramSampler() function in tensorflow::FixedUnigramSampler
264 float FixedUnigramSampler::Probability(int64 value) const { in Probability()
271 int64 FixedUnigramSampler::Sample(random::SimplePhilox* rnd) const { in Sample()
275 void FixedUnigramSampler::FillReservedIds(int32 num_reserved_ids) { in FillReservedIds()
281 Status FixedUnigramSampler::LoadFromFile(Env* env, const string& vocab_file, in LoadFromFile()
310 void FixedUnigramSampler::LoadFromUnigrams(const std::vector<float>& unigrams, in LoadFromUnigrams()
Drange_sampler.h211 class FixedUnigramSampler : public RangeSampler {
215 FixedUnigramSampler(Env* env, int64 range, const string& vocab_file,
219 FixedUnigramSampler(int64 range, const std::vector<float>& unigrams,
Dcandidate_sampler_ops.cc177 set_sampler(new FixedUnigramSampler(context->env(), range_max, vocab_file, in FixedUnigramCandidateSamplerOp()
181 set_sampler(new FixedUnigramSampler(range_max, unigrams, distortion, in FixedUnigramCandidateSamplerOp()