public static extern CudnnStatus cudnnGetOutputTensor4dDim(
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnConvolutionPath path,
out int n,
out int c,
out int h,
out int w);
public static extern CudnnStatus cudnnConvolutionBackwardFilter(
CudnnHandle handle,
CudnnTensorDescriptorHandle srcDescriptor,
[In] CUdeviceptr srcData,
CudnnTensorDescriptorHandle diffDescriptor,
[In] CUdeviceptr diffData,
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnFilterDescriptorHandle gradientDescriptor,
[In, Out] CUdeviceptr gradientData,
CudnnAccumulateResult accumulate);
public static extern CudnnStatus cudnnConvolutionBackwardData(
CudnnHandle handle,
CudnnFilterDescriptorHandle filterDescriptor,
[In] CUdeviceptr filterData,
CudnnTensorDescriptorHandle diffDescriptor,
[In] CUdeviceptr diffData,
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnTensorDescriptorHandle gradientDescriptor,
[In, Out] CUdeviceptr gradientData,
CudnnAccumulateResult accumulate);
public static extern CudnnStatus cudnnConvolutionForward(
CudnnHandle handle,
CudnnTensorDescriptorHandle srcDescriptor,
[In] CUdeviceptr srcData,
CudnnFilterDescriptorHandle filterDescriptor,
[In] CUdeviceptr filterData,
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnTensorDescriptorHandle destDescriptor,
[In, Out] CUdeviceptr destData,
CudnnAccumulateResult accumulate);
public static extern CudnnStatus cudnnSetConvolutionDescriptor(
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnTensorDescriptorHandle inputTensorDescriptor,
CudnnFilterDescriptorHandle filterDescriptor,
int paddingHeight, // zero-padding height
int paddingWidth, // zero-padding width
int verticalStride, // vertical filter stride
int horizontalStride, // horizontal filter stride
int upscaleVertical, // upscale the input in x-direction
int upscaleHorizontal, // upscale the input in y-direction
CudnnConvolutionMode mode);
public static extern CudnnStatus cudnnCreateConvolutionDescriptor(out CudnnConvolutionDescriptorHandle convolutionDescriptor);
public static extern CudnnStatus cudnnConvolutionForward(
CudnnHandle handle,
CudnnTensorDescriptorHandle srcDescriptor,
[In] CUdeviceptr srcData,
CudnnFilterDescriptorHandle filterDescriptor,
[In] CUdeviceptr filterData,
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnTensorDescriptorHandle destDescriptor,
[In, Out] CUdeviceptr destData,
CudnnAccumulateResult accumulate);
public static extern CudnnStatus cudnnSetConvolutionDescriptorEx(
CudnnConvolutionDescriptorHandle convolutionDescriptor,
int n,
int c,
int h,
int w,
int k,
int r,
int s,
int paddingHeight, // zero-padding height
int paddingWidth, // zero-padding width
int verticalStride, // vertical filter stride
int horizontalStride, // horizontal filter stride
int upscaleVertical, // upscale the input in x-direction
int upscaleHorizontal, // upscale the input in y-direction
CudnnConvolutionMode mode);
public static extern CudnnStatus cudnnGetOutputTensor4dDim(
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnConvolutionPath path,
out int n,
out int c,
out int h,
out int w);
public static extern CudnnStatus cudnnDestroyConvolutionDescriptor(CudnnConvolutionDescriptorHandle convolutionDescriptor);
public static extern CudnnStatus cudnnDestroyConvolutionDescriptor(CudnnConvolutionDescriptorHandle convolutionDescriptor);
public static extern CudnnStatus cudnnCreateConvolutionDescriptor(out CudnnConvolutionDescriptorHandle convolutionDescriptor);
internal CudnnConvolutionDescriptor(CudnnConvolutionDescriptorHandle handle)
{
this.Handle = handle;
}
