Library/ManagedCuda/CudaDNN/CudaDNNNativeMethods.cs
/// <summary>
/// This function is used to query the amount of reserved space required for training the 
/// RNN described by rnnDesc with inputs dimensions defined by xDesc. The same reserve 
/// space must be passed to cudnnRNNForwardTraining, cudnnRNNBackwardData and cudnnRNNBackwardWeights.
/// </summary>
/// <param name="handle">Handle to a previously created cuDNN library descriptor.</param>
/// <param name="rnnDesc">A previously initialized RNN descriptor.</param>
/// <param name="seqLength">Number of iterations to unroll over.</param>
/// <param name="xDesc">An array of tensor descriptors describing the input to each recurrent iteration.</param>
/// <param name="sizeInBytes">Minimum amount of GPU memory needed as reserve space to be able to train an RNN with the specified descriptor and input tensors.</param>
This function is used to query the amount of reserved space required for training the
RNN described by rnnDesc with inputs dimensions defined by xDesc. The same reserve
space must be passed to cudnnRNNForwardTraining, cudnnRNNBackwardData and cudnnRNNBackwardWeights.
This function is used to query the amount of reserved space required for training the RNN described by rnnDesc with inputs dimensions defined by xDesc . The same reserve space must be passed to cudnnRNNForwardTraining cudnnRNNBackwardData and cudnnRNNBackwardWeights .
[DllImport(CUDNN_API_DLL_NAME)]
        public static extern cudnnStatus cudnnGetRNNTrainingReserveSize(cudnnHandle handle,
                                                          cudnnRNNDescriptor rnnDesc,
                                                          int seqLength,
                                                          cudnnTensorDescriptor[] xDesc,
                                                          ref SizeT                     sizeInBytes
                                                    );
[ DllImport ( CUDNN_API_DLL_NAME ) ] public static extern cudnnStatus cudnnGetRNNTrainingReserveSize ( cudnnHandle handle , cudnnRNNDescriptor rnnDesc , int seqLength , cudnnTensorDescriptor [ ] xDesc , ref SizeT sizeInBytes ) ;
This function is used to query the amount of reserved space required for training the
RNN described by rnnDesc with inputs dimensions defined by xDesc.
This function is used to query the amount of reserved space required for training the RNN described by rnnDesc with inputs dimensions defined by xDesc .
handle
cudnnHandle
Handle to a previously created cuDNN library descriptor.
Handle to a previously created cuDNN library descriptor .
rnnDesc
cudnnRNNDescriptor
A previously initialized RNN descriptor.
A previously initialized RNN descriptor .
seqLength
int
Number of iterations to unroll over.
Number of iterations to unroll over .
xDesc
cudnnTensorDescriptor[]
An array of tensor descriptors describing the input to each recurrent iteration.
An array of tensor descriptors describing the input to each recurrent iteration .
SizeT
ref
sizeInBytes
Minimum amount of GPU memory needed as reserve space to be able to train an RNN with the specified descriptor and input tensors.
Minimum amount of GPU memory needed as reserve space to be able to train an RNN with the specified descriptor and input tensors .