Package | Description |
---|---|
org.bytedeco.cuda.global |
Modifier and Type | Method and Description |
---|---|
static int |
cudnn.cudnnCreateSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc) |
static int |
cudnn.cudnnDestroySeqDataDescriptor(cudnnSeqDataStruct seqDataDesc) |
static int |
cudnn.cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
int[] dataType,
int[] nbDims,
int nbDimsRequested,
int[] dimA,
int[] axes,
SizeTPointer seqLengthArraySize,
long seqLengthSizeRequested,
int[] seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
IntBuffer dataType,
IntBuffer nbDims,
int nbDimsRequested,
IntBuffer dimA,
IntBuffer axes,
SizeTPointer seqLengthArraySize,
long seqLengthSizeRequested,
IntBuffer seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
IntPointer dataType,
IntPointer nbDims,
int nbDimsRequested,
IntPointer dimA,
IntPointer axes,
SizeTPointer seqLengthArraySize,
long seqLengthSizeRequested,
IntPointer seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnMultiHeadAttnBackwardData(cudnnContext handle,
cudnnAttnStruct attnDesc,
int[] loWinIdx,
int[] hiWinIdx,
int[] devSeqLengthsDQDO,
int[] devSeqLengthsDKDV,
cudnnSeqDataStruct doDesc,
Pointer dout,
cudnnSeqDataStruct dqDesc,
Pointer dqueries,
Pointer queries,
cudnnSeqDataStruct dkDesc,
Pointer dkeys,
Pointer keys,
cudnnSeqDataStruct dvDesc,
Pointer dvalues,
Pointer values,
long weightSizeInBytes,
Pointer weights,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnBackwardData(cudnnContext handle,
cudnnAttnStruct attnDesc,
IntBuffer loWinIdx,
IntBuffer hiWinIdx,
IntBuffer devSeqLengthsDQDO,
IntBuffer devSeqLengthsDKDV,
cudnnSeqDataStruct doDesc,
Pointer dout,
cudnnSeqDataStruct dqDesc,
Pointer dqueries,
Pointer queries,
cudnnSeqDataStruct dkDesc,
Pointer dkeys,
Pointer keys,
cudnnSeqDataStruct dvDesc,
Pointer dvalues,
Pointer values,
long weightSizeInBytes,
Pointer weights,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnBackwardData(cudnnContext handle,
cudnnAttnStruct attnDesc,
IntPointer loWinIdx,
IntPointer hiWinIdx,
IntPointer devSeqLengthsDQDO,
IntPointer devSeqLengthsDKDV,
cudnnSeqDataStruct doDesc,
Pointer dout,
cudnnSeqDataStruct dqDesc,
Pointer dqueries,
Pointer queries,
cudnnSeqDataStruct dkDesc,
Pointer dkeys,
Pointer keys,
cudnnSeqDataStruct dvDesc,
Pointer dvalues,
Pointer values,
long weightSizeInBytes,
Pointer weights,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnBackwardWeights(cudnnContext handle,
cudnnAttnStruct attnDesc,
int addGrad,
cudnnSeqDataStruct qDesc,
Pointer queries,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct doDesc,
Pointer dout,
long weightSizeInBytes,
Pointer weights,
Pointer dweights,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
int[] loWinIdx,
int[] hiWinIdx,
int[] devSeqLengthsQO,
int[] devSeqLengthsKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer weights,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
IntBuffer loWinIdx,
IntBuffer hiWinIdx,
IntBuffer devSeqLengthsQO,
IntBuffer devSeqLengthsKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer weights,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
IntPointer loWinIdx,
IntPointer hiWinIdx,
IntPointer devSeqLengthsQO,
IntPointer devSeqLengthsKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer weights,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
int dataType,
int nbDims,
int[] dimA,
int[] axes,
long seqLengthArraySize,
int[] seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
int dataType,
int nbDims,
IntBuffer dimA,
IntBuffer axes,
long seqLengthArraySize,
IntBuffer seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
int dataType,
int nbDims,
IntPointer dimA,
IntPointer axes,
long seqLengthArraySize,
IntPointer seqLengthArray,
Pointer paddingFill) |
Copyright © 2024. All rights reserved.