HomeSort by relevance Sort by last modified time
    Searched refs:FixedUnigramSampler (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
range_sampler.cc 232 FixedUnigramSampler::FixedUnigramSampler(Env* env, int64 range,
248 FixedUnigramSampler::FixedUnigramSampler(int64 range,
264 float FixedUnigramSampler::Probability(int64 value) const {
271 int64 FixedUnigramSampler::Sample(random::SimplePhilox* rnd) const {
275 void FixedUnigramSampler::FillReservedIds(int32 num_reserved_ids) {
281 Status FixedUnigramSampler::LoadFromFile(Env* env, const string& vocab_file,
310 void FixedUnigramSampler::LoadFromUnigrams(const std::vector<float>& unigrams,
range_sampler_test.cc 160 sampler_.reset(new FixedUnigramSampler(env, 9, fname, 0.8, 0, 1, 0));
170 sampler_.reset(new FixedUnigramSampler(env, 9, fname, 0.8, 0, 1, 0));
178 sampler_.reset(new FixedUnigramSampler(env, 9, fname, 0.8, 0, 1, 0));
185 sampler_.reset(new FixedUnigramSampler(env, 10, fname, 0.8, 1, 1, 0));
196 sampler_.reset(new FixedUnigramSampler(env, 11, fname, 0.8, 2, 1, 0));
206 sampler_.reset(new FixedUnigramSampler(9, weights, 0.8, 0, 1, 0));
214 sampler_.reset(new FixedUnigramSampler(9, weights, 0.8, 0, 1, 0));
219 sampler_.reset(new FixedUnigramSampler(9, weights, 0.8, 0, 1, 0));
224 sampler_.reset(new FixedUnigramSampler(10, weights, 0.8, 1, 1, 0));
233 sampler_.reset(new FixedUnigramSampler(11, weights, 0.8, 2, 1, 0))
    [all...]
range_sampler.h 211 class FixedUnigramSampler : public RangeSampler {
215 FixedUnigramSampler(Env* env, int64 range, const string& vocab_file,
219 FixedUnigramSampler(int64 range, const std::vector<float>& unigrams,
candidate_sampler_ops.cc 177 set_sampler(new FixedUnigramSampler(context->env(), range_max, vocab_file,
181 set_sampler(new FixedUnigramSampler(range_max, unigrams, distortion,

Completed in 819 milliseconds