Library/ManagedCuda/CudaDNN/CudaDNNNativeMethods.cs
/* Helper function to return the minimum size of the index space to be passed to the reduction given the input and output tensors */
/// <summary>
/// Helper function to return the minimum size of the index space to be passed to the reduction given the input and output tensors
/// </summary>
/// <param name="handle"></param>
/// <param name="reduceTensorDesc"></param>
/// <param name="aDesc"></param>
/// <param name="cDesc"></param>
/// <param name="sizeInBytes"></param>
Helper function to return the minimum size of the index space to be passed to the reduction given the input and output tensors
Helper function to return the minimum size of the index space to be passed to the reduction given the input and output tensors
[DllImport(CUDNN_API_DLL_NAME)]
        public static extern cudnnStatus cudnnGetReductionIndicesSize(
                                        cudnnHandle handle,
                                cudnnReduceTensorDescriptor reduceTensorDesc,
                                cudnnTensorDescriptor aDesc,
                                cudnnTensorDescriptor cDesc,
                                ref SizeT sizeInBytes );
[ DllImport ( CUDNN_API_DLL_NAME ) ] public static extern cudnnStatus cudnnGetReductionIndicesSize ( cudnnHandle handle , cudnnReduceTensorDescriptor reduceTensorDesc , cudnnTensorDescriptor aDesc , cudnnTensorDescriptor cDesc , ref SizeT sizeInBytes ) ;
Helper function to return the minimum size of the index space to be passed to the reduction given the input and output tensors
Helper function to return the minimum size of the index space to be passed to the reduction given the input and output tensors
handle
cudnnHandle
None
reduceTensorDesc
cudnnReduceTensorDescriptor
None
aDesc
cudnnTensorDescriptor
None
cDesc
cudnnTensorDescriptor
None
SizeT
ref
sizeInBytes
None