cudnnActivationBackward() private method

private cudnnActivationBackward ( cudnnHandle handle, cudnnActivationDescriptor activationDesc, double &alpha, cudnnTensorDescriptor srcDesc, ManagedCuda.BasicTypes.CUdeviceptr srcData, cudnnTensorDescriptor srcDiffDesc, ManagedCuda.BasicTypes.CUdeviceptr srcDiffData, cudnnTensorDescriptor destDesc, ManagedCuda.BasicTypes.CUdeviceptr destData, double &beta, cudnnTensorDescriptor destDiffDesc, ManagedCuda.BasicTypes.CUdeviceptr destDiffData ) : cudnnStatus
handle cudnnHandle
activationDesc cudnnActivationDescriptor
alpha double
srcDesc cudnnTensorDescriptor
srcData ManagedCuda.BasicTypes.CUdeviceptr
srcDiffDesc cudnnTensorDescriptor
srcDiffData ManagedCuda.BasicTypes.CUdeviceptr
destDesc cudnnTensorDescriptor
destData ManagedCuda.BasicTypes.CUdeviceptr
beta double
destDiffDesc cudnnTensorDescriptor
destDiffData ManagedCuda.BasicTypes.CUdeviceptr
return cudnnStatus
Exemplo n.º 1
0
 /* Function to perform backward activation  */
 public void ActivationBackward(cudnnActivationMode mode,
                                double alpha,
                                TensorDescriptor srcDesc,
                                CudaDeviceVariable <double> srcData,
                                TensorDescriptor srcDiffDesc,
                                CudaDeviceVariable <double> srcDiffData,
                                TensorDescriptor destDesc,
                                CudaDeviceVariable <double> destData,
                                double beta,
                                TensorDescriptor destDiffDesc,
                                CudaDeviceVariable <double> destDiffData
                                )
 {
     res = CudaDNNNativeMethods.cudnnActivationBackward(_handle, mode, ref alpha, srcDesc.Desc, srcData.DevicePointer, srcDiffDesc.Desc, srcDiffData.DevicePointer, destDesc.Desc, destData.DevicePointer, ref beta, destDiffDesc.Desc, destDiffData.DevicePointer);
     Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnActivationForward", res));
     if (res != cudnnStatus.Success)
     {
         throw new CudaDNNException(res);
     }
 }