| Modifier and Type | Method and Description |
|---|---|
static int |
JCudnn.cudnnCreateRNNDescriptor(cudnnRNNDescriptor rnnDesc) |
static int |
JCudnn.cudnnDestroyRNNDescriptor(cudnnRNNDescriptor rnnDesc) |
static int |
JCudnn.cudnnGetRNNLinLayerBiasParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int layer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerBiasDesc,
jcuda.Pointer linLayerBias) |
static int |
JCudnn.cudnnGetRNNLinLayerMatrixParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int layer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerMatDesc,
jcuda.Pointer linLayerMat) |
static int |
JCudnn.cudnnGetRNNParamsSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes,
int dataType) |
static int |
JCudnn.cudnnGetRNNTrainingReserveSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes) |
static int |
JCudnn.cudnnGetRNNWorkspaceSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes) |
static int |
JCudnn.cudnnRNNBackwardData(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor dhyDesc,
jcuda.Pointer dhy,
cudnnTensorDescriptor dcyDesc,
jcuda.Pointer dcy,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor hxDesc,
jcuda.Pointer hx,
cudnnTensorDescriptor cxDesc,
jcuda.Pointer cx,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx,
cudnnTensorDescriptor dhxDesc,
jcuda.Pointer dhx,
cudnnTensorDescriptor dcxDesc,
jcuda.Pointer dcx,
jcuda.Pointer workspace,
long workSpaceSizeInBytes,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnRNNBackwardWeights(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor hxDesc,
jcuda.Pointer hx,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterDescriptor dwDesc,
jcuda.Pointer dw,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnRNNForwardInference(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor hxDesc,
jcuda.Pointer hx,
cudnnTensorDescriptor cxDesc,
jcuda.Pointer cx,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor hyDesc,
jcuda.Pointer hy,
cudnnTensorDescriptor cyDesc,
jcuda.Pointer cy,
jcuda.Pointer workspace,
long workSpaceSizeInBytes) |
static int |
JCudnn.cudnnRNNForwardTraining(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor hxDesc,
jcuda.Pointer hx,
cudnnTensorDescriptor cxDesc,
jcuda.Pointer cx,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor hyDesc,
jcuda.Pointer hy,
cudnnTensorDescriptor cyDesc,
jcuda.Pointer cy,
jcuda.Pointer workspace,
long workSpaceSizeInBytes,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
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.