Searched defs:max_indices_ (Results 1 – 1 of 1) sorted by relevance
1378 const Tensor &max_indices_, in _embedding_bag_backward()1393 const Tensor &max_indices_, in _embedding_bag_backward_symint()1597 const Tensor& max_indices_, int64_t num_weights, in _embedding_bag_dense_backward_cpu()