Searched defs:per_sample_weights_opt (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | EmbeddingBag.cpp | 1229 const int64_t mode, bool sparse, const std::optional<Tensor>& per_sample_weights_opt, in embedding_bag()
1261 const int64_t mode, bool sparse, const std::optional<Tensor>& per_sample_weights_opt, in embedding_bag()
1272 …const int64_t mode, bool sparse, const std::optional<Tensor>& per_sample_weights_opt, bool include… in _embedding_bag_forward_only_cpu()
1295 …const int64_t mode, bool sparse, const std::optional<Tensor>& per_sample_weights_opt, bool include… in _embedding_bag_cpu()
1381 bool sparse, const std::optional<Tensor>& per_sample_weights_opt, in _embedding_bag_backward()
1396 bool sparse, const std::optional<Tensor>& per_sample_weights_opt, in _embedding_bag_backward_symint()
1753 bool scale_grad_by_freq, int64_t mode, const std::optional<Tensor>& per_sample_weights_opt, in _embedding_bag_sparse_backward_symint()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | EmbeddingBag.cu | 312 … const int64_t mode, bool sparse, const std::optional<Tensor>& per_sample_weights_opt, in _embedding_bag_forward_only_cuda()
335 … const int64_t mode, bool sparse, const std::optional<Tensor>& per_sample_weights_opt, in _embedding_bag_cuda()
432 … bool scale_grad_by_freq, int64_t mode, const std::optional<Tensor>& per_sample_weights_opt, in _embedding_bag_dense_backward_cuda()
|