| 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,
jcuda.Pointer dy,
cudnnTensorDescriptor dxdesc,
jcuda.Pointer dx,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnDropoutForward(cudnnHandle handle,
cudnnDropoutDescriptor dropoutDesc,
cudnnTensorDescriptor xdesc,
jcuda.Pointer x,
cudnnTensorDescriptor ydesc,
jcuda.Pointer y,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnSetDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
cudnnHandle handle,
float dropout,
jcuda.Pointer states,
long stateSizeInBytes,
long seed) |
static int |
JCudnn.cudnnSetRNNDescriptor(cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int mode,
int dataType) |
Copyright © 2017. All rights reserved.