| 215 | } |
| 216 | |
| 217 | cudnnStatus_t cudnnFindConvolutionForwardAlgorithm( |
| 218 | cudnnHandle_t handle, const cudnnTensorDescriptor_t xDesc, |
| 219 | const cudnnFilterDescriptor_t wDesc, |
| 220 | const cudnnConvolutionDescriptor_t convDesc, |
| 221 | const cudnnTensorDescriptor_t yDesc, const int requestedAlgoCount, |
| 222 | int *returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t *perfResults) { |
| 223 | return getCudnnPlugin().cudnnFindConvolutionForwardAlgorithm( |
| 224 | handle, xDesc, wDesc, convDesc, yDesc, requestedAlgoCount, |
| 225 | returnedAlgoCount, perfResults); |
| 226 | } |
| 227 | |
| 228 | cudnnStatus_t cudnnFindConvolutionBackwardFilterAlgorithm( |
| 229 | cudnnHandle_t handle, const cudnnTensorDescriptor_t xDesc, |
no outgoing calls
no test coverage detected