| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_op_gpu.cu.cc | 43 FillPhiloxRandomKernelLaunch(random::PhiloxRandom base_gen, in FillPhiloxRandomKernelLaunch()
54 OpKernelContext*, const GPUDevice& d, random::PhiloxRandom gen, in operator ()()
71 GPUDevice, random::UniformDistribution<random::PhiloxRandom, Eigen::half> >;
73 GPUDevice, random::UniformDistribution<random::PhiloxRandom, float> >;
75 GPUDevice, random::UniformDistribution<random::PhiloxRandom, double> >;
77 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int32> >;
79 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int64> >;
81 GPUDevice, random::NormalDistribution<random::PhiloxRandom, Eigen::half> >;
83 GPUDevice, random::NormalDistribution<random::PhiloxRandom, float> >;
85 GPUDevice, random::NormalDistribution<random::PhiloxRandom, double> >;
[all …]
|
| D | random_op.cc | 56 using random::PhiloxRandom;
65 void operator()(OpKernelContext*, const Device&, random::PhiloxRandom gen, in operator ()()
80 static void Run(random::PhiloxRandom gen, T* data, int64 size, in Run()
111 static void Run(random::PhiloxRandom base_gen, T* data, int64 size, in Run()
117 PhiloxRandom::kResultElementCount; in Run()
129 PhiloxRandom gen = base_gen; in Run()
131 SingleSampleAdapter<PhiloxRandom> single_samples(&gen); in Run()
140 PhiloxRandom gen = base_gen; in Run()
142 SingleSampleAdapter<PhiloxRandom> single_samples(&gen); in Run()
155 OpKernelContext* context, const CPUDevice&, random::PhiloxRandom gen, in operator ()()
[all …]
|
| D | stateless_random_ops.cc | 30 Status GenerateKey(Tensor seed, random::PhiloxRandom::Key* out_key, in GenerateKey()
31 random::PhiloxRandom::ResultType* out_counter) { in GenerateKey()
56 const auto mix = random::PhiloxRandom(*out_counter, *out_key)(); in GenerateKey()
87 random::PhiloxRandom::Key key; in Compute()
88 random::PhiloxRandom::ResultType counter; in Compute()
92 Fill(context, random::PhiloxRandom(counter, key), output); in Compute()
96 virtual void Fill(OpKernelContext* context, random::PhiloxRandom random,
105 void Fill(OpKernelContext* context, random::PhiloxRandom random, in Fill()
121 void Fill(OpKernelContext* context, random::PhiloxRandom random, in Fill()
141 typedef random::UniformDistribution<random::PhiloxRandom, IntType> in Fill()
[all …]
|
| D | stateful_random_ops_cpu_gpu.h | 43 PHILOX_DEVICE_INLINE PhiloxRandom
45 PhiloxRandom::ResultType counter; in GetPhiloxRandomFromMem()
46 PhiloxRandom::Key key; in GetPhiloxRandomFromMem()
50 return PhiloxRandom(counter, key); in GetPhiloxRandomFromMem()
53 PHILOX_DEVICE_INLINE void WritePhiloxRandomToMem(PhiloxRandom const& philox, in WritePhiloxRandomToMem()
55 PhiloxRandom::ResultType const& counter = philox.counter(); in WritePhiloxRandomToMem()
56 PhiloxRandom::Key const& key = philox.key(); in WritePhiloxRandomToMem()
62 PHILOX_DEVICE_INLINE void UpdateMemWithPhiloxRandom(PhiloxRandom const& philox, in UpdateMemWithPhiloxRandom()
|
| D | stateful_random_ops_gpu.cu.cc | 26 using random::PhiloxRandom;
37 __shared__ char philox_raw[sizeof(PhiloxRandom)]; in FillKernel()
38 auto philox = reinterpret_cast<PhiloxRandom*>(philox_raw); in FillKernel()
88 GPUDevice, random::NormalDistribution<random::PhiloxRandom, Eigen::half> >;
90 GPUDevice, random::NormalDistribution<random::PhiloxRandom, float> >;
92 GPUDevice, random::NormalDistribution<random::PhiloxRandom, double> >;
|
| D | random_op_gpu.h | 34 PHILOX_DEVICE_FUNC void Run(random::PhiloxRandom gen, T* data, int64 size,
41 PHILOX_DEVICE_FUNC void Run(const random::PhiloxRandom& base_gen, T* data,
132 random::PhiloxRandom gen, T* data, int64 size, Distribution dist) {
163 const random::PhiloxRandom& base_gen, T* data, int64 size,
165 using random::PhiloxRandom;
172 PhiloxRandom::kResultElementCount;
182 PhiloxRandom gen = base_gen;
184 SingleSampleAdapter<PhiloxRandom> single_samples(&gen);
|
| D | random_poisson_op.cc | 74 const random::PhiloxRandom& rng, U* samples_flat);
81 const random::PhiloxRandom& rng, U* samples_flat) { in operator ()()
103 typedef random::UniformDistribution<random::PhiloxRandom, CT> Uniform; in operator ()()
137 random::PhiloxRandom gen = rng; in operator ()()
199 random::PhiloxRandom gen = rng; in operator ()()
266 6 * random::PhiloxRandom::kElementCost; in operator ()()
305 random::PhiloxRandom rng = generator_.ReserveRandomOutputs( in Compute()
|
| D | stateful_random_ops.h | 35 using random::PhiloxRandom;
38 (PhiloxRandom::ResultType::kElementCount +
39 PhiloxRandom::Key::kElementCount) /
|
| D | random_op.h | 40 random::PhiloxRandom gen,
51 random::PhiloxRandom gen,
63 random::PhiloxRandom gen,
|
| D | parameterized_truncated_normal_op.cc | 46 using random::PhiloxRandom;
58 const random::PhiloxRandom& gen, in operator ()()
77 random::PhiloxRandom gen_copy = gen; in operator ()()
83 typedef random::UniformDistribution<random::PhiloxRandom, T> Uniform; in operator ()()
85 typedef random::NormalDistribution<random::PhiloxRandom, T> Normal; in operator ()()
292 random::PhiloxRandom::kElementCost + in operator ()()
293 random::UniformDistribution<random::PhiloxRandom, T>::kElementCost; in operator ()()
424 random::PhiloxRandom rng = in Compute()
|
| D | stateless_random_ops.h | 29 Status GenerateKey(Tensor seed_t, random::PhiloxRandom::Key* out_key,
30 random::PhiloxRandom::ResultType* out_counter);
|
| D | parameterized_truncated_normal_op_gpu.cu.cc | 54 TruncatedNormalKernel(random::PhiloxRandom gen, T* data, int64 num_batches, in TruncatedNormalKernel()
64 typedef random::UniformDistribution<random::PhiloxRandom, T> Uniform; in TruncatedNormalKernel()
65 typedef random::NormalDistribution<random::PhiloxRandom, T> Normal; in TruncatedNormalKernel()
241 const random::PhiloxRandom& gen, in operator ()()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | guarded_philox_random.h | 52 void Init(random::PhiloxRandom::ResultType counter,
53 random::PhiloxRandom::Key key);
58 random::PhiloxRandom ReserveSamples128(int64 samples);
61 random::PhiloxRandom ReserveSamples32(int64 samples) { in ReserveSamples32()
66 random::PhiloxRandom ReserveRandomOutputs(int64 output_count, in ReserveRandomOutputs()
74 random::PhiloxRandom generator_ GUARDED_BY(mu_);
|
| D | guarded_philox_random.cc | 42 generator_ = random::PhiloxRandom(seed, seed2); in Init()
46 void GuardedPhiloxRandom::Init(random::PhiloxRandom::ResultType counter, in Init()
47 random::PhiloxRandom::Key key) { in Init()
50 generator_ = random::PhiloxRandom(counter, key); in Init()
54 random::PhiloxRandom GuardedPhiloxRandom::ReserveSamples128(int64 samples) { in ReserveSamples128()
|
| /external/tensorflow/tensorflow/core/lib/random/ |
| D | philox_random_test.cc | 38 static constexpr int kResultElementCount = PhiloxRandom::kResultElementCount;
39 typedef PhiloxRandom::ResultType ResultType;
40 typedef PhiloxRandom::ResultElementType ResultElementType;
43 ResultType operator()(PhiloxRandom* gen) { return (*gen)(); } in operator ()()
55 PhiloxRandom gen(test_seed); in TEST()
62 PhiloxRandom gen(test_seed); in TEST()
|
| D | simple_philox_test.cc | 30 PhiloxRandom philox(7, 7); in TEST()
59 PhiloxRandom philox1(1, 1), philox2(17, 17); in TEST()
66 PhiloxRandom philox1(1, 1), philox2(2, 1); in TEST()
76 PhiloxRandom philox1(0, 1), philox2(1, 1); in TEST()
96 PhiloxRandom philox(17, 17); in TEST()
115 PhiloxRandom philox(17, 17); in TEST()
|
| D | random_distributions_test.cc | 47 void FillRandomsWithSingles(PhiloxRandom gen, in FillRandomsWithSingles()
55 SingleSampleAdapter<PhiloxRandom> single_samples(&gen); in FillRandomsWithSingles()
156 PhiloxRandom gen(seed); in UniformMomentsTest()
157 FillRandoms<UniformDistribution<PhiloxRandom, T> >(gen, &v1[0], v1.size()); in UniformMomentsTest()
187 PhiloxRandom gen(seed); in NormalMomentsTest()
188 FillRandoms<NormalDistribution<PhiloxRandom, T> >(gen, &v1[0], v1.size()); in NormalMomentsTest()
247 PhiloxRandom gen(seed); in RandomParametersMomentsTest()
249 TruncatedNormalDistribution<SingleSampleAdapter<PhiloxRandom>, T> >( in RandomParametersMomentsTest()
364 SingleSampleAdapterSkipTest<PhiloxRandom>(); in TEST()
|
| D | philox_random.h | 103 class PhiloxRandom {
116 PhiloxRandom() {} in PhiloxRandom() function
119 explicit PhiloxRandom(uint64 seed) { in PhiloxRandom() function
125 explicit PhiloxRandom(uint64 seed_lo, uint64 seed_hi) { in PhiloxRandom() function
133 PhiloxRandom(ResultType counter, Key key) : counter_(counter), key_(key) {} in PhiloxRandom() function
|
| D | simple_philox.h | 33 explicit SimplePhilox(PhiloxRandom* gen) : single_(gen) {} in SimplePhilox()
70 SingleSampleAdapter<PhiloxRandom> single_;
|
| D | weighted_picker_test.cc | 37 PhiloxRandom philox(testing::RandomSeed(), 17); in TEST()
93 PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST()
108 PhiloxRandom philox(testing::RandomSeed() + 10, 17); in TEST()
257 PhiloxRandom philox(301, 17); in BM_Pick()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | input_data.h | 61 single_rand_ = std::unique_ptr<random::PhiloxRandom>( in TensorDataSet()
62 new random::PhiloxRandom(random::New64())); in TensorDataSet()
64 single_rand_ = std::unique_ptr<random::PhiloxRandom>( in TensorDataSet()
65 new random::PhiloxRandom(split_sampling_random_seed_)); in TensorDataSet()
122 std::unique_ptr<random::PhiloxRandom> single_rand_;
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | random_dataset_op.cc | 131 random::PhiloxRandom(dataset()->seed_, dataset()->seed2_); in RestoreInternal()
132 generator_ = random::SingleSampleAdapter<random::PhiloxRandom>( in RestoreInternal()
139 random::SingleSampleAdapter<random::PhiloxRandom>::ResultType Random() in Random()
146 random::PhiloxRandom parent_generator_ GUARDED_BY(mu_);
147 random::SingleSampleAdapter<random::PhiloxRandom> generator_
|
| /external/tensorflow/tensorflow/core/lib/io/ |
| D | table_test.cc | 434 random::PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST_F()
463 random::PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST_F()
473 random::PhiloxRandom philox(testing::RandomSeed() + 2, 17); in TEST_F()
483 random::PhiloxRandom philox(testing::RandomSeed() + 3, 17); in TEST_F()
495 random::PhiloxRandom philox(testing::RandomSeed() + 3, 17); in TEST_F()
508 random::PhiloxRandom philox(testing::RandomSeed() + 4, 17); in TEST_F()
518 random::PhiloxRandom philox(testing::RandomSeed() + 5, 17); in TEST_F()
590 random::PhiloxRandom philox(301, 17); in TEST()
613 random::PhiloxRandom philox(301, 17); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 190 parent_generator_ = random::PhiloxRandom(seed_, seed2_); in ResetRngs()
191 generator_ = random::SingleSampleAdapter<random::PhiloxRandom>( in ResetRngs()
321 random::SingleSampleAdapter<random::PhiloxRandom>::ResultType Random() in Random()
333 random::PhiloxRandom parent_generator_ GUARDED_BY(mu_);
334 random::SingleSampleAdapter<random::PhiloxRandom> generator_
442 parent_generator_ = random::PhiloxRandom(seed_, seed2_); in Reset()
443 generator_ = random::SingleSampleAdapter<random::PhiloxRandom>( in Reset()
452 random::PhiloxRandom parent_generator_ GUARDED_BY(mu_);
453 random::SingleSampleAdapter<random::PhiloxRandom> generator_
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
| D | stochastic_hard_routing_function_op.cc | 100 single_rand_ = std::unique_ptr<random::PhiloxRandom>( in StochasticHardRoutingFunction()
101 new random::PhiloxRandom(random_seed_)); in StochasticHardRoutingFunction()
186 std::unique_ptr<random::PhiloxRandom> single_rand_;
|