Home
last modified time | relevance | path

Searched refs:get_embedding_static_quant_module_mappings (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_qat.py29 get_embedding_static_quant_module_mappings,
419 model = convert(model, mapping=get_embedding_static_quant_module_mappings())
447 model = convert(model, mapping=get_embedding_static_quant_module_mappings())
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_workflow_module.py26 get_embedding_static_quant_module_mappings,
1315 mapping=get_embedding_static_quant_module_mappings())
1351 mapping=get_embedding_static_quant_module_mappings())
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dquantization_mappings.py225 def get_embedding_static_quant_module_mappings() -> Dict[Callable, Any]: function