Function wrapper

exir/control_flow.py:190–199

(
            *args: torch.Tensor, **kwargs: Tuple[torch.Tensor]
        )

        f: Callable[..., Tuple[torch.Tensor]]
    ) -> Callable[..., Union[torch.Tensor, Tuple[torch.Tensor]]]:
        def wrapper(
            *args: torch.Tensor, **kwargs: Tuple[torch.Tensor]
        ) -> Tuple[torch.Tensor]:
            if kwargs:
                raise ExportError(
                    ExportErrorType.NOT_SUPPORTED,
                    "kwargs are not supported for @tracing_context decorated functions.",
                )

            return f(*args)

        wrapper.__tracing_inputs__ = inputs  # pyre-ignore
        return wrapper