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