Searched defs:current_target_prime (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossCTC.cpp | 181 …auto current_target_prime = get_target_prime(targets_data, tg_batch_offset, tg_target_stride, s, B… in ctc_loss_cpu_template() local
328 …auto current_target_prime = get_target_prime(targets_data, tg_batch_offset, tg_target_stride, 2*ta… in ctc_loss_backward_cpu_template() local
345 …auto current_target_prime = get_target_prime(targets_data, tg_batch_offset, tg_target_stride, s, B… in ctc_loss_backward_cpu_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LossCTC.cu | 352 int64_t current_target_prime = get_target_prime( in ctc_loss_backward_log_beta_gpu_kernel() local
370 int64_t current_target_prime; in ctc_loss_backward_log_beta_gpu_kernel() local
535 int64_t current_target_prime = get_target_prime( in ctc_loss_backward_collect_gpu_kernel() local
|