public static extern CudnnStatus cudnnPoolingForward( CudnnHandle handle, CudnnPoolingDescriptorHandle poolingDescriptor, CudnnTensorDescriptorHandle srcDescriptor, [In] CUdeviceptr srcData, CudnnTensorDescriptorHandle destDescriptor, [In, Out] CUdeviceptr destData);
public static extern CudnnStatus cudnnConvolutionBackwardBias( CudnnHandle handle, CudnnTensorDescriptorHandle srcDescriptor, [In] CUdeviceptr srcData, CudnnTensorDescriptorHandle destDescriptor, [In, Out] CUdeviceptr destData, CudnnAccumulateResult accumulate);
public static extern CudnnStatus cudnnActivationForward( CudnnHandle handle, CudnnActivationMode mode, CudnnTensorDescriptorHandle srcDescriptor, [In] CUdeviceptr srcData, CudnnTensorDescriptorHandle destDescriptor, [In, Out] CUdeviceptr destData);
public static extern CudnnStatus cudnnSoftmaxForward( CudnnHandle handle, CudnnSoftmaxAlgorithm algorithm, CudnnSoftmaxMode mode, CudnnTensorDescriptorHandle srcDescriptor, [In] CUdeviceptr srcData, CudnnTensorDescriptorHandle destDescriptor, [In, Out] CUdeviceptr destData);
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);
private CudnnContext( CudnnHandle handle, CudaStream stream ) { if (handle.Pointer == IntPtr.Zero) throw new ArgumentException("handle"); Contract.EndContractBlock(); this.handle = handle; this.stream = stream; }
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 cudnnSetTensor4d(CudnnHandle handle, CudnnTensorDescriptorHandle tensorDescriptor, [In, Out] CUdeviceptr tensorData, [In] CUdeviceptr value);
public static extern CudnnStatus cudnnDestroy(CudnnHandle handle);
public static extern CudnnStatus cudnnAddTensor4d(CudnnHandle handle, CudnnAdditionMode mode, CudnnTensorDescriptorHandle biasDescriptor, [In] CUdeviceptr biasData, CudnnTensorDescriptorHandle srcDestDescriptor, [In, Out] CUdeviceptr srcDestData);
public static extern CudnnStatus cudnnGetStream(CudnnHandle handle, out CUstream streamId);
public static extern CudnnStatus cudnnTransformTensor4d(CudnnHandle handle, CudnnTensorDescriptorHandle srcDescriptor, [In] CUdeviceptr srcData, CudnnTensorDescriptorHandle destDescriptor, [In] CUdeviceptr destData);
public static extern CudnnStatus cudnnSetStream(CudnnHandle handle, CUstream streamId);
public static extern CudnnStatus cudnnCreate(out CudnnHandle handle);