| /external/tensorflow/tensorflow/core/lib/random/ |
| D | random_distributions.h | 73 static constexpr int kResultElementCount = Generator::kResultElementCount;
79 typedef Array<Eigen::half, kResultElementCount> ResultType;
86 for (int i = 0; i < kResultElementCount; ++i) { in operator()
97 static constexpr int kResultElementCount = Generator::kResultElementCount;
103 typedef Array<bfloat16, kResultElementCount> ResultType;
110 for (int i = 0; i < kResultElementCount; ++i) { in operator()
121 static constexpr int kResultElementCount = Generator::kResultElementCount;
127 typedef Array<float, kResultElementCount> ResultType;
134 for (int i = 0; i < kResultElementCount; ++i) { in operator()
145 static constexpr int kResultElementCount = Generator::kResultElementCount / 2;
[all …]
|
| D | philox_random_test.cc | 38 static constexpr int kResultElementCount = PhiloxRandom::kResultElementCount; member in tensorflow::random::__anon997242510111::TrivialPhiloxDistribution
|
| D | philox_random_test_utils.h | 36 const int granularity = Distribution::kResultElementCount; in FillRandoms()
|
| D | random_distributions_test.cc | 50 int granularity = Distribution::kResultElementCount; in FillRandomsWithSingles()
310 static constexpr int kResultElementCount = 1; member in tensorflow::random::__anonf269a2300111::MockGenerator
|
| D | philox_random.h | 108 static constexpr int kResultElementCount = 4; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_op_cpu.h | 86 const int kGroupSize = Distribution::kResultElementCount;
117 const int kGroupSize = Distribution::kResultElementCount;
121 PhiloxRandom::kResultElementCount;
163 const int kGroupSize = Distribution::kResultElementCount;
170 random::PhiloxRandom::kResultElementCount *
|
| D | parameterized_truncated_normal_op_gpu.cu.cc | 69 const int kDistSize = Uniform::kResultElementCount; in TruncatedNormalKernel()
140 remaining_samples -= gen.kResultElementCount; in TruncatedNormalKernel()
170 remaining_samples -= gen.kResultElementCount; in TruncatedNormalKernel()
179 remaining_samples -= gen.kResultElementCount; in TruncatedNormalKernel()
206 remaining_samples -= gen.kResultElementCount; in TruncatedNormalKernel()
|
| D | random_op_gpu.h | 144 const int kGroupSize = Distribution::kResultElementCount;
183 const int kGroupSize = Distribution::kResultElementCount;
186 PhiloxRandom::kResultElementCount;
|
| D | stateless_random_gamma_op.h | 64 remaining_numbers_ = Distribution::kResultElementCount; in operator()
|
| D | parameterized_truncated_normal_op.cc | 351 Uniform::kResultElementCount - 1) / in operator ()()
352 Uniform::kResultElementCount); in operator ()()
356 Eigen::array<T, Uniform::kResultElementCount> z; in operator ()()
357 Eigen::array<T, Uniform::kResultElementCount> g; in operator ()()
|
| D | stateless_random_gamma_op_gpu.cu.cc | 77 double u = uniform(&gen)[Uniform::kResultElementCount - 1]; in FillKernel()
|
| D | random_op.cc | 182 uniform_remaining = Uniform::kResultElementCount; \ in Compute()
275 norm_remaining = Normal::kResultElementCount; in Compute()
|
| D | stateful_random_ops_gpu.cu.cc | 83 const int kGroupSize = Distribution::kResultElementCount; in operator ()()
|
| D | stateless_random_gamma_op.cc | 106 double u = uniform(&gen)[Uniform::kResultElementCount - 1]; in Fill()
|
| D | random_poisson_op.cc | 49 uniform_remaining = Uniform::kResultElementCount; \
|
| D | random_binomial_op.cc | 49 uniform_remaining = Uniform::kResultElementCount; \
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_random_uniform.cc | 50 size_t num_samples = Distribution::kResultElementCount; in ComputeRandomUniformArray()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 182 size_t num_samples = Distribution::kResultElementCount; in matchAndRewrite()
|