Function wrapper

python/paddle/jit/api.py:828–837

(
        layer: Layer | Callable[..., Any],
        path: str,
        input_spec: Sequence[InputSpec | Tensor | object] | None = None,
        **configs: Unpack[_SaveOptions],
    )

@wrap_decorator
def _run_save_pre_hooks(func: _SaveFunction) -> _SaveFunction:
    def wrapper(
        layer: Layer | Callable[..., Any],
        path: str,
        input_spec: Sequence[InputSpec | Tensor | object] | None = None,
        **configs: Unpack[_SaveOptions],
    ) -> None:
        global _save_pre_hooks
        for hook in _save_pre_hooks:
            hook(layer, input_spec, configs)
        func(layer, path, input_spec, **configs)

    return wrapper