Searched defs:grad_start (Results 1 – 2 of 2) sorted by relevance
299 const scalar_t * const grad_start = grad.const_data_ptr<scalar_t>(); in run_backward_parallel_pdist() local370 const scalar_t * const grad_start = grad.const_data_ptr<scalar_t>(); in run_backward_parallel_cdist() local
32 uint64_t grad_start = res.get()[0].toInt(); in execute_forward_backward() local