Esempio n. 1
0
 public static extern CudnnStatus cudnnPoolingForward(
     CudnnHandle handle,
     CudnnPoolingDescriptorHandle poolingDescriptor,
     CudnnTensorDescriptorHandle srcDescriptor,
     [In] CUdeviceptr srcData,
     CudnnTensorDescriptorHandle destDescriptor,
     [In, Out] CUdeviceptr destData);
Esempio n. 2
0
 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);
Esempio n. 5
0
 public static extern CudnnStatus cudnnActivationForward(
     CudnnHandle handle,
     CudnnActivationMode mode,
     CudnnTensorDescriptorHandle srcDescriptor,
     [In] CUdeviceptr srcData,
     CudnnTensorDescriptorHandle destDescriptor,
     [In, Out] CUdeviceptr destData);
Esempio n. 6
0
 public static extern CudnnStatus cudnnSoftmaxForward(
     CudnnHandle handle,
     CudnnSoftmaxAlgorithm algorithm,
     CudnnSoftmaxMode mode,
     CudnnTensorDescriptorHandle srcDescriptor,
     [In] CUdeviceptr srcData,
     CudnnTensorDescriptorHandle destDescriptor,
     [In, Out] CUdeviceptr destData);
Esempio n. 7
0
 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);
Esempio n. 9
0
        private CudnnContext( CudnnHandle handle, CudaStream stream )
        {
            if (handle.Pointer == IntPtr.Zero)
                throw new ArgumentException("handle");

            Contract.EndContractBlock();

            this.handle = handle;
            this.stream = stream;
        }
Esempio n. 10
0
 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);
Esempio n. 12
0
 public static extern CudnnStatus cudnnSetTensor4d(CudnnHandle handle,
                                                   CudnnTensorDescriptorHandle tensorDescriptor,
                                                   [In, Out] CUdeviceptr tensorData,
                                                   [In] CUdeviceptr value);
Esempio n. 13
0
 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);
Esempio n. 15
0
 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);
Esempio n. 22
0
 public static extern CudnnStatus cudnnAddTensor4d(CudnnHandle handle,
                                                   CudnnAdditionMode mode,
                                                   CudnnTensorDescriptorHandle biasDescriptor,
                                                   [In] CUdeviceptr biasData,
                                                   CudnnTensorDescriptorHandle srcDestDescriptor,
                                                   [In, Out] CUdeviceptr srcDestData);
Esempio n. 23
0
 public static extern CudnnStatus cudnnTransformTensor4d(CudnnHandle handle,
                                                         CudnnTensorDescriptorHandle srcDescriptor,
                                                         [In] CUdeviceptr srcData,
                                                         CudnnTensorDescriptorHandle destDescriptor,
                                                         [In] CUdeviceptr destData);
 public static extern CudnnStatus cudnnDestroy(CudnnHandle handle);
Esempio n. 25
0
 public static extern CudnnStatus cudnnCreate(out CudnnHandle handle);