Implementations of the custom layers in C++ and Native Pytorch for CPU support.#212

Open

zsameem wants to merge 2 commits intoNVIDIA:masterfrom zsameem:cpu_support

+721-47

Commits on May 16, 2020

Implementations of the custom layers i.e the Correlation layer, Resample2D layer and the ChannelNorm layer in native Pytorch and C++ to support inference on CPU.
zsameem
committed
Revert to use CUDA layers by default
zsameem
committed