Library/ManagedCuda/CudaDNN/CudaDNNNativeMethods.cs
/// <summary>
/// This function reduces tensor A by implementing the equation C = alpha * reduce op ( A )
/// + beta* C, given tensors A and C and scaling factors alpha and beta.The reduction op
/// to use is indicated by the descriptor reduceTensorDesc.Currently-supported ops are
/// listed by the cudnnReduceTensorOp_t enum.
/// </summary>
/// <param name="handle">Handle to a previously created cuDNN context.</param>
/// <param name="reduceTensorDesc">Handle to a previously initialized reduce tensor descriptor.</param>
/// <param name="indices">Handle to a previously allocated space for writing indices.</param>
/// <param name="indicesSizeInBytes">Size of the above previously allocated space.</param>
/// <param name="workspace">Handle to a previously allocated space for the reduction implementation.</param>
/// <param name="workspaceSizeInBytes">Size of the above previously allocated space.</param>
/// <param name="alpha">Pointer to scaling factor (in host memory) used to blend the source value
/// with prior value in the destination tensor as indicated by the above op equation.</param>
/// <param name="aDesc">Handle to a previously initialized tensor descriptor.</param>
/// <param name="A">Pointer to data of the tensor described by the aDesc descriptor.</param>
/// <param name="beta">Pointer to scaling factor (in host memory) used to blend the source value
/// with prior value in the destination tensor as indicated by the above op equation.</param>
/// <param name="cDesc">Handle to a previously initialized tensor descriptor.</param>
/// <param name="C">Pointer to data of the tensor described by the cDesc descriptor.</param>
This function reduces tensor A by implementing the equation C = alpha * reduce op ( A )
+ beta* C, given tensors A and C and scaling factors alpha and beta.The reduction op
to use is indicated by the descriptor reduceTensorDesc.Currently-supported ops are
listed by the cudnnReduceTensorOp_t enum.
This function reduces tensor A by implementing the equation C = alpha * reduce op ( A ) + beta * C given tensors A and C and scaling factors alpha and beta . The reduction op to use is indicated by the descriptor reduceTensorDesc . Currently - supported ops are listed by the cudnnReduceTensorOp_t enum .
[DllImport(CUDNN_API_DLL_NAME)]
        public static extern cudnnStatus cudnnReduceTensor(
                                cudnnHandle handle,
                                cudnnReduceTensorDescriptor reduceTensorDesc,
                                CUdeviceptr indices,
                                SizeT indicesSizeInBytes,
                                CUdeviceptr workspace,
                                SizeT workspaceSizeInBytes,
                                ref double alpha,
                                cudnnTensorDescriptor aDesc,
                                CUdeviceptr A,
                                ref double beta,
                                cudnnTensorDescriptor cDesc,
                                CUdeviceptr C);
[ DllImport ( CUDNN_API_DLL_NAME ) ] public static extern cudnnStatus cudnnReduceTensor ( cudnnHandle handle , cudnnReduceTensorDescriptor reduceTensorDesc , CUdeviceptr indices , SizeT indicesSizeInBytes , CUdeviceptr workspace , SizeT workspaceSizeInBytes , ref double alpha , cudnnTensorDescriptor aDesc , CUdeviceptr A , ref double beta , cudnnTensorDescriptor cDesc , CUdeviceptr C ) ;
This function reduces tensor A by implementing the equation C = alpha * reduce op ( A )
+ beta* C, given tensors A and C and scaling factors alpha and beta.The reduction op
to use is indicated by the descriptor reduceTensorDesc.Currently-supported ops are
listed by the cudnnReduceTensorOp_t enum.
This function reduces tensor A by implementing the equation C = alpha * reduce op ( A ) + beta * C given tensors A and C and scaling factors alpha and beta . The reduction op to use is indicated by the descriptor reduceTensorDesc . Currently - supported ops are listed by the cudnnReduceTensorOp_t enum .
handle
cudnnHandle
Handle to a previously created cuDNN context.
Handle to a previously created cuDNN context .
reduceTensorDesc
cudnnReduceTensorDescriptor
Handle to a previously initialized reduce tensor descriptor.
Handle to a previously initialized reduce tensor descriptor .
indices
CUdeviceptr
Handle to a previously allocated space for writing indices.
Handle to a previously allocated space for writing indices .
indicesSizeInBytes
SizeT
Size of the above previously allocated space.
Size of the above previously allocated space .
workspace
CUdeviceptr
Handle to a previously allocated space for the reduction implementation.
Handle to a previously allocated space for the reduction implementation .
workspaceSizeInBytes
SizeT
Size of the above previously allocated space.
Size of the above previously allocated space .
double
ref
aDesc
cudnnTensorDescriptor
Handle to a previously initialized tensor descriptor.
Handle to a previously initialized tensor descriptor .
A
CUdeviceptr
Pointer to data of the tensor described by the aDesc descriptor.
Pointer to data of the tensor described by the aDesc descriptor .
cDesc
cudnnTensorDescriptor
Handle to a previously initialized tensor descriptor.
Handle to a previously initialized tensor descriptor .
C
CUdeviceptr
Pointer to data of the tensor described by the cDesc descriptor.
Pointer to data of the tensor described by the cDesc descriptor .
alpha
Pointer to scaling factor (in host memory) used to blend the source value
with prior value in the destination tensor as indicated by the above op equation.
Pointer to scaling factor ( in host memory ) used to blend the source value with prior value in the destination tensor as indicated by the above op equation .
beta
Pointer to scaling factor (in host memory) used to blend the source value
with prior value in the destination tensor as indicated by the above op equation.
Pointer to scaling factor ( in host memory ) used to blend the source value with prior value in the destination tensor as indicated by the above op equation .