Searched defs:zero_infinity (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossCTC.cpp | 234 … const Tensor& neg_log_likelihood, const Tensor& log_alpha, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_cpu_template()
407 …argets, IntArrayRef input_lengths, IntArrayRef target_lengths, int64_t BLANK, bool zero_infinity) { in ctc_loss_meta()
423 …argets, IntArrayRef input_lengths, IntArrayRef target_lengths, int64_t BLANK, bool zero_infinity) { in ctc_loss_cpu()
435 …ts, const Tensor& input_lengths, const Tensor& target_lengths, int64_t BLANK, bool zero_infinity) { in ctc_loss_tensor()
448 … const Tensor& neg_log_likelihood, const Tensor& log_alpha, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_cpu()
467 bool zero_infinity) { in ctc_loss_backward_tensor()
498 … input_lengths, LengthsType target_lengths, int64_t BLANK, int64_t reduction, bool zero_infinity) { in ctc_loss_impl()
536 … input_lengths, IntArrayRef target_lengths, int64_t BLANK, int64_t reduction, bool zero_infinity) { in ctc_loss()
541 …nput_lengths, const Tensor& target_lengths, int64_t BLANK, int64_t reduction, bool zero_infinity) { in ctc_loss()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | LossCTC.cpp | 54 bool zero_infinity) { in _cudnn_ctc_loss()
65 bool zero_infinity) { in _cudnn_ctc_loss_tensor()
171 bool zero_infinity) { in _cudnn_ctc_loss()
256 bool zero_infinity) { in _cudnn_ctc_loss_tensor()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LossCTC.cu | 463 int64_t batch_size, bool zero_infinity) { in ctc_loss_backward_collect_nonblank_gpu_kernel()
515 … int64_t batch_size, int64_t num_labels, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_collect_gpu_kernel()
606 … const Tensor& neg_log_likelihood, const Tensor& log_alpha, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_gpu_template()
778 …argets, IntArrayRef input_lengths, IntArrayRef target_lengths, int64_t BLANK, bool zero_infinity) { in ctc_loss_gpu()
790 … const Tensor& neg_log_likelihood, const Tensor& log_alpha, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_gpu()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | loss.h | 743 bool zero_infinity) { in ctc_loss()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 10695 def _helper(zero_infinity): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 5130 bool zero_infinity) { in _cudnn_ctc_loss_backward()
|