Searched refs:adaptive_avg_pooling2d_fn (Results 1 – 1 of 1) sorted by relevance
11 using adaptive_avg_pooling2d_fn = void(*)(Tensor& output, const Tensor& input, IntArrayRef output_s… variable13 DECLARE_DISPATCH(adaptive_avg_pooling2d_fn, adaptive_avg_pool2d_kernel);