Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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()
H A Dcheck_numerics_op_gpu.cu.cc90 (d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor()) / in Run()
108 (d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor()) / in Run()
H A Drandom_op_gpu.h241 d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor(),
H A Ddepthwise_conv_op_gpu.h756 : device.getNumGpuMultiProcessors();
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A 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() *
H A DTensorDeviceGpu.h318 EIGEN_STRONG_INLINE int getNumGpuMultiProcessors() const { in getNumGpuMultiProcessors() function
H A DTensorExecutor.h604 const int max_blocks = device.getNumGpuMultiProcessors() *
H A DTensorConvolution.h885 const int numMultiProcessors = m_device.getNumGpuMultiProcessors();