| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | LossCTC.cpp | 32 IntArrayRef input_lengths, in _use_cudnn_ctc_loss()
41 const Tensor& input_lengths, in _use_cudnn_ctc_loss_tensor()
50 IntArrayRef input_lengths, in _cudnn_ctc_loss()
61 const Tensor& input_lengths, in _cudnn_ctc_loss_tensor()
92 IntArrayRef input_lengths, in _use_cudnn_ctc_loss()
123 const Tensor& input_lengths, in _use_cudnn_ctc_loss_tensor()
191 std::vector<int> input_lengths(input_lengths_.begin(), input_lengths_.end()); in _cudnn_ctc_loss() local
252 const Tensor& input_lengths, in _cudnn_ctc_loss_tensor()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossCTC.cpp | 56 …llocate_outputs(const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_allocate_outputs()
129 …ss_cpu_template(const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_cpu_template()
233 …nsor& grad_out, const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_backward_cpu_template()
407 …> ctc_loss_meta(const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_meta()
423 …r> ctc_loss_cpu(const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_cpu()
435 …tc_loss_tensor(const Tensor& log_probs, const Tensor& targets, const Tensor& input_lengths, const … in ctc_loss_tensor()
447 …t Tensor& grad, const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_backward_cpu()
462 const Tensor& input_lengths, in ctc_loss_backward_tensor()
498 Tensor ctc_loss_impl(const Tensor& log_probs_, const Tensor& targets, LengthsType input_lengths, Le… in ctc_loss_impl()
536 Tensor ctc_loss(const Tensor& log_probs_, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LossCTC.cu | 79 …const scalar_t*log_probs_data, const int64_t* __restrict__ input_lengths, int64_t max_input_length, in ctc_loss_log_alpha_gpu_kernel()
221 …ss_gpu_template(const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_gpu_template()
323 …const scalar_t*log_probs_data, const int64_t* __restrict__ input_lengths, int64_t max_input_length, in ctc_loss_backward_log_beta_gpu_kernel()
455 … const scalar_t*log_probs_data, const int64_t* __restrict__ input_lengths, in ctc_loss_backward_collect_nonblank_gpu_kernel()
507 …const scalar_t*log_probs_data, const int64_t* __restrict__ input_lengths, int64_t max_input_length, in ctc_loss_backward_collect_gpu_kernel()
578 const int64_t* __restrict__ input_lengths, /* (B, ) layout */ in ctc_loss_zero_padded_gradients()
605 …nsor& grad_out, const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_backward_gpu_template()
778 …r> ctc_loss_gpu(const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_gpu()
789 …t Tensor& grad, const Tensor& log_probs, const Tensor& targets, IntArrayRef input_lengths, IntArra… in ctc_loss_backward_gpu()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | loss.h | 739 const Tensor& input_lengths, in ctc_loss()
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | functional.cpp | 2571 const auto input_lengths = torch::tensor({50, 50, 50}); in TEST_F() local
2589 const auto input_lengths = torch::tensor({50, 50, 50}); in TEST_F() local
2600 const auto input_lengths = torch::tensor({50, 50, 50}); in TEST_F() local
2617 const auto input_lengths = torch::tensor({50, 50, 50}); in TEST_F() local
|
| H A D | modules.cpp | 3194 const auto input_lengths = torch::tensor({50, 50, 50}); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | loss.cpp | 305 const Tensor& input_lengths, in forward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_evaluator.cc | 1639 const auto input_lengths = GetDimensionLengths(input_literal); in ComputeFft() local
1987 const absl::Span<const int64_t> input_lengths, in CopyDataFromInput()
2077 const absl::Span<const int64_t> input_lengths, in CopyDataFromInput()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_nn.py | 3020 def ctcloss_reference(log_probs, targets, input_lengths, target_lengths, blank=0, reduction='mean'): argument
|