Searched refs:DistributionSampler (Results 1 – 7 of 7) sorted by relevance
46 class DistributionSampler {48 explicit DistributionSampler(const gtl::ArraySlice<float>& weights);50 ~DistributionSampler() {} in ~DistributionSampler()88 TF_DISALLOW_COPY_AND_ASSIGN(DistributionSampler);
24 DistributionSampler::DistributionSampler( in DistributionSampler() function in tensorflow::random::DistributionSampler
38 DistributionSampler sampler(weights); in TestWeights()93 DistributionSampler picker(weights); in BM_DistributionSampler()
248 dist_sampler_.reset(new random::DistributionSampler(weights_)); in FixedUnigramSampler()264 dist_sampler_.reset(new random::DistributionSampler(weights_)); in FixedUnigramSampler()
258 sampler_ = new random::DistributionSampler(vocab_weights); in NegTrainOp()350 random::DistributionSampler* sampler_ = nullptr;
229 std::unique_ptr<random::DistributionSampler> dist_sampler_;
267 random::DistributionSampler sampler(probabilities_); in SampleAdaptiveProbabilities()