Home
last modified time | relevance | path

Searched refs:SingleSampleAdapter (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drandom_op_gpu.cu.cc88 random::SingleSampleAdapter<random::PhiloxRandom>, Eigen::half> >;
91 random::SingleSampleAdapter<random::PhiloxRandom>, float> >;
94 random::SingleSampleAdapter<random::PhiloxRandom>, double> >;
Drandom_op.cc57 using random::SingleSampleAdapter;
131 SingleSampleAdapter<PhiloxRandom> single_samples(&gen); in Run()
142 SingleSampleAdapter<PhiloxRandom> single_samples(&gen); in Run()
464 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE>>; \
487 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE>>); \
541 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE>>);
634 using random::SingleSampleAdapter; in operator ()()
655 SingleSampleAdapter<PhiloxRandom> single_samples(&gen); in operator ()()
732 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE>>);
Drandom_op_gpu.h166 using random::SingleSampleAdapter;
184 SingleSampleAdapter<PhiloxRandom> single_samples(&gen);
Drandom_shuffle_op.cc80 random::SingleSampleAdapter<random::PhiloxRandom> single(&local_gen); in Compute()
Dfractional_pool_common.cc93 random::SingleSampleAdapter<random::PhiloxRandom> single(&local_gen); in GeneratePoolingSequenceRandom()
Dstateless_random_ops.cc179 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE> >);
Drandom_shuffle_queue_op.cc84 random::SingleSampleAdapter<random::PhiloxRandom> generator_ GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drandom_dataset_op.cc132 generator_ = random::SingleSampleAdapter<random::PhiloxRandom>( in RestoreInternal()
139 random::SingleSampleAdapter<random::PhiloxRandom>::ResultType Random() in Random()
147 random::SingleSampleAdapter<random::PhiloxRandom> generator_
/external/tensorflow/tensorflow/core/lib/random/
Drandom_distributions_test.cc55 SingleSampleAdapter<PhiloxRandom> single_samples(&gen); in FillRandomsWithSingles()
249 TruncatedNormalDistribution<SingleSampleAdapter<PhiloxRandom>, T> >( in RandomParametersMomentsTest()
334 SingleSampleAdapter<T> gen(&parent_gen); in SingleSampleAdapterSkipTest()
338 SingleSampleAdapter<T> gen_to_skip(&parent_gen_to_skip); in SingleSampleAdapterSkipTest()
Drandom_distributions.cc22 void SingleSampleAdapter<PhiloxRandom>::SkipFromGenerator(uint64 num_skips) { in SkipFromGenerator()
Dsimple_philox.h70 SingleSampleAdapter<PhiloxRandom> single_;
Drandom_distributions.h242 class SingleSampleAdapter {
252 explicit SingleSampleAdapter(Generator* gen) in SingleSampleAdapter() function
/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.cc191 generator_ = random::SingleSampleAdapter<random::PhiloxRandom>( in ResetRngs()
321 random::SingleSampleAdapter<random::PhiloxRandom>::ResultType Random() in Random()
334 random::SingleSampleAdapter<random::PhiloxRandom> generator_
443 generator_ = random::SingleSampleAdapter<random::PhiloxRandom>( in Reset()
453 random::SingleSampleAdapter<random::PhiloxRandom> generator_