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

This commit is contained in:
perseusdg
2021-11-10 20:09:57 +05:30
parent 7dd33cd118
commit 9e328c0daa
4 changed files with 24 additions and 6 deletions
+1
View File
@@ -17,6 +17,7 @@ Pooling::Pooling( Network *net, int winH, int winW, int strideH, int strideW,
this->pool_mode = pool_mode;
this->paddingH = paddingH;
this->paddingW = paddingW;
this->padding = winH -1;
checkCUDNN( cudnnCreatePoolingDescriptor(&poolingDesc) );