pytorch
7a3cf67f - Implement channels last upsample2d/3d forward pass kernel. (#34597)

Commit
5 years ago
Implement channels last upsample2d/3d forward pass kernel. (#34597) Summary: Thi PR implement channel last upsampling nearest for 2D/3D. This is supposed to be faster, plus, avoids converting formats going in and out of operator. Will post benchmarking numbers. Pull Request resolved: https://github.com/pytorch/pytorch/pull/34597 Test Plan: python test/test_nn.py TestNN.test_upsamplingNearest3d_channels_last Differential Revision: D20390583 Pulled By: kimishpatel fbshipit-source-id: e0162fb97604a261887f38fc957d3f787c80954e
Author
Parents
Loading