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

  /external/tensorflow/tensorflow/python/ops/
candidate_sampling_ops.py 212 unigrams=(),
250 one of `vocab_file` and `unigrams` needs to be passed to this operation.
268 unigrams: A list of unigram counts or probabilities, one per ID in
269 sequential order. Exactly one of `vocab_file` and `unigrams` should be
290 unigrams=unigrams, seed=seed1, seed2=seed2, name=name)
  /external/tensorflow/tensorflow/core/kernels/
candidate_sampler_ops.cc 158 std::vector<float> unigrams; local
159 OP_REQUIRES_OK(context, context->GetAttr("unigrams", &unigrams));
161 context, !vocab_file.empty() || !unigrams.empty(),
162 errors::InvalidArgument("Must provide either vocab_file or unigrams."));
163 OP_REQUIRES(context, vocab_file.empty() || unigrams.empty(),
165 "Must only provide one of vocab_file and unigrams."));
181 set_sampler(new FixedUnigramSampler(range_max, unigrams, distortion,
range_sampler.h 219 FixedUnigramSampler(int64 range, const std::vector<float>& unigrams,
247 void LoadFromUnigrams(const std::vector<float>& unigrams, float distortion);
range_sampler.cc 249 const std::vector<float>& unigrams,
258 LoadFromUnigrams(unigrams, distortion);
310 void FixedUnigramSampler::LoadFromUnigrams(const std::vector<float>& unigrams,
313 for (float w : unigrams) {

Completed in 764 milliseconds