Searched refs:embedding_backward_cuda_kernel (Results 1 – 4 of 4) sorted by relevance
10 Tensor embedding_backward_cuda_kernel(
212 Tensor embedding_backward_cuda_kernel( in embedding_backward_cuda_kernel() function
343 return embedding_backward_cuda_kernel(grad, orig_indices, in embedding_dense_backward_cuda()
234 return embedding_backward_cuda_kernel(grad, orig_indices, sorted_indices, in embedding_bag_backward_cuda_sum_avg()