Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet

2020-02-07 09:06:29 +01:00
parent c21a0687ce 2503eba173
commit 5c501f529b
6 changed files with 34 additions and 15 deletions
@@ -229,6 +229,9 @@ public:
    dnnType *offset, *mask;
    dnnType *output_conv;

+    cublasStatus_t stat;
+    cublasHandle_t handle;
+
 protected:

    cudnnTensorDescriptor_t biasTensorDesc;