Ejemplos de ManagedCuda.CudaDNN CudaDNNNativeMethods.cudnnAddTensor en C# (CSharp)

cudnnAddTensor() private method

private cudnnAddTensor ( cudnnHandle handle, double &alpha, cudnnTensorDescriptor biasDesc, ManagedCuda.BasicTypes.CUdeviceptr biasData, double &beta, cudnnTensorDescriptor srcDestDesc, ManagedCuda.BasicTypes.CUdeviceptr srcDestData ) : cudnnStatus
handle	cudnnHandle
alpha	double
biasDesc	cudnnTensorDescriptor
biasData	ManagedCuda.BasicTypes.CUdeviceptr
beta	double
srcDestDesc	cudnnTensorDescriptor
srcDestData	ManagedCuda.BasicTypes.CUdeviceptr
return	cudnnStatus

Ejemplo n.º 1

 /* Tensor Bias addition : srcDest = alpha * bias + beta * srcDestDesc  */
 public void AddTensor(cudnnAddMode mode,
                       double alpha,
                       TensorDescriptor biasDesc,
                       CudaDeviceVariable <double> biasData,
                       double beta,
                       TensorDescriptor srcDestDesc,
                       CudaDeviceVariable <double> srcDestData
                       )
 {
     res = CudaDNNNativeMethods.cudnnAddTensor(_handle, mode, ref alpha, biasDesc.Desc, biasData.DevicePointer, ref beta, srcDestDesc.Desc, srcDestData.DevicePointer);
     Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnAddTensor", res));
     if (res != cudnnStatus.Success)
     {
         throw new CudaDNNException(res);
     }
 }