Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreateCTCLossDescriptor(cudnnCTCLossDescriptor ctcLossDesc)
CTC (Connectionist Temporal Classification) loss descriptor create/destory/set/get functions
|
static int |
JCudnn.cudnnCTCLoss(cudnnHandle handle,
cudnnTensorDescriptor probsDesc,
Pointer probs,
int[] labels,
int[] labelLengths,
int[] inputLengths,
Pointer costs,
cudnnTensorDescriptor gradientsDesc,
Pointer gradients,
int algo,
cudnnCTCLossDescriptor ctcLossDesc,
Pointer workspace,
long workSpaceSizeInBytes)
return the ctc costs and gradients, given the probabilities and labels
|
static int |
JCudnn.cudnnDestroyCTCLossDescriptor(cudnnCTCLossDescriptor ctcLossDesc) |
static int |
JCudnn.cudnnGetCTCLossDescriptor(cudnnCTCLossDescriptor ctcLossDesc,
int[] compType) |
static int |
JCudnn.cudnnGetCTCLossDescriptorEx(cudnnCTCLossDescriptor ctcLossDesc,
int[] compType,
int[] normMode,
int[] gradMode) |
static int |
JCudnn.cudnnGetCTCLossWorkspaceSize(cudnnHandle handle,
cudnnTensorDescriptor probsDesc,
cudnnTensorDescriptor gradientsDesc,
int[] labels,
int[] labelLengths,
int[] inputLengths,
int algo,
cudnnCTCLossDescriptor ctcLossDesc,
long[] sizeInBytes)
return the workspace size needed for ctc
|
static int |
JCudnn.cudnnSetCTCLossDescriptor(cudnnCTCLossDescriptor ctcLossDesc,
int compType) |
static int |
JCudnn.cudnnSetCTCLossDescriptorEx(cudnnCTCLossDescriptor ctcLossDesc,
int compType,
int normMode,
int gradMode) |
Copyright © 2019. All rights reserved.