mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

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

Browse Source
This commit is contained in:
perseusdg
2021-10-28 23:35:37 +05:30
parent 8c36dd0431
commit c5e66c6bf6
37 changed files with 1042 additions and 700 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Flatten.cpp
+2 -2
View File
@@ -17,8 +17,8 @@ Flatten::Flatten(Network *net) : Layer(net) {
this->h = 1;
this->w = 1;
this->rows = input_dim.w;
this->cols = input_dim.h * input_dim.c;
this->rows = input_dim.c;
this->cols = input_dim.h * input_dim.w;
this->c = input_dim.w * input_dim.h * input_dim.c;
}