Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreateCTCLossDescriptor(cudnnCTCLossDescriptor ctcLossDesc)
Create an instance of a CTC (Connectionist Temporal Classification) loss descriptor
|
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.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) |
Copyright © 2018. All rights reserved.