Home
last modified time | relevance | path

Searched refs:maxGpuThreadsPerMultiProcessor (Results 1 – 8 of 8) sorted by relevance

/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(),
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dgpu_launch_config.h133 d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor(), in GetGpuLaunchConfig()
240 d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor(); in GetGpu2DLaunchConfig()
316 d.getNumGpuMultiProcessors() * d.maxGpuThreadsPerMultiProcessor(); in GetGpu3DLaunchConfig()
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorReductionGpu.h770 device.maxGpuThreadsPerMultiProcessor() / block_size;
778 device.maxGpuThreadsPerMultiProcessor() / 1024;
816 device.maxGpuThreadsPerMultiProcessor() / block_size;
938 device.maxGpuThreadsPerMultiProcessor() / block_size;
946 device.maxGpuThreadsPerMultiProcessor() / 1024;
H A DTensorDeviceGpu.h324 EIGEN_STRONG_INLINE int maxGpuThreadsPerMultiProcessor() const { in maxGpuThreadsPerMultiProcessor() function
H A DTensorExecutor.h605 device.maxGpuThreadsPerMultiProcessor() / block_size;
H A DTensorConvolution.h884 … const int maxBlocksPerProcessor = m_device.maxGpuThreadsPerMultiProcessor() / maxThreadsPerBlock;