Home
last modified time | relevance | path

Searched defs:CalculateFwdVar (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dctcloss_cpu_kernel.cc131 void CTCLossCpuKernelMod::CalculateFwdVar(const std::vector<uint32_t> &label_with_blank, in CalculateFwdVar() function in mindspore::kernel::CTCLossCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dctcloss_impl.cu290 cudaError_t CalculateFwdVar(T *log_alpha_b, int *label_value_with_blank, T *softmax_probs, const in… in CalculateFwdVar() function