Searched defs:maxThreadsDim (Results 1 – 5 of 5) sorted by relevance
359 int* maxThreadsDim = at::cuda::getCurrentDeviceProperties()->maxThreadsDim; in TORCH_IMPL_FUNC() local494 int* maxThreadsDim = at::cuda::getCurrentDeviceProperties()->maxThreadsDim; in TORCH_IMPL_FUNC() local
492 int* maxThreadsDim = at::cuda::getCurrentDeviceProperties()->maxThreadsDim; in adaptive_avg_pool2d_out_cuda_template() local644 int* maxThreadsDim = at::cuda::getCurrentDeviceProperties()->maxThreadsDim; in adaptive_avg_pool2d_backward_out_cuda_template() local
698 int* maxThreadsDim = at::cuda::getCurrentDeviceProperties()->maxThreadsDim; in upsample_gen2d_aa_out_cuda_template() local797 int* maxThreadsDim = at::cuda::getCurrentDeviceProperties()->maxThreadsDim; in upsample_gen2d_aa_backward_out_cuda_template() local
280 int* maxThreadsDim = at::cuda::getCurrentDeviceProperties()->maxThreadsDim; in upsample_nearest2d_out_cuda_template() local
200 int maxThreadsDim[3]; /**< Maximum size of each dimension of a block */ member