Searched defs:tg_batch_offsets_data (Results 1 – 1 of 1) sorted by relevance
244 auto tg_batch_offsets_data = tg_batch_offsets.mutable_data_ptr<int64_t>(); in ctc_loss_gpu_template() local615 auto tg_batch_offsets_data = tg_batch_offsets.mutable_data_ptr<int64_t>(); in ctc_loss_backward_gpu_template() local