Searched defs:targets_data (Results 1 – 2 of 2) sorted by relevance
80 …const target_t* __restrict__ targets_data, const int64_t* __restrict__ target_lengths, int64_t max… in ctc_loss_log_alpha_gpu_kernel()324 …const target_t* __restrict__ targets_data, const int64_t* __restrict__ target_lengths, int64_t max… in ctc_loss_backward_log_beta_gpu_kernel()456 … const target_t* __restrict__ targets_data, const int64_t* __restrict__ target_lengths, in ctc_loss_backward_collect_nonblank_gpu_kernel()508 …const target_t* __restrict__ targets_data, const int64_t* __restrict__ target_lengths, int64_t max… in ctc_loss_backward_collect_gpu_kernel()
153 auto targets_data = targets.const_data_ptr<target_t>(); in ctc_loss_cpu_template() local277 auto targets_data = targets.const_data_ptr<target_t>(); in ctc_loss_backward_cpu_template() local