Add PP-YOLOE support

nvdsinfer_custom_impl_Yolo/layers/shortcut_layer.h

@@ -6,17 +6,21 @@
 #ifndef __SHORTCUT_LAYER_H__
 #define __SHORTCUT_LAYER_H__
 
+#include <map>
+
 #include "NvInfer.h"
 
 #include "activation_layer.h"
 
-nvinfer1::ILayer* shortcutLayer(
+nvinfer1::ITensor* shortcutLayer(
     int layerIdx,
+    std::string mode,
     std::string activation,
     std::string inputVol,
     std::string shortcutVol,
+    std::map<std::string, std::string>& block,
     nvinfer1::ITensor* input,
-    nvinfer1::ITensor* shortcutTensor,
+    nvinfer1::ITensor* shortcut,
     nvinfer1::INetworkDefinition* network);
 
 #endif