Searched defs:numclass (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | ctcloss_impl.cu | 43 int numclass = blank + 1; in CalculateFwdVarKernel() local
102 int numclass = blank + 1; in CalculateBwdVarKernel() local
170 int SOffSet, int maxtime, int numclass, const int *sequence_length, in CTCLossKernel()
221 int batch, int numclass) { in InnerSoftMaxKernel()
304 int numclass, cudaStream_t stream) { in InnerSoftMax()
431 … int SOffSet, int maxtime, int numclass, const int *sequence_length, int *label_squence_length, in CTCLoss()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ctcloss_gpu_kernel.h | 352 int numclass; variable
|