public static extern CudnnStatus cudnnGetFilterDescriptor(CudnnFilterDescriptorHandle filterDescriptor,
out CudnnType dataType, // image data type
out int k, // number of output feature maps
out int c, // number of input feature maps
out int h, // height of each input filter
out int w // width of each input filter
);
internal CudnnFilterDescriptor(CudnnFilterDescriptorHandle handle)
{
if (handle.Pointer == IntPtr.Zero)
throw new ArgumentException("The handle pointer is null.", "handle");
Contract.EndContractBlock();
this.Handle = handle;
}
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 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 cudnnSetFilterDescriptor(CudnnFilterDescriptorHandle filterDescriptor,
CudnnType dataType, // image data type
int k, // number of output feature maps
int c, // number of input feature maps
int h, // height of each input filter
int w // width of each input filter
);
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 cudnnDestroyFilterDescriptor(CudnnFilterDescriptorHandle filterDescriptor);
public static extern CudnnStatus cudnnCreateFilterDescriptor(out CudnnFilterDescriptorHandle filterDescriptor);
public static extern CudnnStatus cudnnDestroyFilterDescriptor(CudnnFilterDescriptorHandle filterDescriptor);
public static extern CudnnStatus cudnnCreateFilterDescriptor(out CudnnFilterDescriptorHandle filterDescriptor);
