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

yolo TensorRT almost DONE

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-03 15:52:08 +02:00
parent 2ef76209a1
commit 858b3501fa
6 changed files with 96 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/Layer.h
-1
View File
@@ -251,7 +251,6 @@ public:
virtual value_type* infer(dataDim_t &dim, value_type* srcData);
protected:
int stride;
};
include/NetworkRT.h
+2
View File
@@ -38,6 +38,8 @@ public:
nvinfer1::ITensor* convert_layer(nvinfer1::ITensor *input, Dense *l);
nvinfer1::ITensor* convert_layer(nvinfer1::ITensor *input, Pooling *l);
nvinfer1::ITensor* convert_layer(nvinfer1::ITensor *input, Softmax *l);
nvinfer1::ITensor* convert_layer(nvinfer1::ITensor *input, Route *l);
nvinfer1::ITensor* convert_layer(nvinfer1::ITensor *input, Reorg *l);
};