Searched refs:kResultElementCount (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/lib/random/
H A Drandom_distributions.h44 // invocation. It needs to define kResultElementCount for the
58 static const int kResultElementCount = Generator::kResultElementCount; member in class:tensorflow::random::UniformDistribution
64 typedef Array<Eigen::half, kResultElementCount> ResultType;
71 for (int i = 0; i < kResultElementCount; ++i) {
82 static const int kResultElementCount = Generator::kResultElementCount; member in class:tensorflow::random::UniformDistribution
88 typedef Array<float, kResultElementCount> ResultType;
95 for (int i = 0; i < kResultElementCount; ++i) {
106 static const int kResultElementCount member in class:tensorflow::random::UniformDistribution
130 static const int kResultElementCount = Generator::kResultElementCount; member in class:tensorflow::random::UniformDistribution
164 static const int kResultElementCount = Generator::kResultElementCount / 2; member in class:tensorflow::random::UniformDistribution
201 static const int kResultElementCount = 1; member in class:tensorflow::random::SingleSampleAdapter
285 static const int kResultElementCount = Generator::kResultElementCount; member in class:tensorflow::random::NormalDistribution
312 static const int kResultElementCount = Generator::kResultElementCount; member in class:tensorflow::random::NormalDistribution
336 static const int kResultElementCount = Generator::kResultElementCount / 2; member in class:tensorflow::random::NormalDistribution
379 static const int kResultElementCount = member in class:tensorflow::random::TruncatedNormalDistribution
422 static const int kResultElementCount = member in class:tensorflow::random::TruncatedNormalDistribution
465 static const int kResultElementCount = member in class:tensorflow::random::TruncatedNormalDistribution
[all...]
H A Dphilox_random_test.cc38 static constexpr int kResultElementCount = PhiloxRandom::kResultElementCount; member in class:tensorflow::random::__anon26702::TrivialPhiloxDistribution
H A Dphilox_random_test_utils.h36 const int granularity = Distribution::kResultElementCount;
H A Drandom_distributions_test.cc46 int granularity = Distribution::kResultElementCount;
290 static const int kResultElementCount = 1; member in class:tensorflow::random::__anon26704::MockGenerator
H A Dphilox_random.h107 static const int kResultElementCount = 4; member in class:tensorflow::random::PhiloxRandom
/external/tensorflow/tensorflow/core/kernels/
H A Dparameterized_truncated_normal_op_gpu.cu.cc63 const int kDistSize = Uniform::kResultElementCount;
130 remaining_samples -= gen.kResultElementCount;
139 remaining_samples -= gen.kResultElementCount;
168 remaining_samples -= gen.kResultElementCount;
H A Drandom_op_gpu.cu.cc132 const int kGroupSize = Distribution::kResultElementCount;
170 const int kGroupSize = Distribution::kResultElementCount;
173 PhiloxRandom::kResultElementCount;
H A Drandom_op.cc82 const int kGroupSize = Distribution::kResultElementCount;
113 const int kGroupSize = Distribution::kResultElementCount;
117 PhiloxRandom::kResultElementCount;
158 const int kGroupSize = Distribution::kResultElementCount;
165 random::PhiloxRandom::kResultElementCount *
306 uniform_remaining = Uniform::kResultElementCount; \
398 norm_remaining = Normal::kResultElementCount;
577 const size_t kGroupSize = Distribution::kResultElementCount;
628 const size_t kGroupSize = Distribution::kResultElementCount;
631 PhiloxRandom::kResultElementCount;
[all...]
H A Drandom_poisson_op.cc48 uniform_remaining = Uniform::kResultElementCount; \

Completed in 147 milliseconds