Home
last modified time | relevance | path

Searched full:embedding_4bit (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/executorch/exir/passes/
H A D_quant_patterns_and_replacements.py308 "embedding_4bit(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, "
313 "embedding_4bit.dtype(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, "
318 "embedding_4bit.out(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, "
323 "embedding_4bit.dtype_out(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, "
328 @impl(quantized_decomposed_lib, "embedding_4bit", "CompositeExplicitAutograd")
329 def embedding_4bit( function
360 @register_fake("quantized_decomposed::embedding_4bit.out")
370 return embedding_4bit(
380 @impl(quantized_decomposed_lib, "embedding_4bit.dtype", "CompositeExplicitAutograd")
413 @register_fake("quantized_decomposed::embedding_4bit.dtype_out")
/aosp_15_r20/external/executorch/kernels/quantized/
H A Dquantized.yaml61 - func: quantized_decomposed::embedding_4bit.out(Tensor weight, Tensor weight_scales, Tensor? weigh…
67 - func: quantized_decomposed::embedding_4bit.dtype_out(Tensor weight, Tensor weight_scales, Tensor?…
/aosp_15_r20/external/executorch/kernels/quantized/cpu/
H A Dop_embedding4b.cpp28 * Corresponds as the out variant to torch.ops.quantized.embedding_4bit
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml686 - quantized::embedding_4bit
1035 quantized::embedding_4bit: 1
H A Dmodel_ops.yaml428 quantized::embedding_4bit: 18
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dembedding_ops.py162 return torch.ops.quantized.embedding_4bit(
/aosp_15_r20/external/executorch/examples/models/llama/source_transformation/
H A Dquantize.py719 return torch.ops.quantized_decomposed.embedding_4bit.dtype(
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag.cpp1124 TORCH_SELECTIVE_NAME("quantized::embedding_4bit"), in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A Dlibrary.cpp142 …m.def(TORCH_SELECTIVE_SCHEMA("quantized::embedding_4bit(__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]