/TRT_PixelShuffle

Primary LanguageC++

TensorRT 8.6 Plugin ( C++ && Python ) Implementation of nn.PixelShuffle

Usage

  1. modify TensorRT path in CMakeLists.txt
  2. Build the plugin with the following commands:
mkdir build && cd build
cmake ..
make