Migrate tkDNN max pooling plugin creation to pluginRegistry from the default method

2021-11-10 20:09:57 +05:30
parent 7dd33cd118
commit 9e328c0daa
4 changed files with 24 additions and 6 deletions
@@ -500,6 +500,7 @@ public:
    int winH, winW;
    int strideH, strideW;
    int paddingH, paddingW;
+    int padding;
    bool size;
    tkdnnPoolingMode_t pool_mode;