Searched refs:embedding_backward_cuda_kernel_unique_by_key (Results 1 – 2 of 2) sorted by relevance
89 int64_t embedding_backward_cuda_kernel_unique_by_key(const Tensor &sorted_indices, Tensor &segment_… in embedding_backward_cuda_kernel_unique_by_key() function108 int64_t embedding_backward_cuda_kernel_unique_by_key<int>(const Tensor &sorted_indices, Tensor &seg…110 int64_t embedding_backward_cuda_kernel_unique_by_key<int64_t>(const Tensor &sorted_indices, Tensor …
209 int64_t embedding_backward_cuda_kernel_unique_by_key(const Tensor &sorted_indices, Tensor &segment_…239 …int64_t num_of_segments = embedding_backward_cuda_kernel_unique_by_key<index_t>(sorted_indices, se… in embedding_backward_cuda_kernel()