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
Exemplo n.º 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);
     }
 }