| 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[] seqLengthArrayDQDO,
int[] seqLengthArrayDKDV,
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 w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnBackwardData(cudnnContext handle,
cudnnAttnStruct attnDesc,
IntBuffer loWinIdx,
IntBuffer hiWinIdx,
IntBuffer seqLengthArrayDQDO,
IntBuffer seqLengthArrayDKDV,
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 w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnBackwardData(cudnnContext handle,
cudnnAttnStruct attnDesc,
IntPointer loWinIdx,
IntPointer hiWinIdx,
IntPointer seqLengthArrayDQDO,
IntPointer seqLengthArrayDKDV,
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 w,
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 w,
Pointer dw,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
int[] loWinIdx,
int[] hiWinIdx,
int[] seqLengthArrayQRO,
int[] seqLengthArrayKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
IntBuffer loWinIdx,
IntBuffer hiWinIdx,
IntBuffer seqLengthArrayQRO,
IntBuffer seqLengthArrayKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnn.cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
IntPointer loWinIdx,
IntPointer hiWinIdx,
IntPointer seqLengthArrayQRO,
IntPointer seqLengthArrayKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer w,
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 © 2019. All rights reserved.