| 195 | } |
| 196 | |
| 197 | cudnnStatus_t cudnnGetConvolutionForwardWorkspaceSize( |
| 198 | cudnnHandle_t handle, const cudnnTensorDescriptor_t xDesc, |
| 199 | const cudnnFilterDescriptor_t wDesc, |
| 200 | const cudnnConvolutionDescriptor_t convDesc, |
| 201 | const cudnnTensorDescriptor_t yDesc, cudnnConvolutionFwdAlgo_t algo, |
| 202 | size_t *sizeInBytes) { |
| 203 | return getCudnnPlugin().cudnnGetConvolutionForwardWorkspaceSize( |
| 204 | handle, xDesc, wDesc, convDesc, yDesc, algo, sizeInBytes); |
| 205 | } |
| 206 | |
| 207 | cudnnStatus_t cudnnGetConvolutionBackwardFilterWorkspaceSize( |
| 208 | cudnnHandle_t handle, const cudnnTensorDescriptor_t xDesc, |
no outgoing calls
no test coverage detected