Searched defs:max_target_length (Results 1 – 3 of 3) sorted by relevance
80 …* __restrict__ targets_data, const int64_t* __restrict__ target_lengths, int64_t max_target_length, in ctc_loss_log_alpha_gpu_kernel()242 int64_t max_target_length = 0; in ctc_loss_gpu_template() local324 …* __restrict__ targets_data, const int64_t* __restrict__ target_lengths, int64_t max_target_length, in ctc_loss_backward_log_beta_gpu_kernel()508 …* __restrict__ targets_data, const int64_t* __restrict__ target_lengths, int64_t max_target_length, in ctc_loss_backward_collect_gpu_kernel()613 int64_t max_target_length; in ctc_loss_backward_gpu_template() local
58 max_target_length = 10 variable
75 int64_t max_target_length = 0; in ctc_loss_allocate_outputs() local246 int64_t max_target_length; in ctc_loss_backward_cpu_template() local