diff options
author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2017-11-02 01:28:08 +0300 |
---|---|---|
committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2017-11-02 01:28:08 +0300 |
commit | cd6fca847dd9aa0b5236306beaaa8f778e2d8726 (patch) | |
tree | 495608b6ad01298da0b6e5478c3c6505e0cc2841 /src/graph/node_operators_unary.h | |
parent | 289c25cb2ca0df9077480f8e843e99ea2ef8f018 (diff) |
commented out offending code make it compile, needs to be fixed when USE_CUDNN=on
Diffstat (limited to 'src/graph/node_operators_unary.h')
-rw-r--r-- | src/graph/node_operators_unary.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index 844f8905..fcf1c290 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -1047,14 +1047,14 @@ public: padWidth, strideHeight, strideWidth)); - + /* @TODO: does not compile CUDNN_CALL(cudnnGetPooling2dForwardOutputDim(poolingDesc_, xDesc_, shape_.begin(), shape_.begin() + 1, shape_.begin() + 2, shape_.begin() + 3)); - +*/ CUDNN_CALL(cudnnCreateTensorDescriptor(&yDesc_)); CUDNN_CALL(cudnnSetTensor4dDescriptor(yDesc_, CUDNN_TENSOR_NCHW, |