Searched refs:SingleSampleAdapter (Results 1 – 21 of 21) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_op_gpu.cu.cc | 71 random::SingleSampleAdapter<random::PhiloxRandom>, Eigen::half> >;
74 random::SingleSampleAdapter<random::PhiloxRandom>, float> >;
77 random::SingleSampleAdapter<random::PhiloxRandom>, double> >;
|
| D | stateful_random_ops_gpu.cu.cc | 129 random::SingleSampleAdapter<random::PhiloxRandom>,
133 random::SingleSampleAdapter<random::PhiloxRandom>,
137 random::SingleSampleAdapter<random::PhiloxRandom>,
|
| D | random_op_cpu.h | 55 using random::SingleSampleAdapter;
135 SingleSampleAdapter<PhiloxRandom> single_samples(&gen);
146 SingleSampleAdapter<PhiloxRandom> single_samples(&gen);
|
| D | random_op.cc | 337 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE>>; \
360 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE>>); \
423 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE>>);
|
| D | random_op_gpu.h | 180 using random::SingleSampleAdapter;
201 SingleSampleAdapter<PhiloxRandom> single_samples(&gen);
|
| D | random_shuffle_op.cc | 80 random::SingleSampleAdapter<random::PhiloxRandom> single(&local_gen); in Compute()
|
| D | fractional_pool_common.cc | 93 random::SingleSampleAdapter<random::PhiloxRandom> single(&local_gen); in GeneratePoolingSequenceRandom()
|
| D | stateless_random_ops.cc | 230 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE> >)
|
| D | stateless_random_ops_v2.cc | 291 random::SingleSampleAdapter<random::PhiloxRandom>, TYPE> >)
|
| D | random_shuffle_queue_op.cc | 84 random::SingleSampleAdapter<random::PhiloxRandom> generator_
|
| D | stateful_random_ops.cc | 382 random::SingleSampleAdapter<PhiloxRandom>, TYPE> >);
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | random_dataset_op.cc | 126 random::SingleSampleAdapter<random::PhiloxRandom>(&parent_generator_); in RestoreInternal()
132 random::SingleSampleAdapter<random::PhiloxRandom>::ResultType Random() in Random()
141 random::SingleSampleAdapter<random::PhiloxRandom> generator_
|
| D | sampling_dataset_op.cc | 149 random::SingleSampleAdapter<random::PhiloxRandom>(&parent_generator_); in ResetRngs()
212 random::SingleSampleAdapter<random::PhiloxRandom> generator_
|
| D | random_dataset_op_test.cc | 37 random::SingleSampleAdapter<random::PhiloxRandom>(&parent_generator); in GenerateExpectedData()
|
| /external/tensorflow/tensorflow/core/lib/random/ |
| D | random_distributions_test.cc | 55 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()
|
| D | random_distributions.cc | 22 void SingleSampleAdapter<PhiloxRandom>::SkipFromGenerator(uint64 num_skips) { in SkipFromGenerator()
|
| D | simple_philox.h | 70 SingleSampleAdapter<PhiloxRandom> single_;
|
| D | random_distributions.h | 306 class SingleSampleAdapter {
316 explicit SingleSampleAdapter(Generator* gen) in SingleSampleAdapter() function
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | random_seed_ops.cc | 61 random::SingleSampleAdapter<random::PhiloxRandom>(&parent_generator_); in Reset()
|
| D | random_seed_ops.h | 125 random::SingleSampleAdapter<random::PhiloxRandom> generator_
|
| D | shuffle_dataset_op.cc | 276 random::SingleSampleAdapter<random::PhiloxRandom>(&parent_generator_); in ResetRngs()
399 random::SingleSampleAdapter<random::PhiloxRandom>::ResultType Random() in Random()
421 random::SingleSampleAdapter<random::PhiloxRandom> generator_
|