Home
last modified time | relevance | path

Searched full:embedding_byte (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/executorch/exir/passes/
H A D_quant_patterns_and_replacements.py29 "embedding_byte(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, "
34 "embedding_byte.dtype(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, "
39 "embedding_byte.out(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, "
44 "embedding_byte.dtype_out(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, "
80 @impl(quantized_decomposed_lib, "embedding_byte", "CompositeExplicitAutograd")
81 def embedding_byte( function
106 @register_fake("quantized_decomposed::embedding_byte.out")
116 return embedding_byte(
126 @impl(quantized_decomposed_lib, "embedding_byte.dtype", "CompositeExplicitAutograd")
153 @register_fake("quantized_decomposed::embedding_byte.dtype_out")
[all …]
/aosp_15_r20/external/executorch/kernels/quantized/cpu/
H A Dop_embedding.cpp209 * Corresponds as the out variant to torch.ops.quantized.embedding_byte
241 constexpr auto name = "quantized_decomposed::embedding_byte.out"; in quantized_embedding_byte_out()
302 constexpr auto name = "quantized_decomposed::embedding_byte.dtype_out"; in quantized_embedding_byte_dtype_out()
/aosp_15_r20/external/executorch/kernels/quantized/
H A Dquantized.yaml37 - func: quantized_decomposed::embedding_byte.out(Tensor weight, Tensor weight_scales, Tensor? weigh…
43 - func: quantized_decomposed::embedding_byte.dtype_out(Tensor weight, Tensor weight_scales, Tensor?…
H A Dtargets.bzl12 # Excluding embedding_byte ops because we choose to define them
/aosp_15_r20/external/executorch/exir/tests/
H A Dtest_quant_fusion_pass.py312 # ).check("executorch_exir_dialects_edge__ops_quantized_decomposed.embedding_byte.out"
368 # ).check("executorch_exir_dialects_edge__ops_quantized_decomposed.embedding_byte.out"
/aosp_15_r20/external/executorch/extension/aten_util/test/
H A Dmake_aten_functor_from_et_functor_test.cpp302 …"embedding_byte.out(Tensor weight, Tensor weight_scales, Tensor weight_zero_points, int weight_qua… in TORCH_LIBRARY()
328 c10::Dispatcher::singleton().findSchema({"my_op::embedding_byte", "out"}); in TEST_F()
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml687 - quantized::embedding_byte
1036 quantized::embedding_byte: 14
H A Dmodel_ops.yaml429 quantized::embedding_byte: 34
/aosp_15_r20/external/executorch/extension/llm/export/
H A Dquantizer_lib.py101 _ = torch.ops.quantized_decomposed.embedding_byte.out
/aosp_15_r20/external/executorch/docs/source/
H A Dcompiler-backend-dialect.md175 * `quantized_decomposed::embedding_byte(Tensor weight, Tensor weight_scales, Tensor weight_zero_poi…
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dembedding_ops.py166 return torch.ops.quantized.embedding_byte(
/aosp_15_r20/external/executorch/examples/models/llama/source_transformation/
H A Dquantize.py708 return torch.ops.quantized_decomposed.embedding_byte.dtype(
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag.cpp1121 TORCH_SELECTIVE_NAME("quantized::embedding_byte"), in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A Dlibrary.cpp141 …m.def(TORCH_SELECTIVE_SCHEMA("quantized::embedding_byte(__torch__.torch.classes.quantized.Embeddin… in TORCH_LIBRARY()
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_module.py1244 embedding_funcs = [torch.ops.quantized.embedding_4bit, torch.ops.quantized.embedding_byte]
H A Dtest_quantized_op.py4775 quant_ops = [torch.ops.quantized.embedding_byte, torch.ops.quantized.embedding_4bit]
4811 quant_op = torch.ops.quantized.embedding_byte