Searched refs:ThreadUnsafeUnigramSampler (Results 1 – 3 of 3) sorted by relevance
164 class ThreadUnsafeUnigramSampler : public RangeSampler {166 explicit ThreadUnsafeUnigramSampler(int64_t range);167 ~ThreadUnsafeUnigramSampler() override {} in ~ThreadUnsafeUnigramSampler()203 ThreadUnsafeUnigramSampler unsafe_sampler_ TF_GUARDED_BY(mu_);
181 ThreadUnsafeUnigramSampler::ThreadUnsafeUnigramSampler(int64_t range) in ThreadUnsafeUnigramSampler() function in tensorflow::ThreadUnsafeUnigramSampler186 int64_t ThreadUnsafeUnigramSampler::Sample(random::SimplePhilox* rnd) const { in Sample()190 float ThreadUnsafeUnigramSampler::Probability(int64_t value) const { in Probability()194 void ThreadUnsafeUnigramSampler::Update(ArraySlice<int64_t> values) { in Update()
135 SimpleCandidateSamplerOp<ThreadUnsafeUnigramSampler>);