Searched defs:sharedMemPerBlock (Results 1 – 5 of 5) sorted by relevance
494 size_t sharedMemPerBlock = at::cuda::getCurrentDeviceProperties()->sharedMemPerBlock; in adaptive_avg_pool2d_out_cuda_template() local646 size_t sharedMemPerBlock = at::cuda::getCurrentDeviceProperties()->sharedMemPerBlock; in adaptive_avg_pool2d_backward_out_cuda_template() local
697 size_t sharedMemPerBlock = at::cuda::getCurrentDeviceProperties()->sharedMemPerBlock; in upsample_gen2d_aa_out_cuda_template() local829 size_t sharedMemPerBlock = at::cuda::getCurrentDeviceProperties()->sharedMemPerBlock; in upsample_gen2d_aa_backward_out_cuda_template() local
670 EIGEN_STRONG_INLINE size_t sharedMemPerBlock() const { in sharedMemPerBlock() function984 EIGEN_STRONG_INLINE size_t sharedMemPerBlock() const { in sharedMemPerBlock() function
327 EIGEN_STRONG_INLINE int sharedMemPerBlock() const { in sharedMemPerBlock() function
202 int sharedMemPerBlock; /**< Shared memory available per block in bytes */ member