frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Pytorch] Remove run_on_bundled_input (#58344)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58344

remove a helper function thats more trouble then its worth.

ghstack-source-id: 129131889

Test Plan: ci and {P414950111}

Reviewed By: dhruvbird

Differential Revision: D28460607

fbshipit-source-id: 31bd6c1cc169785bb360e3113d258b612cad47fc
This commit is contained in:
Jacob Szwejbka
2021-05-17 12:42:58 -07:00
committed by Facebook GitHub Bot
parent 443ce1e8a1
commit 1891e4bf1e
4 changed files with 25 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/utils/mobile_optimizer.py
View File

@ -117,7 +117,6 @@ def _get_bundled_inputs_preserved_attributes(script_module: torch.jit.ScriptModu
if hasattr(script_module, 'get_all_bundled_inputs'):
bundled_inputs_attributes.append('get_all_bundled_inputs')
bundled_inputs_attributes.append('get_num_bundled_inputs')
bundled_inputs_attributes.append('run_on_bundled_input')
# Bundled inputs in module after the change that introduced bundled inputs for multiple functions
if hasattr(script_module, 'get_bundled_inputs_functions_and_info'):