Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreatePersistentRNNPlan(cudnnRNNDescriptor rnnDesc,
int minibatch,
int dataType,
cudnnPersistentRNNPlan plan)
Expensive.
|
static int |
JCudnn.cudnnCreateRNNDescriptor(cudnnRNNDescriptor rnnDesc) |
static int |
JCudnn.cudnnDestroyRNNDescriptor(cudnnRNNDescriptor rnnDesc) |
static int |
JCudnn.cudnnFindRNNBackwardDataAlgorithmEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor[] dyDesc,
Pointer dy,
cudnnTensorDescriptor dhyDesc,
Pointer dhy,
cudnnTensorDescriptor dcyDesc,
Pointer dcy,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnTensorDescriptor[] dxDesc,
Pointer dx,
cudnnTensorDescriptor dhxDesc,
Pointer dhx,
cudnnTensorDescriptor dcxDesc,
Pointer dcx,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformance[] perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor[] yDesc,
Pointer y,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformance[] perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterDescriptor dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnFindRNNForwardInferenceAlgorithmEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformance[] perfResults,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
JCudnn.cudnnFindRNNForwardTrainingAlgorithmEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformance[] perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnGetRNNBackwardDataAlgorithmMaxCount(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] count) |
static int |
JCudnn.cudnnGetRNNBackwardWeightsAlgorithmMaxCount(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] count) |
static int |
JCudnn.cudnnGetRNNBiasMode(cudnnRNNDescriptor rnnDesc,
int[] biasMode) |
static int |
JCudnn.cudnnGetRNNDescriptor(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] hiddenSize,
int[] numLayers,
cudnnDropoutDescriptor dropoutDesc,
int[] inputMode,
int[] direction,
int[] mode,
int[] algo,
int[] mathPrec) |
static int |
JCudnn.cudnnGetRNNForwardInferenceAlgorithmMaxCount(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] count) |
static int |
JCudnn.cudnnGetRNNForwardTrainingAlgorithmMaxCount(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] count) |
static int |
JCudnn.cudnnGetRNNLinLayerBiasParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int pseudoLayer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerBiasDesc,
Pointer linLayerBias) |
static int |
JCudnn.cudnnGetRNNLinLayerMatrixParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int pseudoLayer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerMatDesc,
Pointer linLayerMat) |
static int |
JCudnn.cudnnGetRNNMatrixMathType(cudnnRNNDescriptor rnnDesc,
int[] mType) |
static int |
JCudnn.cudnnGetRNNPaddingMode(cudnnRNNDescriptor rnnDesc,
int[] paddingMode) |
static int |
JCudnn.cudnnGetRNNParamsSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes,
int dataType) |
static int |
JCudnn.cudnnGetRNNProjectionLayers(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] recProjSize,
int[] outProjSize) |
static int |
JCudnn.cudnnGetRNNTrainingReserveSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
long[] sizeInBytes) |
static int |
JCudnn.cudnnGetRNNWorkspaceSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
long[] sizeInBytes)
dataType in weight descriptors and input descriptors is used to describe storage
|
static int |
JCudnn.cudnnRNNBackwardData(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor[] dyDesc,
Pointer dy,
cudnnTensorDescriptor dhyDesc,
Pointer dhy,
cudnnTensorDescriptor dcyDesc,
Pointer dcy,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnTensorDescriptor[] dxDesc,
Pointer dx,
cudnnTensorDescriptor dhxDesc,
Pointer dhx,
cudnnTensorDescriptor dcxDesc,
Pointer dcx,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
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) |
static int |
JCudnn.cudnnRNNBackwardWeights(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor[] yDesc,
Pointer y,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterDescriptor dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
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) |
static int |
JCudnn.cudnnRNNForwardInference(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
Pointer workspace,
long workSpaceSizeInBytes) |
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) |
static int |
JCudnn.cudnnRNNForwardTraining(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
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) |
static int |
JCudnn.cudnnRNNGetClip(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] clipMode,
int[] clipNanOpt,
double[] lclip,
double[] rclip) |
static int |
JCudnn.cudnnRNNSetClip(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int clipMode,
int clipNanOpt,
double lclip,
double rclip) |
static int |
JCudnn.cudnnSetPersistentRNNPlan(cudnnRNNDescriptor rnnDesc,
cudnnPersistentRNNPlan plan) |
static int |
JCudnn.cudnnSetRNNAlgorithmDescriptor(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnAlgorithmDescriptor algoDesc) |
static int |
JCudnn.cudnnSetRNNBiasMode(cudnnRNNDescriptor rnnDesc,
int biasMode) |
static int |
JCudnn.cudnnSetRNNDescriptor_v5(cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int mode,
int mathPrec) |
static int |
JCudnn.cudnnSetRNNDescriptor_v6(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int mode,
int algo,
int mathPrec)
DEPRECATED routines to be removed next release :
User should use the non-suffixed version (which has the API and functionality of _v6 version)
Routines with _v5 suffix has the functionality of the non-suffixed routines in the CUDNN V6
|
static int |
JCudnn.cudnnSetRNNDescriptor(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int mode,
int algo,
int mathPrec)
dropout is between RNN layers, not between recurrent steps
|
static int |
JCudnn.cudnnSetRNNMatrixMathType(cudnnRNNDescriptor rnnDesc,
int mType) |
static int |
JCudnn.cudnnSetRNNPaddingMode(cudnnRNNDescriptor rnnDesc,
int paddingMode)
RNN EX API
|
static int |
JCudnn.cudnnSetRNNProjectionLayers(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int recProjSize,
int outProjSize) |
Copyright © 2019. All rights reserved.