| 12345678910111213141516 |
- from ._lower_to_native_backend import _lower_to_native_backend
- from ..qconfig import QConfigAny
- from torch.fx import GraphModule
- from typing import Dict, Tuple
- __all__ = ['lower_to_fbgemm']
- def lower_to_fbgemm(
- model: GraphModule,
- qconfig_map: Dict[str, QConfigAny],
- node_name_to_scope: Dict[str, Tuple[str, type]]
- ) -> GraphModule:
- """ Lower a quantized reference model (with reference quantized operator patterns)
- to fbgemm
- """
- return _lower_to_native_backend(model, qconfig_map, node_name_to_scope)
|
