cudnnConvolutionBackwardFilter() private method

private cudnnConvolutionBackwardFilter ( cudnnHandle handle, double &alpha, cudnnTensorDescriptor srcDesc, ManagedCuda.BasicTypes.CUdeviceptr srcData, cudnnTensorDescriptor diffDesc, ManagedCuda.BasicTypes.CUdeviceptr diffData, cudnnConvolutionDescriptor convDesc, cudnnConvolutionBwdFilterAlgo algo, ManagedCuda.BasicTypes.CUdeviceptr workSpace, ManagedCuda.BasicTypes.SizeT workSpaceSizeInBytes, double &beta, cudnnFilterDescriptor gradDesc, ManagedCuda.BasicTypes.CUdeviceptr gradData ) : cudnnStatus
handle cudnnHandle
alpha double
srcDesc cudnnTensorDescriptor
srcData ManagedCuda.BasicTypes.CUdeviceptr
diffDesc cudnnTensorDescriptor
diffData ManagedCuda.BasicTypes.CUdeviceptr
convDesc cudnnConvolutionDescriptor
algo cudnnConvolutionBwdFilterAlgo
workSpace ManagedCuda.BasicTypes.CUdeviceptr
workSpaceSizeInBytes ManagedCuda.BasicTypes.SizeT
beta double
gradDesc cudnnFilterDescriptor
gradData ManagedCuda.BasicTypes.CUdeviceptr
return cudnnStatus
CudaDNNNativeMethods Class Documentation
Esempio n. 1
0
 public void ConvolutionBackwardFilter(double alpha,
                                       TensorDescriptor srcDesc,
                                       CudaDeviceVariable <double> srcData,
                                       TensorDescriptor diffDesc,
                                       CudaDeviceVariable <double> diffData,
                                       ConvolutionDescriptor convDesc,
                                       double beta,
                                       FilterDescriptor gradDesc,
                                       CudaDeviceVariable <double> gradData
                                       )
 {
     res = CudaDNNNativeMethods.cudnnConvolutionBackwardFilter(_handle, ref alpha, srcDesc.Desc, srcData.DevicePointer, diffDesc.Desc, diffData.DevicePointer, convDesc.Desc, ref beta, gradDesc.Desc, gradData.DevicePointer);
     Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnConvolutionBackwardFilter", res));
     if (res != cudnnStatus.Success)
     {
         throw new CudaDNNException(res);
     }
 }