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