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
@@ -8,7 +8,10 @@ namespace tk { namespace dnn {
 Reshape::Reshape(Network *net, dataDim_t new_dim) : Layer(net) {

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