torch.fx.passes.pass_manager.loop_pass

torch.fx.passes.pass_manager.loop_pass(base_pass, n_iter=None, predicate=None) [source]

Удобная обёртка для проходов, которые требуется применять несколько раз.

Необходимо указать ровно один из параметров: n_iter`or `predicate.

Параметры:

• base_pass (Callable[Object, Object]) – проход, выполняемый в цикле
• n_iter (int, optional) – количество повторений прохода
• predicate (Callable[Object, bool], optional) –

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

Callable[[_T], _T]