cudnnConvolutionBackwardData() private method

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