| 279 | } |
| 280 | |
| 281 | cudnnStatus_t cudnnConvolutionForward( |
| 282 | cudnnHandle_t handle, const void *alpha, |
| 283 | const cudnnTensorDescriptor_t xDesc, const void *x, |
| 284 | const cudnnFilterDescriptor_t wDesc, const void *w, |
| 285 | const cudnnConvolutionDescriptor_t convDesc, cudnnConvolutionFwdAlgo_t algo, |
| 286 | void *workSpace, size_t workSpaceSizeInBytes, const void *beta, |
| 287 | const cudnnTensorDescriptor_t yDesc, void *y) { |
| 288 | return getCudnnPlugin().cudnnConvolutionForward( |
| 289 | handle, alpha, xDesc, x, wDesc, w, convDesc, algo, workSpace, |
| 290 | workSpaceSizeInBytes, beta, yDesc, y); |
| 291 | } |
| 292 | |
| 293 | cudnnStatus_t cudnnConvolutionBackwardFilter( |
| 294 | cudnnHandle_t handle, const void *alpha, |