Searched refs:adaptive_average_pool (Results 1 – 1 of 1) sorted by relevance
53 __global__ void adaptive_average_pool(const scalar_t *input, scalar_t *output, in adaptive_average_pool() function585 adaptive_average_pool <<<blocks, threads, 0, at::cuda::getCurrentCUDAStream()>>> ( in adaptive_avg_pool2d_out_cuda_template()