Searched defs:tg_batch_stride (Results 1 – 2 of 2) sorted by relevance
93 int64_t tg_batch_stride = targets.stride(0); in ctc_loss_allocate_outputs() local262 int64_t tg_batch_stride = targets.stride(0); in ctc_loss_backward_cpu_template() local
261 int64_t tg_batch_stride = targets.stride(0); in ctc_loss_gpu_template() local629 int64_t tg_batch_stride = targets.stride(0); in ctc_loss_backward_gpu_template() local