static int |
cudnn.cudnnCreateDropoutDescriptor(cudnnDropoutStruct dropoutDesc) |
static int |
cudnn.cudnnDestroyDropoutDescriptor(cudnnDropoutStruct dropoutDesc) |
static int |
cudnn.cudnnDropoutBackward(cudnnContext handle,
cudnnDropoutStruct dropoutDesc,
cudnnTensorStruct dydesc,
Pointer dy,
cudnnTensorStruct dxdesc,
Pointer dx,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnn.cudnnDropoutForward(cudnnContext handle,
cudnnDropoutStruct dropoutDesc,
cudnnTensorStruct xdesc,
Pointer x,
cudnnTensorStruct ydesc,
Pointer y,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnn.cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc,
int[] attnMode,
int[] nHeads,
double[] smScaler,
int[] dataType,
int[] computePrec,
int[] mathType,
cudnnDropoutStruct attnDropoutDesc,
cudnnDropoutStruct postDropoutDesc,
int[] qSize,
int[] kSize,
int[] vSize,
int[] qProjSize,
int[] kProjSize,
int[] vProjSize,
int[] oProjSize,
int[] qoMaxSeqLength,
int[] kvMaxSeqLength,
int[] maxBatchSize,
int[] maxBeamSize) |
static int |
cudnn.cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc,
IntBuffer attnMode,
IntBuffer nHeads,
DoubleBuffer smScaler,
IntBuffer dataType,
IntBuffer computePrec,
IntBuffer mathType,
cudnnDropoutStruct attnDropoutDesc,
cudnnDropoutStruct postDropoutDesc,
IntBuffer qSize,
IntBuffer kSize,
IntBuffer vSize,
IntBuffer qProjSize,
IntBuffer kProjSize,
IntBuffer vProjSize,
IntBuffer oProjSize,
IntBuffer qoMaxSeqLength,
IntBuffer kvMaxSeqLength,
IntBuffer maxBatchSize,
IntBuffer maxBeamSize) |
static int |
cudnn.cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc,
IntPointer attnMode,
IntPointer nHeads,
DoublePointer smScaler,
IntPointer dataType,
IntPointer computePrec,
IntPointer mathType,
cudnnDropoutStruct attnDropoutDesc,
cudnnDropoutStruct postDropoutDesc,
IntPointer qSize,
IntPointer kSize,
IntPointer vSize,
IntPointer qProjSize,
IntPointer kProjSize,
IntPointer vProjSize,
IntPointer oProjSize,
IntPointer qoMaxSeqLength,
IntPointer kvMaxSeqLength,
IntPointer maxBatchSize,
IntPointer maxBeamSize) |
static int |
cudnn.cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
float[] dropout,
Pointer states,
long[] seed) |
static int |
cudnn.cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
FloatBuffer dropout,
Pointer states,
LongBuffer seed) |
static int |
cudnn.cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
FloatPointer dropout,
Pointer states,
LongPointer seed) |
static int |
cudnn.cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
FloatPointer dropout,
PointerPointer states,
LongPointer seed) |
static int |
cudnn.cudnnGetRNNDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] hiddenSize,
int[] numLayers,
cudnnDropoutStruct dropoutDesc,
int[] inputMode,
int[] direction,
int[] mode,
int[] algo,
int[] mathPrec) |
static int |
cudnn.cudnnGetRNNDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer hiddenSize,
IntBuffer numLayers,
cudnnDropoutStruct dropoutDesc,
IntBuffer inputMode,
IntBuffer direction,
IntBuffer mode,
IntBuffer algo,
IntBuffer mathPrec) |
static int |
cudnn.cudnnGetRNNDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer hiddenSize,
IntPointer numLayers,
cudnnDropoutStruct dropoutDesc,
IntPointer inputMode,
IntPointer direction,
IntPointer mode,
IntPointer algo,
IntPointer mathPrec) |
static int |
cudnn.cudnnRestoreDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
float dropout,
Pointer states,
long stateSizeInBytes,
long seed) |
static int |
cudnn.cudnnSetAttnDescriptor(cudnnAttnStruct attnDesc,
int attnMode,
int nHeads,
double smScaler,
int dataType,
int computePrec,
int mathType,
cudnnDropoutStruct attnDropoutDesc,
cudnnDropoutStruct postDropoutDesc,
int qSize,
int kSize,
int vSize,
int qProjSize,
int kProjSize,
int vProjSize,
int oProjSize,
int qoMaxSeqLength,
int kvMaxSeqLength,
int maxBatchSize,
int maxBeamSize) |
static int |
cudnn.cudnnSetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
float dropout,
Pointer states,
long stateSizeInBytes,
long seed) |
static int |
cudnn.cudnnSetRNNDescriptor_v5(cudnnRNNStruct rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutStruct dropoutDesc,
int inputMode,
int direction,
int mode,
int mathPrec) |
static int |
cudnn.cudnnSetRNNDescriptor_v6(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutStruct dropoutDesc,
int inputMode,
int direction,
int mode,
int algo,
int mathPrec) |
static int |
cudnn.cudnnSetRNNDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutStruct dropoutDesc,
int inputMode,
int direction,
int mode,
int algo,
int mathPrec) |