Rate this Page

torch.utils.mobile_optimizer#

Created On: Jul 15, 2020 | Last Updated On: May 20, 2025

PyTorch Mobile is no longer actively supported. Redirecting to ExecuTorch documentation.

Warning

PyTorch Mobile is no longer actively supported. Please check out ExecuTorch, PyTorch’s all-new on-device inference library. You can also review documentation on XNNPACK and Vulkan delegates.

torch.utils.mobile_optimizer.optimize_for_mobile(script_module, optimization_blocklist=None, preserved_methods=None, backend='CPU')[source]#

Optimize a torch script module for mobile deployment.

Parameters
  • script_module (ScriptModule) – An instance of torch script module with type of ScriptModule.

  • optimization_blocklist (Optional[set[torch._C._MobileOptimizerType]]) – A set with type of MobileOptimizerType. When set is not passed, optimization method will run all the optimizer pass; otherwise, optimizer method will run the optimization pass that is not included inside optimization_blocklist.

  • preserved_methods (Optional[list[~AnyStr]]) – A list of methods that needed to be preserved when freeze_module pass is invoked

  • backend (str) – Device type to use for running the result model (‘CPU’(default), ‘Vulkan’ or ‘Metal’).

Returns

A new optimized torch script module

Return type

RecursiveScriptModule