def qat_mod_config(
    mod: _Type[_nn.Module], qat_mod: _Type[_nn.Module], ref_quant_mod: _Type[_nn.Module]
) -> _BackendPatternConfig:
    """
    Returns backend pattern config for QAT modules.
    """
    return (
        _BackendPatternConfig(qat_mod)
        .set_observation_type(_ObservationType.OUTPUT_USE_DIFFERENT_OBSERVER_AS_INPUT)
        .set_dtype_configs(weighted_dtype_configs)
        .set_root_module(mod)
        .set_reference_quantized_module(ref_quant_mod)
    )

    Returns backend pattern config for QAT modules.
    
coremltools/optimize/torch/quantization/_backend_config_utils.py