Ejemplo n.º 1
0
 public static extern CudnnStatus cudnnConvolutionBackwardBias(
                                     CudnnHandle handle,
                                     CudnnTensorDescriptorHandle srcDescriptor,
                                     [In] CUdeviceptr srcData,
                                     CudnnTensorDescriptorHandle destDescriptor,
                                     [In, Out] CUdeviceptr destData,
                                     CudnnAccumulateResult accumulate);
Ejemplo n.º 2
0
 public static extern CudnnStatus cudnnActivationForward(
                                     CudnnHandle handle,
                                     CudnnActivationMode mode,
                                     CudnnTensorDescriptorHandle srcDescriptor,
                                     [In] CUdeviceptr srcData,
                                     CudnnTensorDescriptorHandle destDescriptor,
                                     [In, Out] CUdeviceptr destData);
Ejemplo n.º 3
0
 public static extern CudnnStatus cudnnActivationForward(
     CudnnHandle handle,
     CudnnActivationMode mode,
     CudnnTensorDescriptorHandle srcDescriptor,
     [In] CUdeviceptr srcData,
     CudnnTensorDescriptorHandle destDescriptor,
     [In, Out] CUdeviceptr destData);
Ejemplo n.º 4
0
 public static extern CudnnStatus cudnnSetTensor4dDescriptor(CudnnTensorDescriptorHandle tensorDesc,
                                                             CudnnTensorFormat format,
                                                             CudnnType dataType,    // image data type
                                                             int n,                 // number of inputs (batch size)
                                                             int c,                 // number of input feature maps
                                                             int h,                 // height of input section
                                                             int w);                // width of input section
Ejemplo n.º 5
0
 public static extern CudnnStatus cudnnPoolingForward(
     CudnnHandle handle,
     CudnnPoolingDescriptorHandle poolingDescriptor,
     CudnnTensorDescriptorHandle srcDescriptor,
     [In] CUdeviceptr srcData,
     CudnnTensorDescriptorHandle destDescriptor,
     [In, Out] CUdeviceptr destData);
Ejemplo n.º 6
0
 public static extern CudnnStatus cudnnConvolutionBackwardBias(
     CudnnHandle handle,
     CudnnTensorDescriptorHandle srcDescriptor,
     [In] CUdeviceptr srcData,
     CudnnTensorDescriptorHandle destDescriptor,
     [In, Out] CUdeviceptr destData,
     CudnnAccumulateResult accumulate);
Ejemplo n.º 7
0
 public static extern CudnnStatus cudnnSoftmaxForward(
     CudnnHandle handle,
     CudnnSoftmaxAlgorithm algorithm,
     CudnnSoftmaxMode mode,
     CudnnTensorDescriptorHandle srcDescriptor,
     [In] CUdeviceptr srcData,
     CudnnTensorDescriptorHandle destDescriptor,
     [In, Out] CUdeviceptr destData);
        internal CudnnTensorDescriptor(CudnnTensorDescriptorHandle handle)
        {
            if (handle.Pointer == IntPtr.Zero)
                throw new ArgumentException("The handle pointer is null.", "handle");

            Contract.EndContractBlock();

            this.Handle = handle;
        }
Ejemplo n.º 9
0
 public static extern CudnnStatus cudnnGetTensor4dDescriptor(CudnnTensorDescriptorHandle tensorDesc,
                                                             out CudnnType dataType,      // image data type
                                                             out int n,                   // number of inputs (batch size)
                                                             out int c,                   // number of input feature maps
                                                             out int h,                   // height of input section
                                                             out int w,
                                                             out int nStride,
                                                             out int cStride,
                                                             out int hStride,
                                                             out int wStride);
Ejemplo n.º 10
0
 public static extern CudnnStatus cudnnSetTensor4dDescriptorEx(CudnnTensorDescriptorHandle tensorDesc,
                                                               CudnnType dataType,    // image data type
                                                               int n,                 // number of inputs (batch size)
                                                               int c,                 // number of input feature maps
                                                               int h,                 // height of input section
                                                               int w,
                                                               int nStride,
                                                               int cStride,
                                                               int hStride,
                                                               int wStride);
Ejemplo n.º 11
0
 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);
Ejemplo n.º 12
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);
Ejemplo n.º 13
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);
Ejemplo n.º 14
0
 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);
Ejemplo n.º 15
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);
Ejemplo n.º 16
0
 public static extern CudnnStatus cudnnDestroyTensor4dDescriptor(CudnnTensorDescriptorHandle tensorDesc);
