Searched defs:gr_batch_stride (Results 1 – 1 of 1) sorted by relevance
458 … int64_t gr_input_stride, int64_t gr_batch_stride, int64_t gr_char_stride, in ctc_loss_backward_collect_nonblank_gpu_kernel()510 … int64_t gr_input_stride, int64_t gr_batch_stride, int64_t gr_char_stride, in ctc_loss_backward_collect_gpu_kernel()580 int64_t gr_batch_stride, in ctc_loss_zero_padded_gradients()