torch.fx.passes.infra.pass_manager.pass_result_wrapper

torch.fx.passes.infra.pass_manager.pass_result_wrapper(fn) [исходный код]

Обёртка для проходов, которые в настоящее время не возвращают PassResult. Эта обёртка заставляет их возвращать PassResult, содержащий изменённый объект и значение True для флага «modified».

Параметры:

fn (Callable[Module, Any]) –

Возвращает:

wrapped_fn (Callable[Module, PassResult])

Тип возвращаемого значения:

Callable[[…], PassResult | None]

Предупреждение

Этот API является экспериментальным и НЕ гарантирует обратную совместимость.