Searched defs:log_beta_b (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ctcloss_gpu_kernel.h | 304 …void MemManageForCus(T **log_alpha_b, T **log_beta_b, int **label_value_with_blank, int *cum_label… in MemManageForCus()
320 void FreeMem(int *label_value_with_blank, T *log_alpha_b, T *log_beta_b) { in FreeMem()
360 T *log_beta_b; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | ctcloss_cpu_kernel.cc | 214 … const std::vector<std::vector<TT>> &log_beta_b, const TT log_pzx, in CalculateGrad()
338 std::vector<std::vector<T>> log_beta_b; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | ctcloss_impl.cu | 89 __global__ void CalculateBwdVarKernel(T *log_beta_b, int *label_value_with_blank, T *softmax_probs, in CalculateBwdVarKernel()
169 __global__ void CTCLossKernel(T *log_alpha_b, T *log_beta_b, T *softmax_probs, int *label_value_wit… in CTCLossKernel()
277 cudaError_t CalculateBwdVar(T *log_beta_b, int *label_value_with_blank, T *softmax_probs, const int… in CalculateBwdVar()
430 cudaError_t CTCLoss(T *log_alpha_b, T *log_beta_b, T *softmax_probs, int *label_value_with_blank, i… in CTCLoss()
|