├── CMakeLists.txt ├── README.md ├── compile.sh ├── flattenConcatCustom.cpp ├── flattenConcatCustom.h └── load_trt_engine.cpp /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YirongMao/TensorRT-Custom-Plugin/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YirongMao/TensorRT-Custom-Plugin/HEAD/README.md -------------------------------------------------------------------------------- /compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YirongMao/TensorRT-Custom-Plugin/HEAD/compile.sh -------------------------------------------------------------------------------- /flattenConcatCustom.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YirongMao/TensorRT-Custom-Plugin/HEAD/flattenConcatCustom.cpp -------------------------------------------------------------------------------- /flattenConcatCustom.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YirongMao/TensorRT-Custom-Plugin/HEAD/flattenConcatCustom.h -------------------------------------------------------------------------------- /load_trt_engine.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YirongMao/TensorRT-Custom-Plugin/HEAD/load_trt_engine.cpp --------------------------------------------------------------------------------