Searched refs:embedding_backward_feature_kernel (Results 1 – 1 of 1) sorted by relevance
48 __global__ void embedding_backward_feature_kernel in embedding_backward_feature_kernel() function278 embedding_backward_feature_kernel<scalar_t, accscalar_t, index_t> in embedding_dense_backward_cuda()