| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qembeddingbag.cpp | 46 bool include_last_offset, in embedding_lookup_fallback_impl()
232 bool include_last_offset, in embedding_bag_nbit_impl()
404 bool include_last_offset, in embedding_bag_byte_impl()
565 bool include_last_offset, in embedding_bag_byte_helper()
663 bool include_last_offset, in _embedding_bag_nbit_helper()
767 bool include_last_offset, in embeddingbag_byte()
788 bool include_last_offset, in embeddingbag_4bit()
828 bool include_last_offset) { in embedding_bag_byte_rowwise_offsets_out()
851 bool include_last_offset) { in embedding_bag_4bit_rowwise_offsets_out()
886 bool include_last_offset) { in embedding_bag_2bit_rowwise_offsets_out()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | EmbeddingBag.cpp | 197 bool include_last_offset, in index_select_add()
375 bool include_last_offset, in index_select_add()
562 bool include_last_offset, in index_select_scale_add()
751 bool include_last_offset, in index_select_scale_add()
876 bool include_last_offset) { in check_arguments()
922 const bool include_last_offset, in make_bag_size_out()
950 bool include_last_offset) { in make_max_indices_out()
984 bool include_last_offset = (output.size(0) == offsets_size - 1); in make_offset2bag_out() local
1003 const bool include_last_offset, in make_bag_size()
1016 bool include_last_offset) { in make_max_indices()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/ |
| H A D | EmbeddingBag.cu | 94 const bool include_last_offset, in embedding_bag_nbits_rowwise_offsets_kernel()
197 bool include_last_offset, in embedding_bag_byte_impl()
276 bool include_last_offset) { in embedding_bag_byte_rowwise_offsets()
380 bool include_last_offset) { in embedding_bag_4bit_impl()
458 bool include_last_offset) { in embedding_bag_4bit_rowwise_offsets()
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_embedding.py | 497 def gen_1D_indices_offsets(include_last_offset, allpad): argument
537 indices_1D, offsets, include_last_offset, padding_idx argument
926 include_last_offset=False, argument
1131 mode, trainable_scale, include_last_offset, has_weight=True argument
|
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | qembeddingbag_test.py | 24 include_last_offset, argument
|
| H A D | embeddingbag_test.py | 21 include_last_offset, argument
|
| H A D | qatembedding_ops_test.py | 25 include_last_offset, argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/ops/ |
| H A D | test_embedding_bag.py | 38 include_last_offset=False, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | EmbeddingBag.cu | 313 bool include_last_offset, int64_t padding_idx) { in _embedding_bag_forward_only_cuda()
336 bool include_last_offset, int64_t padding_idx) { in _embedding_bag_cuda()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | embedding.h | 99 bool include_last_offset, in embedding_bag()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/qat/modules/ |
| H A D | embedding_ops.py | 152 include_last_offset=False, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 4571 include_last_offset, fallback_to_no_sparse, sparsity, atol, rtol): argument
4618 include_last_offset, weights, per_sample_weights, argument
4706 include_last_offset, argument
4731 include_last_offset, argument
4758 include_last_offset, argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset18.py | 238 include_last_offset, argument
|
| H A D | symbolic_opset10.py | 599 include_last_offset, argument
|
| H A D | symbolic_opset11.py | 1212 include_last_offset, argument
|
| H A D | symbolic_helper.py | 1949 include_last_offset, argument
|
| H A D | symbolic_opset9.py | 940 include_last_offset, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | ops.cpp | 1139 const auto include_last_offset = p_node->Input(8).toBool(); in __anon11f46a8b3102() local
1177 const auto include_last_offset = p_node->Input(8).toBool(); in __anon11f46a8b3302() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | shim_common.cpp | 470 int32_t include_last_offset, in aoti_torch__embedding_bag()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 3357 include_last_offset=False, argument
|