| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | LossCTC.cpp | 30 const Tensor& log_probs, in _use_cudnn_ctc_loss()
39 const Tensor& log_probs, in _use_cudnn_ctc_loss_tensor()
48 const Tensor& log_probs, in _cudnn_ctc_loss()
59 const Tensor& log_probs, in _cudnn_ctc_loss_tensor()
90 const Tensor& log_probs, in _use_cudnn_ctc_loss()
121 const Tensor& log_probs, in _use_cudnn_ctc_loss_tensor()
174 const TensorArg log_probs{log_probs_t, "log_probs", 1}; in _cudnn_ctc_loss() local
262 const TensorArg log_probs{log_probs_t, "log_probs", 1}; in _cudnn_ctc_loss_tensor() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossCTC.cpp | 56 … Tensor, size_t, std::vector<int64_t>> ctc_loss_allocate_outputs(const Tensor& log_probs, const Te… in ctc_loss_allocate_outputs()
129 std::tuple<Tensor, Tensor> ctc_loss_cpu_template(const Tensor& log_probs, const Tensor& targets, In… in ctc_loss_cpu_template()
233 Tensor ctc_loss_backward_cpu_template(const Tensor& grad_out, const Tensor& log_probs, const Tensor… in ctc_loss_backward_cpu_template()
407 std::tuple<Tensor, Tensor> ctc_loss_meta(const Tensor& log_probs, const Tensor& targets, IntArrayRe… in ctc_loss_meta()
423 std::tuple<Tensor, Tensor> ctc_loss_cpu(const Tensor& log_probs, const Tensor& targets, IntArrayRef… in ctc_loss_cpu()
435 std::tuple<Tensor, Tensor> ctc_loss_tensor(const Tensor& log_probs, const Tensor& targets, const Te… in ctc_loss_tensor()
447 Tensor ctc_loss_backward_cpu(const Tensor& grad, const Tensor& log_probs, const Tensor& targets, In… in ctc_loss_backward_cpu()
460 const Tensor& log_probs, in ctc_loss_backward_tensor()
500 Tensor log_probs = is_batched ? log_probs_ : log_probs_.unsqueeze(1); in ctc_loss_impl() local
541 Tensor ctc_loss(const Tensor& log_probs, const Tensor& targets, const Tensor& input_lengths, const … in ctc_loss()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LossCTC.cu | 221 std::tuple<Tensor, Tensor> ctc_loss_gpu_template(const Tensor& log_probs, const Tensor& targets, In… in ctc_loss_gpu_template()
605 Tensor ctc_loss_backward_gpu_template(const Tensor& grad_out, const Tensor& log_probs, const Tensor… in ctc_loss_backward_gpu_template()
778 std::tuple<Tensor, Tensor> ctc_loss_gpu(const Tensor& log_probs, const Tensor& targets, IntArrayRef… in ctc_loss_gpu()
789 Tensor ctc_loss_backward_gpu(const Tensor& grad, const Tensor& log_probs, const Tensor& targets, In… in ctc_loss_backward_gpu()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ctc/ |
| H A D | ctc_beam_search.h | 403 int n, std::vector<std::vector<int>>* paths, std::vector<T>* log_probs, in TopPaths()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ctc/ |
| H A D | ctc_beam_search.h | 398 int n, std::vector<std::vector<int>>* paths, std::vector<float>* log_probs, in TopPaths()
|
| H A D | ctc_beam_search_decoder.cc | 225 std::vector<float> log_probs; in Eval() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | loss.h | 737 const Tensor& log_probs, in ctc_loss()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ctc_decoder_ops.cc | 340 std::vector<T> log_probs; in Compute() local
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | functional.cpp | 2574 const auto log_probs = in TEST_F() local
2591 const auto log_probs = in TEST_F() local
2603 const auto log_probs = in TEST_F() local
2619 const auto log_probs = in TEST_F() local
|
| H A D | modules.cpp | 3197 const auto log_probs = in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | loss.cpp | 303 const Tensor& log_probs, in forward()
|
| /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
|