| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Embedding.cpp | 38 c10::SymInt padding_idx, bool scale_grad_by_freq, bool sparse) { in embedding_symint()
58 c10::SymInt padding_idx, bool scale_grad_by_freq, bool sparse) { in embedding_backward_symint()
77 int64_t padding_idx, bool scale_grad_by_freq) { in embedding_sparse_backward()
114 int64_t padding_idx, bool scale_grad_by_freq) { in embedding_dense_backward_cpu()
|
| H A D | EmbeddingBag.cpp | 1228 const Tensor &offsets, const bool scale_grad_by_freq, in embedding_bag()
1260 const Tensor &offsets, const bool scale_grad_by_freq, in embedding_bag()
1271 const Tensor &offsets, const bool scale_grad_by_freq, in _embedding_bag_forward_only_cpu()
1294 const Tensor &offsets, const bool scale_grad_by_freq, in _embedding_bag_cpu()
1380 bool scale_grad_by_freq, int64_t mode, in _embedding_bag_backward()
1395 bool scale_grad_by_freq, int64_t mode, in _embedding_bag_backward_symint()
1510 bool scale_grad_by_freq, in _embedding_bag_dense_backward_cpu_sum_mean()
1598 … bool scale_grad_by_freq, int64_t mode, const std::optional<Tensor>& per_sample_weights__opt, in _embedding_bag_dense_backward_cpu()
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 | 174 bool scale_grad_by_freq, int64_t mode, in embedding_bag_backward_cuda_sum_avg()
311 const Tensor &offsets, const bool scale_grad_by_freq, in _embedding_bag_forward_only_cuda()
334 const Tensor &offsets_, const bool scale_grad_by_freq, 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()
|
| H A D | Embedding.cu | 250 bool scale_grad_by_freq) { in embedding_dense_backward_cuda()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | embedding.h | 30 bool scale_grad_by_freq, in embedding()
95 bool scale_grad_by_freq, in embedding_bag()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/qat/modules/ |
| H A D | embedding_ops.py | 35 scale_grad_by_freq=False, argument
148 scale_grad_by_freq=False, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 26 c10::SymInt padding_idx, bool scale_grad_by_freq, bool sparse) { in embedding_batch_rule()
57 c10::SymInt num_weights, c10::SymInt padding_idx, bool scale_grad_by_freq) { in embedding_dense_backward_batch_rule()
|
| /aosp_15_r20/external/executorch/backends/cadence/reference/operators/ |
| H A D | op_embedding.cpp | 23 bool scale_grad_by_freq, in embedding_out()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_embedding.cpp | 78 bool scale_grad_by_freq, in embedding_out()
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | sparse.py | 222 scale_grad_by_freq=False, argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset18.py | 234 scale_grad_by_freq, argument
|
| H A D | symbolic_opset10.py | 595 scale_grad_by_freq, argument
|
| H A D | symbolic_opset11.py | 1208 scale_grad_by_freq, argument
|
| H A D | symbolic_helper.py | 1945 scale_grad_by_freq, argument
|
| H A D | symbolic_opset9.py | 909 scale_grad_by_freq, argument
936 scale_grad_by_freq, argument
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_embedding_test.cpp | 31 bool scale_grad_by_freq, in op_embedding_out()
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_operators.py | 1141 def embedding(g, weight, indices, padding_idx, scale_grad_by_freq, sparse): argument
1183 def embedding(g, weight, indices, padding_idx, scale_grad_by_freq, sparse): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 384 bool scale_grad_by_freq, in compute_shape_embedding()
421 bool scale_grad_by_freq) { in compute_shape_embedding_dense_backward()
|
| /aosp_15_r20/external/executorch/backends/apple/mps/serialization/ |
| H A D | mps_graph_schema.py | 461 scale_grad_by_freq: bool = False variable in MPSEmbedding
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | shim_common.cpp | 466 int32_t scale_grad_by_freq, in aoti_torch__embedding_bag()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorMath.cpp | 332 bool scale_grad_by_freq, in NestedTensor_embedding()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 3353 scale_grad_by_freq=False, argument
6357 scale_grad_by_freq, argument
6399 scale_grad_by_freq, argument
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_external_calls.cpp | 395 bool scale_grad_by_freq = false; in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 2960 def embedding(weight, indices, padding_idx=-1, scale_grad_by_freq=False, sparse=False): argument
|