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

tensorRT serialization OK

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-11 17:17:05 +02:00
parent 57c9a6ec99
commit 3b2f062dd9
7 changed files with 54 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pluginsRT/RegionRT.cpp
+9 -1
View File
@@ -70,10 +70,18 @@ public:
virtual size_t getSerializationSize() override {
return 0;
return 6*sizeof(int) + 1*sizeof(float);
}
virtual void serialize(void* buffer) override {
char *buf = reinterpret_cast<char*>(buffer);
tkDNN::writeBUF(buf, classes);
tkDNN::writeBUF(buf, coords);
tkDNN::writeBUF(buf, num);
tkDNN::writeBUF(buf, thresh);
tkDNN::writeBUF(buf, c);
tkDNN::writeBUF(buf, h);
tkDNN::writeBUF(buf, w);
}
int c, h, w;
src/pluginsRT/ReorgRT.cpp
+6 -1
View File
@@ -48,10 +48,15 @@ public:
virtual size_t getSerializationSize() override {
return 0;
return 4*sizeof(int);
}
virtual void serialize(void* buffer) override {
char *buf = reinterpret_cast<char*>(buffer);
tkDNN::writeBUF(buf, stride);
tkDNN::writeBUF(buf, c);
tkDNN::writeBUF(buf, h);
tkDNN::writeBUF(buf, w);
}
int c, h, w, stride;