| Modifier and Type | Method and Description |
|---|---|
static int |
JCudnn.cudnnBuildRNNDynamic(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int miniBatch) |
static int |
JCudnn.cudnnCreateRNNDescriptor(cudnnRNNDescriptor rnnDesc) |
static int |
JCudnn.cudnnDestroyRNNDescriptor(cudnnRNNDescriptor rnnDesc) |
static int |
JCudnn.cudnnGetRNNDescriptor_v8(cudnnRNNDescriptor rnnDesc,
int[] algo,
int[] cellMode,
int[] biasMode,
int[] dirMode,
int[] inputMode,
int[] dataType,
int[] mathPrec,
int[] mathType,
int[] inputSize,
int[] hiddenSize,
int[] projSize,
int[] numLayers,
cudnnDropoutDescriptor dropoutDesc,
int[] auxFlags) |
static int |
JCudnn.cudnnGetRNNTempSpaceSizes(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int fwdMode,
cudnnRNNDataDescriptor xDesc,
long[] workSpaceSize,
long[] reserveSpaceSize) |
static int |
JCudnn.cudnnGetRNNWeightParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int pseudoLayer,
long weightSpaceSize,
Pointer weightSpace,
int linLayerID,
cudnnTensorDescriptor mDesc,
Pointer mAddr,
cudnnTensorDescriptor bDesc,
Pointer bAddr) |
static int |
JCudnn.cudnnGetRNNWeightSpaceSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
long[] weightSpaceSize) |
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.cudnnRNNGetClip_v8(cudnnRNNDescriptor rnnDesc,
int[] clipMode,
int[] clipNanOpt,
double[] lclip,
double[] rclip)
Deprecated.
|
static int |
JCudnn.cudnnRNNGetClip_v9(cudnnRNNDescriptor rnnDesc,
int[] clipMode,
double[] lclip,
double[] rclip) |
static int |
JCudnn.cudnnRNNSetClip_v8(cudnnRNNDescriptor rnnDesc,
int clipMode,
int clipNanOpt,
double lclip,
double rclip)
Deprecated.
|
static int |
JCudnn.cudnnRNNSetClip_v9(cudnnRNNDescriptor rnnDesc,
int clipMode,
double lclip,
double rclip) |
static int |
JCudnn.cudnnSetRNNDescriptor_v8(cudnnRNNDescriptor rnnDesc,
int algo,
int cellMode,
int biasMode,
int dirMode,
int inputMode,
int dataType,
int mathPrec,
int mathType,
int inputSize,
int hiddenSize,
int projSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int auxFlags)
mathPrec in cudnnSetRNNDescriptor_v8() specifies compute precision.
|
Copyright © 2024. All rights reserved.