Searched refs:embedding_dense_backward_cuda_scan (Results 1 – 3 of 3) sorted by relevance
47 void embedding_dense_backward_cuda_scan(Tensor &sorted_indices, Tensor &count) { in embedding_dense_backward_cuda_scan() function82 void embedding_dense_backward_cuda_scan<int>(Tensor &sorted_indices, Tensor &count);84 void embedding_dense_backward_cuda_scan<int64_t>(Tensor &sorted_indices, Tensor &count);
245 void embedding_dense_backward_cuda_scan(Tensor &sorted_indices, Tensor &count);338 embedding_dense_backward_cuda_scan<index_t>(sorted_indices, count); in embedding_dense_backward_cuda()
42 void embedding_dense_backward_cuda_scan(Tensor &sorted_indices, Tensor &count);230 embedding_dense_backward_cuda_scan<index_t>(sorted_indices, count); in embedding_bag_backward_cuda_sum_avg()