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 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 cudnnDestroyFilterDescriptor(CudnnFilterDescriptorHandle filterDescriptor);
public static extern CudnnStatus cudnnCreateFilterDescriptor(out CudnnFilterDescriptorHandle filterDescriptor);