Ejemplo n.º 17
0
 public static extern CudnnStatus cudnnCreateTensor4dDescriptor(out CudnnTensorDescriptorHandle tensorDesc);
Ejemplo n.º 18
0
 public static extern CudnnStatus cudnnTransformTensor4d(CudnnHandle handle,
                                                         CudnnTensorDescriptorHandle srcDescriptor,
                                                         [In] CUdeviceptr srcData,
                                                         CudnnTensorDescriptorHandle destDescriptor,
                                                         [In] CUdeviceptr destData);
Ejemplo n.º 19
0
 public static extern CudnnStatus cudnnAddTensor4d(CudnnHandle handle,
                                                   CudnnAdditionMode mode,
                                                   CudnnTensorDescriptorHandle biasDescriptor,
                                                   [In] CUdeviceptr biasData,
                                                   CudnnTensorDescriptorHandle srcDestDescriptor,
                                                   [In, Out] CUdeviceptr srcDestData);
Ejemplo n.º 20
0
 public static extern CudnnStatus cudnnSoftmaxForward(
                                     CudnnHandle handle,
                                     CudnnSoftmaxAlgorithm algorithm,
                                     CudnnSoftmaxMode mode,
                                     CudnnTensorDescriptorHandle srcDescriptor,
                                     [In] CUdeviceptr srcData,
                                     CudnnTensorDescriptorHandle destDescriptor,
                                     [In, Out] CUdeviceptr destData);
Ejemplo n.º 21
0
 public static extern CudnnStatus cudnnTransformTensor4d(CudnnHandle handle,
                                                         CudnnTensorDescriptorHandle srcDescriptor,
                                                         [In] CUdeviceptr srcData,
                                                         CudnnTensorDescriptorHandle destDescriptor,
                                                         [In] CUdeviceptr destData);
Ejemplo n.º 22
0
 public static extern CudnnStatus cudnnSetTensor4d(CudnnHandle handle,                                                 
                                                   CudnnTensorDescriptorHandle tensorDescriptor,
                                                   [In, Out] CUdeviceptr tensorData,
                                                   [In] CUdeviceptr value);
Ejemplo n.º 23
0
 public static extern CudnnStatus cudnnSetTensor4dDescriptorEx(CudnnTensorDescriptorHandle tensorDesc,
                                                               CudnnType dataType,    // image data type
                                                               int n,                 // number of inputs (batch size)
                                                               int c,                 // number of input feature maps
                                                               int h,                 // height of input section
                                                               int w,
                                                               int nStride,
                                                               int cStride,
                                                               int hStride,
                                                               int wStride);
Ejemplo n.º 24
0
 public static extern CudnnStatus cudnnPoolingForward(
                                     CudnnHandle handle,
                                     CudnnPoolingDescriptorHandle poolingDescriptor,
                                     CudnnTensorDescriptorHandle srcDescriptor,
                                     [In] CUdeviceptr srcData,
                                     CudnnTensorDescriptorHandle destDescriptor,
                                     [In, Out] CUdeviceptr destData);
Ejemplo n.º 25
0
 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);
Ejemplo n.º 26
0
 public static extern CudnnStatus cudnnGetTensor4dDescriptor(CudnnTensorDescriptorHandle tensorDesc,
                                                               out CudnnType dataType,    // image data type
                                                               out int n,                 // number of inputs (batch size)
                                                               out int c,                 // number of input feature maps
                                                               out int h,                 // height of input section
                                                               out int w,
                                                               out int nStride,
                                                               out int cStride,
                                                               out int hStride,
                                                               out int wStride);
Ejemplo n.º 27
0
 public static extern CudnnStatus cudnnDestroyTensor4dDescriptor(CudnnTensorDescriptorHandle tensorDesc);
Ejemplo n.º 28
0
 public static extern CudnnStatus cudnnCreateTensor4dDescriptor(out CudnnTensorDescriptorHandle tensorDesc);
Ejemplo n.º 29
0
 public static extern CudnnStatus cudnnAddTensor4d(CudnnHandle handle,
                                                   CudnnAdditionMode mode,
                                                   CudnnTensorDescriptorHandle biasDescriptor,
                                                   [In] CUdeviceptr biasData,
                                                   CudnnTensorDescriptorHandle srcDestDescriptor,
                                                   [In, Out] CUdeviceptr srcDestData);
Ejemplo n.º 30
0
 public static extern CudnnStatus cudnnSetTensor4d(CudnnHandle handle,
                                                   CudnnTensorDescriptorHandle tensorDescriptor,
                                                   [In, Out] CUdeviceptr tensorData,
                                                   [In] CUdeviceptr value);