Uses of Class
jcuda.jcudnn.cudnnSeqDataDescriptor

Uses of cudnnSeqDataDescriptor in jcuda.jcudnn

Methods in jcuda.jcudnn with parameters of type cudnnSeqDataDescriptor

Modifier and Type	Method and Description
static int	JCudnn.cudnnCreateSeqDataDescriptor(cudnnSeqDataDescriptor seqDataDesc)
static int	JCudnn.cudnnDestroySeqDataDescriptor(cudnnSeqDataDescriptor seqDataDesc)
static int	JCudnn.cudnnGetSeqDataDescriptor(cudnnSeqDataDescriptor seqDataDesc, int[] dataType, int[] nbDims, int nbDimsRequested, int[] dimA, int[] axes, long[] seqLengthArraySize, long seqLengthSizeRequested, int[] seqLengthArray, Pointer paddingFill)
static int	JCudnn.cudnnMultiHeadAttnBackwardData(cudnnHandle handle, cudnnAttnDescriptor attnDesc, int[] loWinIdx, int[] hiWinIdx, Pointer devSeqLengthsDQDO, Pointer devSeqLengthsDKDV, cudnnSeqDataDescriptor doDesc, Pointer dout, cudnnSeqDataDescriptor dqDesc, Pointer dqueries, Pointer queries, cudnnSeqDataDescriptor dkDesc, Pointer dkeys, Pointer keys, cudnnSeqDataDescriptor dvDesc, Pointer dvalues, Pointer values, long weightSizeInBytes, Pointer weights, long workSpaceSizeInBytes, Pointer workSpace, long reserveSpaceSizeInBytes, Pointer reserveSpace)
static int	JCudnn.cudnnMultiHeadAttnBackwardWeights(cudnnHandle handle, cudnnAttnDescriptor attnDesc, int addGrad, cudnnSeqDataDescriptor qDesc, Pointer queries, cudnnSeqDataDescriptor kDesc, Pointer keys, cudnnSeqDataDescriptor vDesc, Pointer values, cudnnSeqDataDescriptor doDesc, Pointer dout, long weightSizeInBytes, Pointer weights, Pointer dweights, long workSpaceSizeInBytes, Pointer workSpace, long reserveSpaceSizeInBytes, Pointer reserveSpace)
static int	JCudnn.cudnnMultiHeadAttnForward(cudnnHandle handle, cudnnAttnDescriptor attnDesc, int currIdx, int[] loWinIdx, int[] hiWinIdx, Pointer devSeqLengthsQO, Pointer devSeqLengthsKV, cudnnSeqDataDescriptor qDesc, Pointer queries, Pointer residuals, cudnnSeqDataDescriptor kDesc, Pointer keys, cudnnSeqDataDescriptor vDesc, Pointer values, cudnnSeqDataDescriptor oDesc, Pointer out, long weightSizeInBytes, Pointer weights, long workSpaceSizeInBytes, Pointer workSpace, long reserveSpaceSizeInBytes, Pointer reserveSpace)
static int	JCudnn.cudnnSetSeqDataDescriptor(cudnnSeqDataDescriptor seqDataDesc, int dataType, int nbDims, int[] dimA, int[] axes, long seqLengthArraySize, int[] seqLengthArray, Pointer paddingFill)