cudnnActivationForward() private method

private cudnnActivationForward ( cudnnHandle handle, cudnnActivationDescriptor activationDesc, double &alpha, cudnnTensorDescriptor srcDesc, ManagedCuda.BasicTypes.CUdeviceptr srcData, double &beta, cudnnTensorDescriptor destDesc, ManagedCuda.BasicTypes.CUdeviceptr destData ) : cudnnStatus
handle cudnnHandle
activationDesc cudnnActivationDescriptor
alpha double
srcDesc cudnnTensorDescriptor
srcData ManagedCuda.BasicTypes.CUdeviceptr
beta double
destDesc cudnnTensorDescriptor
destData ManagedCuda.BasicTypes.CUdeviceptr
return cudnnStatus
Ejemplo n.º 1
0
        /* Activation functions: All of the form "output = alpha * Op(inputs) + beta * output" */

        /* Function to perform forward activation  */
        public void ActivationForward(cudnnActivationMode mode,
                                      double alpha,
                                      TensorDescriptor srcDesc,
                                      CudaDeviceVariable <double> srcData,
                                      double beta,
                                      TensorDescriptor destDesc,
                                      CudaDeviceVariable <double> destData
                                      )
        {
            res = CudaDNNNativeMethods.cudnnActivationForward(_handle, mode, ref alpha, srcDesc.Desc, srcData.DevicePointer, ref beta, destDesc.Desc, destData.DevicePointer);
            Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnActivationForward", res));
            if (res != cudnnStatus.Success)
            {
                throw new CudaDNNException(res);
            }
        }