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
示例#1
0
 /* 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);
     }
 }