Searched refs:_adaptive_avg_pool (Results 1 – 1 of 1) sorted by relevance
181 Tensor _adaptive_avg_pool(const Tensor& input, in _adaptive_avg_pool() function255 return _adaptive_avg_pool<2, scalar_t>(input, output_size, output); in q_adaptive_avg_pool2d()261 return _adaptive_avg_pool<3, scalar_t>(input, output_size, output); in q_adaptive_avg_pool3d()