Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreateDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc) |
static int |
JCudnn.cudnnDestroyDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc) |
static int |
JCudnn.cudnnDropoutBackward(cudnnHandle handle,
cudnnDropoutDescriptor dropoutDesc,
cudnnTensorDescriptor dydesc,
Pointer dy,
cudnnTensorDescriptor dxdesc,
Pointer dx,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnDropoutForward(cudnnHandle handle,
cudnnDropoutDescriptor dropoutDesc,
cudnnTensorDescriptor xdesc,
Pointer x,
cudnnTensorDescriptor ydesc,
Pointer y,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnGetAttnDescriptor(cudnnAttnDescriptor attnDesc,
int[] attnMode,
int[] nHeads,
double[] smScaler,
int[] dataType,
int[] computePrec,
int[] mathType,
cudnnDropoutDescriptor attnDropoutDesc,
cudnnDropoutDescriptor 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 |
JCudnn.cudnnGetDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
cudnnHandle handle,
float[] dropout,
Pointer states,
long[] seed) |
static int |
JCudnn.cudnnGetRNNDescriptor_v6(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] hiddenSize,
int[] numLayers,
cudnnDropoutDescriptor dropoutDesc,
int[] inputMode,
int[] direction,
int[] cellMode,
int[] algo,
int[] mathPrec)
Deprecated.
|
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.cudnnRestoreDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
cudnnHandle handle,
float dropout,
Pointer states,
long stateSizeInBytes,
long seed)
Restores the dropout descriptor to a previously saved-off state
|
static int |
JCudnn.cudnnSetAttnDescriptor(cudnnAttnDescriptor attnDesc,
int attnMode,
int nHeads,
double smScaler,
int dataType,
int computePrec,
int mathType,
cudnnDropoutDescriptor attnDropoutDesc,
cudnnDropoutDescriptor 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 |
JCudnn.cudnnSetDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
cudnnHandle handle,
float dropout,
Pointer states,
long stateSizeInBytes,
long seed) |
static int |
JCudnn.cudnnSetRNNDescriptor_v6(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int cellMode,
int algo,
int mathPrec)
Deprecated.
|
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) |
Copyright © 2020. All rights reserved.