Searched defs:prob_num (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 | 156 __global__ void ProbInitKernel(T *prob_num, int size) { in ProbInitKernel()
171 … int *label_squence_length, int *cum_labels_length, T *cost, T *grads, T *prob_num, in CTCLossKernel()
432 … int *cum_labels_length, T *cost, T *grads, T *prob_num, bool ignore_longer_outputs_than_inputs, in CTCLoss()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ctcloss_gpu_kernel.h | 349 T *prob_num; variable
|