cudnnSetConvolution2dDescriptor() 개인적인 메소드

private cudnnSetConvolution2dDescriptor ( cudnnConvolutionDescriptor convDesc, int pad_h, int pad_w, int u, int v, int upscalex, int upscaley, cudnnConvolutionMode mode, cudnnDataType dataType ) : cudnnStatus
convDesc cudnnConvolutionDescriptor
pad_h int
pad_w int
u int
v int
upscalex int
upscaley int
mode cudnnConvolutionMode
dataType cudnnDataType
리턴 cudnnStatus
예제 #1
0
 public void SetConvolution2dDescriptor(int pad_h,            // zero-padding height
                                        int pad_w,            // zero-padding width
                                        int u,                // vertical filter stride
                                        int v,                // horizontal filter stride
                                        int upscalex,         // upscale the input in x-direction
                                        int upscaley,         // upscale the input in y-direction
                                        cudnnConvolutionMode mode
                                        )
 {
     res = CudaDNNNativeMethods.cudnnSetConvolution2dDescriptor(_desc, pad_h, pad_w, u, v, upscalex, upscaley, mode);
     Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnSetConvolution2dDescriptor", res));
     if (res != cudnnStatus.Success)
     {
         throw new CudaDNNException(res);
     }
 }
예제 #2
0
 /// <summary>
 /// This function initializes a previously created convolution descriptor object into a 2D
 /// correlation. This function assumes that the tensor and filter descriptors corresponds
 /// to the formard convolution path and checks if their settings are valid. That same
 /// convolution descriptor can be reused in the backward path provided it corresponds to
 /// the same layer.
 /// </summary>
 /// <param name="pad_h">zero-padding height: number of rows of zeros implicitly concatenated
 /// onto the top and onto the bottom of input images.</param>
 /// <param name="pad_w">zero-padding width: number of columns of zeros implicitly concatenated
 /// onto the left and onto the right of input images.</param>
 /// <param name="u">Vertical filter stride.</param>
 /// <param name="v">Horizontal filter stride.</param>
 /// <param name="dilation_h">Filter height dilation.</param>
 /// <param name="dilation_w">Filter width dilation.</param>
 /// <param name="mode">Selects between CUDNN_CONVOLUTION and CUDNN_CROSS_CORRELATION.</param>
 /// <param name="dataType">Selects the datatype in which the computation will be done.</param>
 public void SetConvolution2dDescriptor(int pad_h,      // zero-padding height
                                        int pad_w,      // zero-padding width
                                        int u,          // vertical filter stride
                                        int v,          // horizontal filter stride
                                        int dilation_h, // filter dilation in the vertical dimension
                                        int dilation_w, // filter dilation in the horizontal dimension
                                        cudnnConvolutionMode mode,
                                        cudnnDataType dataType
                                        )
 {
     res = CudaDNNNativeMethods.cudnnSetConvolution2dDescriptor(_desc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, dataType);
     Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnSetConvolution2dDescriptor", res));
     if (res != cudnnStatus.Success)
     {
         throw new CudaDNNException(res);
     }
 }