Package | Description |
---|---|
org.bytedeco.cuda.global |
Modifier and Type | Method and Description |
---|---|
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_v6(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] hiddenSize,
int[] numLayers,
cudnnDropoutStruct dropoutDesc,
int[] inputMode,
int[] direction,
int[] cellMode,
int[] algo,
int[] mathPrec)
Deprecated.
|
static int |
cudnn.cudnnGetRNNDescriptor_v6(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer hiddenSize,
IntBuffer numLayers,
cudnnDropoutStruct dropoutDesc,
IntBuffer inputMode,
IntBuffer direction,
IntBuffer cellMode,
IntBuffer algo,
IntBuffer mathPrec)
Deprecated.
|
static int |
cudnn.cudnnGetRNNDescriptor_v6(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer hiddenSize,
IntPointer numLayers,
cudnnDropoutStruct dropoutDesc,
IntPointer inputMode,
IntPointer direction,
IntPointer cellMode,
IntPointer algo,
IntPointer mathPrec)
Deprecated.
|
static int |
cudnn.cudnnGetRNNDescriptor_v8(cudnnRNNStruct 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,
cudnnDropoutStruct dropoutDesc,
int[] auxFlags) |
static int |
cudnn.cudnnGetRNNDescriptor_v8(cudnnRNNStruct rnnDesc,
IntBuffer algo,
IntBuffer cellMode,
IntBuffer biasMode,
IntBuffer dirMode,
IntBuffer inputMode,
IntBuffer dataType,
IntBuffer mathPrec,
IntBuffer mathType,
IntBuffer inputSize,
IntBuffer hiddenSize,
IntBuffer projSize,
IntBuffer numLayers,
cudnnDropoutStruct dropoutDesc,
IntBuffer auxFlags) |
static int |
cudnn.cudnnGetRNNDescriptor_v8(cudnnRNNStruct rnnDesc,
IntPointer algo,
IntPointer cellMode,
IntPointer biasMode,
IntPointer dirMode,
IntPointer inputMode,
IntPointer dataType,
IntPointer mathPrec,
IntPointer mathType,
IntPointer inputSize,
IntPointer hiddenSize,
IntPointer projSize,
IntPointer numLayers,
cudnnDropoutStruct dropoutDesc,
IntPointer auxFlags) |
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_v6(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutStruct dropoutDesc,
int inputMode,
int direction,
int cellMode,
int algo,
int mathPrec)
Deprecated.
|
static int |
cudnn.cudnnSetRNNDescriptor_v8(cudnnRNNStruct 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,
cudnnDropoutStruct dropoutDesc,
int auxFlags) |
Copyright © 2024. All rights reserved.