Package | Description |
---|---|
org.bytedeco.cuda.global |
Modifier and Type | Method and Description |
---|---|
static int |
cudnn.cudnnCreateRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc) |
static int |
cudnn.cudnnDestroyRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc) |
static int |
cudnn.cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
int[] dataType,
int[] layout,
int[] maxSeqLength,
int[] batchSize,
int[] vectorSize,
int arrayLengthRequested,
int[] seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
IntBuffer dataType,
IntBuffer layout,
IntBuffer maxSeqLength,
IntBuffer batchSize,
IntBuffer vectorSize,
int arrayLengthRequested,
IntBuffer seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
IntPointer dataType,
IntPointer layout,
IntPointer maxSeqLength,
IntPointer batchSize,
IntPointer vectorSize,
int arrayLengthRequested,
IntPointer seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnGetRNNTempSpaceSizes(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int fwdMode,
cudnnRNNDataStruct xDesc,
SizeTPointer workSpaceSize,
SizeTPointer reserveSpaceSize) |
static int |
cudnn.cudnnRNNBackwardData_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] devSeqLengths,
cudnnRNNDataStruct yDesc,
Pointer y,
Pointer dy,
cudnnRNNDataStruct xDesc,
Pointer dx,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer dhy,
Pointer dhx,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer dcy,
Pointer dcx,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardData_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer devSeqLengths,
cudnnRNNDataStruct yDesc,
Pointer y,
Pointer dy,
cudnnRNNDataStruct xDesc,
Pointer dx,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer dhy,
Pointer dhx,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer dcy,
Pointer dcx,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardData_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer devSeqLengths,
cudnnRNNDataStruct yDesc,
Pointer y,
Pointer dy,
cudnnRNNDataStruct xDesc,
Pointer dx,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer dhy,
Pointer dhx,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer dcy,
Pointer dcx,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardDataEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnRNNDataStruct dyDesc,
Pointer dy,
cudnnRNNDataStruct dcDesc,
Pointer dcAttn,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnRNNDataStruct dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
cudnnRNNDataStruct dkDesc,
Pointer dkeys,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
cudnn.cudnnRNNBackwardWeights_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int addGrad,
int[] devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hDesc,
Pointer hx,
cudnnRNNDataStruct yDesc,
Pointer y,
long weightSpaceSize,
Pointer dweightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardWeights_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int addGrad,
IntBuffer devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hDesc,
Pointer hx,
cudnnRNNDataStruct yDesc,
Pointer y,
long weightSpaceSize,
Pointer dweightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardWeights_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int addGrad,
IntPointer devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hDesc,
Pointer hx,
cudnnRNNDataStruct yDesc,
Pointer y,
long weightSpaceSize,
Pointer dweightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardWeightsEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnRNNDataStruct yDesc,
Pointer y,
Pointer workSpace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
cudnn.cudnnRNNForward(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int fwdMode,
int[] devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer hy,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer cy,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNForward(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int fwdMode,
IntBuffer devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer hy,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer cy,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNForward(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int fwdMode,
IntPointer devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer hy,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer cy,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNForwardInferenceEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
cudnnRNNDataStruct kDesc,
Pointer keys,
cudnnRNNDataStruct cDesc,
Pointer cAttn,
cudnnRNNDataStruct iDesc,
Pointer iAttn,
cudnnRNNDataStruct qDesc,
Pointer queries,
Pointer workSpace,
long workSpaceSizeInBytes)
Deprecated.
|
static int |
cudnn.cudnnRNNForwardTrainingEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
cudnnRNNDataStruct kDesc,
Pointer keys,
cudnnRNNDataStruct cDesc,
Pointer cAttn,
cudnnRNNDataStruct iDesc,
Pointer iAttn,
cudnnRNNDataStruct qDesc,
Pointer queries,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
cudnn.cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
int dataType,
int layout,
int maxSeqLength,
int batchSize,
int vectorSize,
int[] seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
int dataType,
int layout,
int maxSeqLength,
int batchSize,
int vectorSize,
IntBuffer seqLengthArray,
Pointer paddingFill) |
static int |
cudnn.cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
int dataType,
int layout,
int maxSeqLength,
int batchSize,
int vectorSize,
IntPointer seqLengthArray,
Pointer paddingFill) |
Copyright © 2024. All rights reserved.