Home
last modified time | relevance | path

Searched refs:kGroupSize (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drandom_op_gpu.h133 const int kGroupSize = Distribution::kResultElementCount;
137 int32 offset = thread_id * kGroupSize;
140 const SampleCopier<T, kGroupSize> copier;
141 while (offset + kGroupSize <= size) {
145 offset += total_thread_count * kGroupSize;
150 for (int i = 0; i < kGroupSize; ++i) {
169 const int kGroupSize = Distribution::kResultElementCount;
170 const int kGeneratorSkipPerOutputGroup = kGroupSize *
177 int64 offset = group_index * kGroupSize;
188 for (int i = 0; i < kGroupSize; ++i) {
[all …]
Drandom_op.cc82 const int kGroupSize = Distribution::kResultElementCount; in Run() local
85 int64 offset = start_group * kGroupSize; in Run()
88 int64 limit_group_full = std::min(limit_group, size / kGroupSize); in Run()
91 std::copy(&samples[0], &samples[0] + kGroupSize, data + offset); in Run()
92 offset += kGroupSize; in Run()
97 int64 remaining_size = size - limit_group_full * kGroupSize; in Run()
113 const int kGroupSize = Distribution::kResultElementCount; in Run() local
116 kGroupSize * kReservedSamplesPerOutput / in Run()
119 int64 offset = start_group * kGroupSize; in Run()
122 int64 limit_group_full = std::min(limit_group, size / kGroupSize); in Run()
[all …]
Dstateful_random_ops_gpu.cu.cc70 const int kGroupSize = Distribution::kResultElementCount; in operator ()() local
71 int work_element_count = (output_size + kGroupSize - 1) / kGroupSize; in operator ()()