Home
last modified time | relevance | path

Searched refs:getNumGpuMultiProcessors (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dgpu_launch_config.h133 d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor(), in GetGpuLaunchConfig()
138 d.getNumGpuMultiProcessors()); in GetGpuLaunchConfig()
208 block_count = std::min(block_count * d.getNumGpuMultiProcessors(), in GetGpuLaunchConfigFixedBlockSize()
240 d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor(); in GetGpu2DLaunchConfig()
316 d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor(); in GetGpu3DLaunchConfig()
319 d.getNumGpuMultiProcessors()); in GetGpu3DLaunchConfig()
/external/tensorflow/tensorflow/core/kernels/
Ddebug_ops_gpu.cu.cc160 (d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor()) / in Run()
184 (d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor()) / in Run()
208 (d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor()) / in Run()
232 (d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor()) / in Run()
Dcheck_numerics_op_gpu.cu.cc90 (d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor()) / in Run()
108 (d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor()) / in Run()
Drandom_op_gpu.h241 d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor(),
Ddepthwise_conv_op_gpu.h754 : device.getNumGpuMultiProcessors();
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorReductionGpu.h769 const int max_blocks = device.getNumGpuMultiProcessors() *
777 const int max_blocks = device.getNumGpuMultiProcessors() *
815 const int max_blocks = device.getNumGpuMultiProcessors() *
937 const int max_blocks = device.getNumGpuMultiProcessors() *
945 const int max_blocks = device.getNumGpuMultiProcessors() *
DTensorDeviceGpu.h318 EIGEN_STRONG_INLINE int getNumGpuMultiProcessors() const { in getNumGpuMultiProcessors() function
DTensorExecutor.h604 const int max_blocks = device.getNumGpuMultiProcessors() *
DTensorConvolution.h885 const int numMultiProcessors = m_device.getNumGpuMultiProcessors();