diff options
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, |