| 143 | } |
| 144 | |
| 145 | cudnnStatus_t cudnnSetTensor4dDescriptor(cudnnTensorDescriptor_t tensorDesc, |
| 146 | cudnnTensorFormat_t format, |
| 147 | cudnnDataType_t dataType, int n, int c, |
| 148 | int h, int w) { |
| 149 | return getCudnnPlugin().cudnnSetTensor4dDescriptor(tensorDesc, format, |
| 150 | dataType, n, c, h, w); |
| 151 | } |
| 152 | |
| 153 | cudnnStatus_t cudnnGetConvolutionBackwardDataWorkspaceSize( |
| 154 | cudnnHandle_t handle, const cudnnFilterDescriptor_t wDesc, |