TRT8 works with almost every nerual network now!!!!(including demo3d)

2021-10-28 23:35:37 +05:30
parent 8c36dd0431
commit c5e66c6bf6
37 changed files with 1042 additions and 700 deletions
@@ -9,6 +9,9 @@ Shortcut::Shortcut(Network *net, Layer *backLayer, bool mul) : Layer(net) {

    this->backLayer = backLayer;
    this->mul = mul;
+    this->c = input_dim.c;
+    this->h = input_dim.h;
+    this->w = input_dim.w;
    checkCuda( cudaMalloc(&dstData, output_dim.tot()*sizeof(dnnType)) );

    if( ( backLayer->output_dim.c != input_dim.c && mul ) ||