torch.fx.passes.pass_manager.inplace_wrapper#
- torch.fx.passes.pass_manager.inplace_wrapper(fn)[source]#
Convenience wrapper for passes which modify an object inplace. This wrapper makes them return the modified object instead.
- Parameters:
fn (Callable[Object, Any]) –
- Returns:
wrapped_fn (Callable[Object, Object])
- Return type:
Callable[[Concatenate[_T, ~_P]], _T]