pytorch
48424a6c - Avoid dynamic dispatch inside the omp loop in AdaptiveAvgPool2d (#20366)

Commit
5 years ago
Avoid dynamic dispatch inside the omp loop in AdaptiveAvgPool2d (#20366) Summary: This PR changes CPU implementation of `AdaptiveAveragePool2D` by - move dispatch to outside the OpenMP loop - support fp16 Pull Request resolved: https://github.com/pytorch/pytorch/pull/20366 Differential Revision: D15456069 Pulled By: ezyang fbshipit-source-id: 00fa2916f8b136af9f5c8b5db0eca4619f9f5bac
Author
Parents
Loading