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

dects dont works

Browse Source
This commit is contained in:
Francesco Gatti
2019-02-05 20:09:47 +00:00
parent c8dea4668d
commit 0e97452460
4 changed files with 52 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Layer.cpp
+10 -7
View File
@@ -7,14 +7,17 @@ namespace tk { namespace dnn {
Layer::Layer(Network *net) {
this->net = net;
this->input_dim = net->getOutputDim();
this->output_dim = input_dim;
checkCUDNN( cudnnCreateTensorDescriptor(&srcTensorDesc) );
checkCUDNN( cudnnCreateTensorDescriptor(&dstTensorDesc) );
if(!net->addLayer(this))
FatalError("Net reached max number of layers");
if(net != nullptr) {
this->input_dim = net->getOutputDim();
this->output_dim = input_dim;
checkCUDNN( cudnnCreateTensorDescriptor(&srcTensorDesc) );
checkCUDNN( cudnnCreateTensorDescriptor(&dstTensorDesc) );
if(!net->addLayer(this))
FatalError("Net reached max number of layers");
}
}
Layer::~Layer() {