Home
last modified time | relevance | path

Searched refs:embedding_backward_cuda_kernel (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DEmbeddingBackwardKernel.cuh10 Tensor embedding_backward_cuda_kernel(
H A DEmbedding.cu343 return embedding_backward_cuda_kernel(grad, orig_indices, in embedding_dense_backward_cuda()
H A DEmbeddingBackwardKernel.cu212 Tensor embedding_backward_cuda_kernel( in embedding_backward_cuda_kernel() function
H A DEmbeddingBag.cu234 return embedding_backward_cuda_kernel(grad, orig_indices, sorted_indices, in embedding_bag_backward_cuda_sum_avg()