Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreateCTCLossDescriptor(cudnnCTCLossDescriptor ctcLossDesc) |
static int |
JCudnn.cudnnCTCLoss_v8(cudnnHandle handle,
int algo,
cudnnCTCLossDescriptor ctcLossDesc,
cudnnTensorDescriptor probsDesc,
Pointer probs,
Pointer labels,
Pointer labelLengths,
Pointer inputLengths,
Pointer costs,
cudnnTensorDescriptor gradientsDesc,
Pointer gradients,
long workSpaceSizeInBytes,
Pointer workspace)
return the ctc costs and gradients, given the probabilities and labels
|
static int |
JCudnn.cudnnCTCLoss(cudnnHandle handle,
cudnnTensorDescriptor probsDesc,
Pointer probs,
int[] hostLabels,
int[] hostLabelLengths,
int[] hostInputLengths,
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_v8(cudnnCTCLossDescriptor ctcLossDesc,
int[] compType,
int[] normMode,
int[] gradMode,
int[] maxLabelLength) |
static int |
JCudnn.cudnnGetCTCLossDescriptor(cudnnCTCLossDescriptor ctcLossDesc,
int[] compType) |
static int |
JCudnn.cudnnGetCTCLossDescriptorEx(cudnnCTCLossDescriptor ctcLossDesc,
int[] compType,
int[] normMode,
int[] gradMode) |
static int |
JCudnn.cudnnGetCTCLossWorkspaceSize_v8(cudnnHandle handle,
int algo,
cudnnCTCLossDescriptor ctcLossDesc,
cudnnTensorDescriptor probsDesc,
cudnnTensorDescriptor gradientsDesc,
long[] sizeInBytes)
return the workspace size needed for ctc
|
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_v8(cudnnCTCLossDescriptor ctcLossDesc,
int compType,
int normMode,
int gradMode,
int maxLabelLength) |
static int |
JCudnn.cudnnSetCTCLossDescriptor(cudnnCTCLossDescriptor ctcLossDesc,
int compType) |
static int |
JCudnn.cudnnSetCTCLossDescriptorEx(cudnnCTCLossDescriptor ctcLossDesc,
int compType,
int normMode,
int gradMode) |
Copyright © 2020. All rights reserved.