Searched defs:log_alpha_b (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | ctcloss_gpu_kernel.h | 254 …void MemManageForCus(T **log_alpha_b, T **log_beta_b, int **label_value_with_blank, int *cum_label… in MemManageForCus()
273 void FreeMem(int *label_value_with_blank, T *log_alpha_b, T *log_beta_b) { in FreeMem()
315 T *log_alpha_b; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | ctcloss_cpu_kernel.cc | 198 const std::vector<std::vector<TT>> &log_alpha_b, in CalculateGrad()
313 std::vector<std::vector<T>> log_alpha_b; in LaunchKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | ctcloss_impl.cu | 31 __global__ void CalculateFwdVarKernel(T *log_alpha_b, int *label_value_with_blank, T *softmax_probs, in CalculateFwdVarKernel()
170 __global__ void CTCLossKernel(T *log_alpha_b, T *log_beta_b, T *softmax_probs, int *label_value_wit… in CTCLossKernel()
289 void CalculateFwdVar(T *log_alpha_b, int *label_value_with_blank, T *softmax_probs, const int *sequ… in CalculateFwdVar()
420 void CTCLoss(T *log_alpha_b, T *log_beta_b, T *softmax_probs, int *label_value_with_blank, int batc… in CTCLoss()
|