| 97 | } |
| 98 | |
| 99 | cudnnStatus_t cudnnSetConvolution2dDescriptor( |
| 100 | cudnnConvolutionDescriptor_t convDesc, |
| 101 | int pad_h, // zero-padding height |
| 102 | int pad_w, // zero-padding width |
| 103 | int u, // vertical filter stride |
| 104 | int v, // horizontal filter stride |
| 105 | int upscalex, // upscale the input in x-direction |
| 106 | int upscaley, // upscale the input in y-direction |
| 107 | cudnnConvolutionMode_t mode, cudnnDataType_t computeType) { |
| 108 | return |
| 109 | #if CUDNN_VERSION >= 6000 |
| 110 | getCudnnPlugin().cudnnSetConvolution2dDescriptor( |
| 111 | convDesc, pad_h, pad_w, u, v, upscalex, upscaley, mode, |
| 112 | computeType); |
| 113 | #elif CUDNN_VERSION >= 4000 |
| 114 | getCudnnPlugin().cudnnSetConvolution2dDescriptor( |
| 115 | convDesc, pad_h, pad_w, u, v, upscalex, upscaley, mode); |
| 116 | #else |
| 117 | static_assert(1 != 1, "cuDNN version not supported"); |
| 118 | #endif |
| 119 | } |
| 120 | |
| 121 | cudnnStatus_t cudnnSetFilter4dDescriptor(cudnnFilterDescriptor_t filterDesc, |
| 122 | cudnnDataType_t dataType, |
no outgoing calls
no test coverage detected