| Package | Description |
|---|---|
| org.bytedeco.cuda.global |
| Modifier and Type | Method and Description |
|---|---|
static int |
cudnn.cudnnBuildRNNDynamic(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int miniBatch) |
static int |
cudnn.cudnnCreateRNNDescriptor(cudnnRNNStruct rnnDesc) |
static int |
cudnn.cudnnDestroyRNNDescriptor(cudnnRNNStruct rnnDesc) |
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.cudnnGetRNNTempSpaceSizes(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int fwdMode,
cudnnRNNDataStruct xDesc,
SizeTPointer workSpaceSize,
SizeTPointer reserveSpaceSize) |
static int |
cudnn.cudnnGetRNNWeightParams(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int pseudoLayer,
long weightSpaceSize,
Pointer weightSpace,
int linLayerID,
cudnnTensorStruct mDesc,
Pointer mAddr,
cudnnTensorStruct bDesc,
Pointer bAddr) |
static int |
cudnn.cudnnGetRNNWeightParams(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int pseudoLayer,
long weightSpaceSize,
Pointer weightSpace,
int linLayerID,
cudnnTensorStruct mDesc,
PointerPointer mAddr,
cudnnTensorStruct bDesc,
PointerPointer bAddr) |
static int |
cudnn.cudnnGetRNNWeightSpaceSize(cudnnContext handle,
cudnnRNNStruct rnnDesc,
SizeTPointer weightSpaceSize) |
static int |
cudnn.cudnnRNNBackwardData_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] devSeqLengths,
cudnnRNNDataStruct yDesc,
Pointer y,
Pointer dy,
cudnnRNNDataStruct xDesc,
Pointer dx,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer dhy,
Pointer dhx,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer dcy,
Pointer dcx,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardData_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer devSeqLengths,
cudnnRNNDataStruct yDesc,
Pointer y,
Pointer dy,
cudnnRNNDataStruct xDesc,
Pointer dx,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer dhy,
Pointer dhx,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer dcy,
Pointer dcx,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardData_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer devSeqLengths,
cudnnRNNDataStruct yDesc,
Pointer y,
Pointer dy,
cudnnRNNDataStruct xDesc,
Pointer dx,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer dhy,
Pointer dhx,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer dcy,
Pointer dcx,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardWeights_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int addGrad,
int[] devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hDesc,
Pointer hx,
cudnnRNNDataStruct yDesc,
Pointer y,
long weightSpaceSize,
Pointer dweightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardWeights_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int addGrad,
IntBuffer devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hDesc,
Pointer hx,
cudnnRNNDataStruct yDesc,
Pointer y,
long weightSpaceSize,
Pointer dweightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNBackwardWeights_v8(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int addGrad,
IntPointer devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hDesc,
Pointer hx,
cudnnRNNDataStruct yDesc,
Pointer y,
long weightSpaceSize,
Pointer dweightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNForward(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int fwdMode,
int[] devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer hy,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer cy,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNForward(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int fwdMode,
IntBuffer devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer hy,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer cy,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNForward(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int fwdMode,
IntPointer devSeqLengths,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hDesc,
Pointer hx,
Pointer hy,
cudnnTensorStruct cDesc,
Pointer cx,
Pointer cy,
long weightSpaceSize,
Pointer weightSpace,
long workSpaceSize,
Pointer workSpace,
long reserveSpaceSize,
Pointer reserveSpace) |
static int |
cudnn.cudnnRNNGetClip_v8(cudnnRNNStruct rnnDesc,
int[] clipMode,
int[] clipNanOpt,
double[] lclip,
double[] rclip)
Deprecated.
|
static int |
cudnn.cudnnRNNGetClip_v8(cudnnRNNStruct rnnDesc,
IntBuffer clipMode,
IntBuffer clipNanOpt,
DoubleBuffer lclip,
DoubleBuffer rclip)
Deprecated.
|
static int |
cudnn.cudnnRNNGetClip_v8(cudnnRNNStruct rnnDesc,
IntPointer clipMode,
IntPointer clipNanOpt,
DoublePointer lclip,
DoublePointer rclip)
Deprecated.
|
static int |
cudnn.cudnnRNNGetClip_v9(cudnnRNNStruct rnnDesc,
int[] clipMode,
double[] lclip,
double[] rclip) |
static int |
cudnn.cudnnRNNGetClip_v9(cudnnRNNStruct rnnDesc,
IntBuffer clipMode,
DoubleBuffer lclip,
DoubleBuffer rclip) |
static int |
cudnn.cudnnRNNGetClip_v9(cudnnRNNStruct rnnDesc,
IntPointer clipMode,
DoublePointer lclip,
DoublePointer rclip) |
static int |
cudnn.cudnnRNNSetClip_v8(cudnnRNNStruct rnnDesc,
int clipMode,
int clipNanOpt,
double lclip,
double rclip)
Deprecated.
|
static int |
cudnn.cudnnRNNSetClip_v9(cudnnRNNStruct rnnDesc,
int clipMode,
double lclip,
double rclip) |
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 © 2025. All rights reserved.