Searched defs:tg_target_stride (Results 1 – 2 of 2) sorted by relevance
84 … const int64_t* __restrict__ tg_batch_offsets, int64_t tg_target_stride, in ctc_loss_log_alpha_gpu_kernel()240 int64_t tg_target_stride; in ctc_loss_gpu_template() local327 … const int64_t* __restrict__ tg_batch_offsets, int64_t tg_target_stride, in ctc_loss_backward_log_beta_gpu_kernel()462 … const int64_t* __restrict__ tg_batch_offsets, int64_t tg_target_stride, in ctc_loss_backward_collect_nonblank_gpu_kernel()514 … const int64_t* __restrict__ tg_batch_offsets, int64_t tg_target_stride, in ctc_loss_backward_collect_gpu_kernel()611 int64_t tg_target_stride; in ctc_loss_backward_gpu_template() local
74 size_t tg_target_stride; in ctc_loss_allocate_outputs() local136 size_t tg_target_stride; in ctc_loss_cpu_template() local244 int64_t tg_target_stride; in ctc_loss_backward_cpu_template() local