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 fMode,
cudnnRNNDataDescriptor xDesc,
long[] workSpaceSize,
long[] reserveSpaceSize) |
static int |
JCudnn.cudnnRNNBackwardData_v8(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] 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.cudnnRNNBackwardDataEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnRNNDataDescriptor yDesc,
Pointer y,
cudnnRNNDataDescriptor dyDesc,
Pointer dy,
cudnnRNNDataDescriptor dcDesc,
Pointer dcAttn,
cudnnTensorDescriptor dhyDesc,
Pointer dhy,
cudnnTensorDescriptor dcyDesc,
Pointer dcy,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnRNNDataDescriptor dxDesc,
Pointer dx,
cudnnTensorDescriptor dhxDesc,
Pointer dhx,
cudnnTensorDescriptor dcxDesc,
Pointer dcx,
cudnnRNNDataDescriptor dkDesc,
Pointer dkeys,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNBackwardWeights_v8(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int addGrad,
int[] 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.cudnnRNNBackwardWeightsEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnRNNDataDescriptor xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnRNNDataDescriptor yDesc,
Pointer y,
Pointer workSpace,
long workSpaceSizeInBytes,
cudnnFilterDescriptor dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNForward(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int fwdMode,
int[] 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.cudnnRNNForwardInferenceEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnRNNDataDescriptor xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnRNNDataDescriptor yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
cudnnRNNDataDescriptor kDesc,
Pointer keys,
cudnnRNNDataDescriptor cDesc,
Pointer cAttn,
cudnnRNNDataDescriptor iDesc,
Pointer iAttn,
cudnnRNNDataDescriptor qDesc,
Pointer queries,
Pointer workSpace,
long workSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNForwardTrainingEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnRNNDataDescriptor xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnRNNDataDescriptor yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
cudnnRNNDataDescriptor kDesc,
Pointer keys,
cudnnRNNDataDescriptor cDesc,
Pointer cAttn,
cudnnRNNDataDescriptor iDesc,
Pointer iAttn,
cudnnRNNDataDescriptor qDesc,
Pointer queries,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnSetRNNDataDescriptor(cudnnRNNDataDescriptor rnnDataDesc,
int dataType,
int layout,
int maxSeqLength,
int batchSize,
int vectorSize,
int[] seqLengthArray,
Pointer paddingFill) |
Copyright © 2020. All rights reserved.