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

1 command network inference

Browse Source
This commit is contained in:
Francesco Gatti
2017-07-02 20:47:14 +02:00
parent cfa43a8580
commit ea59d7ca53
4 changed files with 43 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Layer.cpp
+3
View File
@@ -12,6 +12,9 @@ Layer::Layer(Network *net, dataDim_t in_dim) {
checkCUDNN( cudnnCreateTensorDescriptor(&srcTensorDesc) );
checkCUDNN( cudnnCreateTensorDescriptor(&dstTensorDesc) );
if(!net->addLayer(this))
FatalError("Net reached max number of layers");
}
Layer::~Layer() {