| 174 | } |
| 175 | |
| 176 | cudnnStatus_t cudnnGetConvolutionNdForwardOutputDim( |
| 177 | const cudnnConvolutionDescriptor_t convDesc, |
| 178 | const cudnnTensorDescriptor_t inputTensorDesc, |
| 179 | const cudnnFilterDescriptor_t filterDesc, int nbDims, |
| 180 | int tensorOuputDimA[]) { |
| 181 | return getCudnnPlugin().cudnnGetConvolutionNdForwardOutputDim( |
| 182 | convDesc, inputTensorDesc, filterDesc, nbDims, tensorOuputDimA); |
| 183 | } |
| 184 | |
| 185 | cudnnStatus_t cudnnGetConvolutionForwardAlgorithmMaxCount(cudnnHandle_t handle, |
| 186 | int *count) { |