Revert D41690203: Clean Up MobileOptimizerType Rewrite Flags Public A…

…PI and Documentation Differential Revision: D41690203 (18de6ff) Original commit changeset: d901bdcbd16a Original Phabricator Diff: D41690203 (18de6ff) fbshipit-source-id: bba9a29d5c2b6b69423160726d29c44843023fb0
facebookresearch · Jan 10, 2023 · 7b2ba6c · 7b2ba6c
1 parent 18de6ff
commit 7b2ba6c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/d2go/export/torchscript.py b/d2go/export/torchscript.py
@@ -17,8 +17,9 @@
 from mobile_cv.common.misc.file_utils import make_temp_directory
 from mobile_cv.common.misc.iter_utils import recursive_iterate
 from torch import nn
+from torch._C import MobileOptimizerType
 from torch.utils.bundled_inputs import augment_model_with_bundled_inputs
-from torch.utils.mobile_optimizer import MobileOptimizerType, optimize_for_mobile
+from torch.utils.mobile_optimizer import optimize_for_mobile
 
 
 logger = logging.getLogger(__name__)