Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dgpu_launch_config.h127 inline GpuLaunchConfig GetGpuLaunchConfig(int work_element_count, in GetGpuLaunchConfig() argument
129 CHECK_GT(work_element_count, 0); in GetGpuLaunchConfig()
131 const int virtual_thread_count = work_element_count; in GetGpuLaunchConfig()
146 inline CudaLaunchConfig GetCudaLaunchConfig(int work_element_count, in GetCudaLaunchConfig() argument
148 return GetGpuLaunchConfig(work_element_count, d); in GetCudaLaunchConfig()
156 GpuLaunchConfig GetGpuLaunchConfig(int work_element_count, in GetGpuLaunchConfig() argument
160 CHECK_GT(work_element_count, 0); in GetGpuLaunchConfig()
178 std::min(block_count, DivUp(work_element_count, thread_per_block)); in GetGpuLaunchConfig()
180 config.virtual_thread_count = work_element_count; in GetGpuLaunchConfig()
193 int work_element_count, const Eigen::GpuDevice& d, DeviceFunc func, in GetGpuLaunchConfigFixedBlockSize() argument
[all …]
Dgpu_kernel_helper_test.cu.cc182 #define TEST_LAUNCH_PARAMETER(work_element_count) \ in TEST_F() argument
188 cfg = GetGpuLaunchConfig(work_element_count, d); \ in TEST_F()
193 EXPECT_EQ(work_element_count, \ in TEST_F()
201 cfg = GetGpuLaunchConfig(work_element_count, d, Count1D, 0, 0); \ in TEST_F()
206 EXPECT_EQ(work_element_count, \ in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dstateful_random_ops_gpu.cu.cc84 int work_element_count = (output_size + kGroupSize - 1) / kGroupSize; in operator ()() local
86 GetGpuLaunchConfig(work_element_count, d, FillKernel<Distribution>, 0, 0); in operator ()()