| Modifier and Type | Method and Description |
|---|---|
static int |
JCudnn.cudnnCreateRNNDataDescriptor(cudnnRNNDataDescriptor rnnDataDesc) |
static int |
JCudnn.cudnnDestroyRNNDataDescriptor(cudnnRNNDataDescriptor rnnDataDesc) |
static int |
JCudnn.cudnnGetRNNDataDescriptor(cudnnRNNDataDescriptor rnnDataDesc,
int[] dataType,
int[] layout,
int[] maxSeqLength,
int[] batchSize,
int[] vectorSize,
int arrayLengthRequested,
int[] seqLengthArray,
Pointer paddingFill)
symbol for filling padding position in output
|
static int |
JCudnn.cudnnGetRNNTempSpaceSizes(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int fwdMode,
cudnnRNNDataDescriptor xDesc,
long[] workSpaceSize,
long[] reserveSpaceSize) |
static int |
JCudnn.cudnnRNNBackwardData_v8(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
Pointer devSeqLengths,
cudnnRNNDataDescriptor yDesc,
Pointer y,
Pointer dy,
cudnnRNNDataDescriptor xDesc,
Pointer dx,
cudnnTensorDescriptor hDesc,
Pointer hx,
Pointer dhy,
Pointer dhx,
cudnnTensorDescriptor cDesc,
Pointer cx,
Pointer dcy,
Pointer dcx,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
JCudnn.cudnnRNNBackwardWeights_v8(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int addGrad,
Pointer devSeqLengths,
cudnnRNNDataDescriptor xDesc,
Pointer x,
cudnnTensorDescriptor hDesc,
Pointer hx,
cudnnRNNDataDescriptor yDesc,
Pointer y,
long weightSpaceSize,
Pointer dweightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
JCudnn.cudnnRNNForward(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int fwdMode,
Pointer devSeqLengths,
cudnnRNNDataDescriptor xDesc,
Pointer x,
cudnnRNNDataDescriptor yDesc,
Pointer y,
cudnnTensorDescriptor hDesc,
Pointer hx,
Pointer hy,
cudnnTensorDescriptor cDesc,
Pointer cx,
Pointer cy,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
JCudnn.cudnnSetRNNDataDescriptor(cudnnRNNDataDescriptor rnnDataDesc,
int dataType,
int layout,
int maxSeqLength,
int batchSize,
int vectorSize,
int[] seqLengthArray,
Pointer paddingFill) |
Copyright © 2024. All rights reserved.