Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.optimization.reset_modules#

torch.fx.experimental.optimization.reset_modules(nodes, modules, old_modules)[source]#

Maps each module that’s been changed with modules_to_mkldnn back to its original.

On this page
Show Source
PyTorch Libraries