Searched defs:log_alpha (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossCTC.cpp | 118 …Tensor log_alpha = at::empty({batch_size, log_probs.size(0), 2*max_target_length+1}, log_probs.opt… in ctc_loss_allocate_outputs() local
135 Tensor neg_log_likelihood, log_alpha; in ctc_loss_cpu_template() local
234 … const Tensor& neg_log_likelihood, const Tensor& log_alpha, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_cpu_template()
411 Tensor neg_log_likelihood, log_alpha; in ctc_loss_meta() local
448 … const Tensor& neg_log_likelihood, const Tensor& log_alpha, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_cpu()
465 const Tensor& log_alpha, in ctc_loss_backward_tensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ctc/ |
| H A D | ctc_loss_calculator.h | 375 Matrix* log_alpha) const { in CalculateForwardVariables()
490 const Matrix& log_alpha, in CalculateGradient()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LossCTC.cu | 290 …Tensor log_alpha = at::empty({batch_size, log_probs.size(0), 2*max_target_length+1}, log_probs.opt… in ctc_loss_gpu_template() local
606 … const Tensor& neg_log_likelihood, const Tensor& log_alpha, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_gpu_template()
790 … const Tensor& neg_log_likelihood, const Tensor& log_alpha, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_gpu()
|