Home
last modified time | relevance | path

Searched defs:maxGridSize (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveAveragePooling.cu493 int* maxGridSize = at::cuda::getCurrentDeviceProperties()->maxGridSize; in adaptive_avg_pool2d_out_cuda_template() local
645 int* maxGridSize = at::cuda::getCurrentDeviceProperties()->maxGridSize; in adaptive_avg_pool2d_backward_out_cuda_template() local
H A DUpSampleBilinear2d.cu700 int* maxGridSize = at::cuda::getCurrentDeviceProperties()->maxGridSize; in upsample_gen2d_aa_out_cuda_template() local
802 int* maxGridSize = at::cuda::getCurrentDeviceProperties()->maxGridSize; in upsample_gen2d_aa_backward_out_cuda_template() local
H A DUpSampleNearest2d.cu281 int* maxGridSize = at::cuda::getCurrentDeviceProperties()->maxGridSize; in upsample_nearest2d_out_cuda_template() local
/aosp_15_r20/hardware/google/aemu/third-party/cuda/include/host-common/
Ddynlink_cuda_cuda.h201 int maxGridSize[3]; /**< Maximum size of each dimension of a grid */ member