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 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 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 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 cudnnSoftmaxForward(
CudnnHandle handle,
CudnnSoftmaxAlgorithm algorithm,
CudnnSoftmaxMode mode,
CudnnTensorDescriptorHandle srcDescriptor,
[In] CUdeviceptr srcData,
CudnnTensorDescriptorHandle destDescriptor,
[In, Out] CUdeviceptr destData);
public static extern CudnnStatus cudnnSetTensor4d(CudnnHandle handle,
CudnnTensorDescriptorHandle tensorDescriptor,
[In, Out] CUdeviceptr tensorData,
[In] CUdeviceptr value);
public static extern CudnnStatus cudnnSetStream(CudnnHandle handle, CUstream streamId);
public static extern CudnnStatus cudnnPoolingForward(
CudnnHandle handle,
CudnnPoolingDescriptorHandle poolingDescriptor,
CudnnTensorDescriptorHandle srcDescriptor,
[In] CUdeviceptr srcData,
CudnnTensorDescriptorHandle destDescriptor,
[In, Out] CUdeviceptr destData);
public static extern CudnnStatus cudnnCreate(out 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 cudnnTransformTensor4d(CudnnHandle handle,
CudnnTensorDescriptorHandle srcDescriptor,
[In] CUdeviceptr srcData,
CudnnTensorDescriptorHandle destDescriptor,
[In] CUdeviceptr destData);
public static extern CudnnStatus cudnnDestroy(CudnnHandle handle);
public static extern CudnnStatus cudnnCreate(out CudnnHandle handle);
