Deconv tensorrt

This commit is contained in:
fbagni
2019-10-30 09:40:29 +01:00
parent f9afee2f3b
commit 33f36ab204
3 changed files with 19 additions and 10 deletions
+1 -2
View File
@@ -161,6 +161,7 @@ public:
virtual dnnType* infer(dataDim_t &dim, dnnType* srcData);
int kernelH, kernelW, strideH, strideW, paddingH, paddingW;
bool deConv;
protected:
cudnnFilterDescriptor_t filterDesc;
@@ -173,8 +174,6 @@ protected:
void inferCUDNN(dnnType* srcData, bool back = false);
void* workSpace;
size_t ws_sizeInBytes;
bool deConv;
};