Home
last modified time | relevance | path

Searched refs:maxGpuThreadsPerBlock (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddebug_ops_gpu.cu.cc158 const int32 block_size = d.maxGpuThreadsPerBlock(); in Run()
182 const int32 block_size = d.maxGpuThreadsPerBlock(); in Run()
206 const int32 block_size = d.maxGpuThreadsPerBlock(); in Run()
230 const int32 block_size = d.maxGpuThreadsPerBlock(); in Run()
Dcheck_numerics_op_gpu.cu.cc88 const int32 block_size = d.maxGpuThreadsPerBlock(); in Run()
106 const int32 block_size = d.maxGpuThreadsPerBlock(); in Run()
Drandom_op_gpu.h238 const int32 block_size = d.maxGpuThreadsPerBlock();
/external/tensorflow/tensorflow/core/util/
Dgpu_launch_config.h135 const int thread_per_block = std::min(1024, d.maxGpuThreadsPerBlock()); in GetGpuLaunchConfig()
317 thread_per_block = std::min(1024, d.maxGpuThreadsPerBlock()); in GetGpu3DLaunchConfig()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceGpu.h321 EIGEN_STRONG_INLINE int maxGpuThreadsPerBlock() const { in maxGpuThreadsPerBlock() function
DTensorExecutor.h603 const int block_size = device.maxGpuThreadsPerBlock();
DTensorConvolution.h883 const int maxThreadsPerBlock = m_device.maxGpuThreadsPerBlock();