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,
int[] devSeqLengthsDQDO,
int[] 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,
int[] devSeqLengthsQO,
int[] 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) |
Copyright © 2020. All rights reserved.