def save_state(self, observed: GraphModule) -> None:
observed._activation_post_process_map = \
self.activation_post_process_map # type: ignore
observed._patterns = self.patterns # type: ignore
observed._qconfig_map = self.qconfig_map # type: ignore
observed._prepare_custom_config_dict = \
self.prepare_custom_config_dict # type: ignore
def save_state(observed: GraphModule, qconfig_map: Dict[str, QConfigAny],
node_name_to_scope: Dict[str, Tuple[str, type]],
patterns: Dict[Pattern, QuantizeHandler],
prepare_custom_config_dict: Dict[str, Any]) -> None:
observed._patterns = patterns # type: ignore[assignment]
observed._qconfig_map = qconfig_map # type: ignore[assignment]
observed._prepare_custom_config_dict = \
prepare_custom_config_dict # type: ignore[assignment]
observed._node_name_to_scope = node_name_to_scope # type: ignore[assignment]
def save_state(
observed: GraphModule,
qconfig_map: Dict[str, QConfigAny],
node_name_to_scope: Dict[str, Tuple[str, type]],
patterns: Dict[Pattern, QuantizeHandler],
prepare_custom_config_dict: Dict[str, Any],
equalization_qconfig_map: Dict[str, Any],
qconfig_dict: Dict[str, Dict[Any, Any]],
is_training: bool,
) -> None:
observed._patterns = patterns # type: ignore[assignment]
observed._qconfig_map = qconfig_map # type: ignore[assignment]
observed._prepare_custom_config_dict = \
prepare_custom_config_dict # type: ignore[assignment]
observed._node_name_to_scope = node_name_to_scope # type: ignore[assignment]
observed._equalization_qconfig_map = equalization_qconfig_map # type: ignore[assignment]
observed._qconfig_dict = qconfig_dict # type: ignore[assignment]
observed._is_training = is_training # type: ignore[assignment]
