Searched refs:get_module_to_qat_module (Results 1 – 3 of 3) sorted by relevance
10 from torch.ao.quantization.backend_config.utils import get_module_to_qat_module393 module_to_qat_module_class = get_module_to_qat_module(backend_config)
24 get_module_to_qat_module,2088 module_to_qat_module = get_module_to_qat_module(backend_config)
90 def get_module_to_qat_module( function