torch.fx.passes.pass_manager.loop_pass

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

Convenience wrapper for passes which need to be applied multiple times.

Exactly one of n_iter`or `predicate must be specified.

Parameters:

• base_pass (Callable[Object, Object]) – pass to be applied in loop

• n_iter (int, optional) – number of times to loop pass

• predicate (Callable[Object, bool], optional) –

Return type:

Callable[[_T], _T]