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.cudnnGetDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
cudnnHandle handle,
float[] dropout,
Pointer states,
long[] seed) |
static int |
JCudnn.cudnnGetRNNDescriptor(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int[] hiddenSize,
int[] numLayers,
cudnnDropoutDescriptor dropoutDesc,
int[] inputMode,
int[] direction,
int[] mode,
int[] algo,
int[] dataType) |
static int |
JCudnn.cudnnRestoreDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
cudnnHandle handle,
float dropout,
Pointer states,
long stateSizeInBytes,
long seed) |
static int |
JCudnn.cudnnSetDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
cudnnHandle handle,
float dropout,
Pointer states,
long stateSizeInBytes,
long seed) |
static int |
JCudnn.cudnnSetRNNDescriptor_v5(cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int mode,
int dataType) |
static int |
JCudnn.cudnnSetRNNDescriptor_v6(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int mode,
int algo,
int dataType)
DEPRECATED routines to be removed next release :
User should use the non-suffixed version (which has the API and functionality of _v6 version)
Routines with _v5 suffix has the functionality of the non-suffixed routines in the CUDNN V6
|
static int |
JCudnn.cudnnSetRNNDescriptor(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int mode,
int algo,
int dataType) |
Copyright © 2018. All rights reserved.