diff --git a/backends/xnnpack/operators/node_visitor.py b/backends/xnnpack/operators/node_visitor.py index 3589efbe435..5ba1416003f 100644 --- a/backends/xnnpack/operators/node_visitor.py +++ b/backends/xnnpack/operators/node_visitor.py @@ -36,7 +36,6 @@ ) from executorch.backends.xnnpack.utils.xnnpack_constants import XNN_INVALID_VALUE_ID -from executorch.exir.dialects._ops import ops as exir_ops from torch.export import ExportedProgram XNN_TYPE_MAP = { diff --git a/exir/delegate.py b/exir/delegate.py index c60bc20e85a..de2be14a8e3 100644 --- a/exir/delegate.py +++ b/exir/delegate.py @@ -24,10 +24,7 @@ ProxyTorchDispatchMode, track_tensor_tree, ) -from torch.utils._python_dispatch import ( - _get_current_dispatch_mode, - _pop_mode_temporarily, -) +from torch.utils._python_dispatch import _get_current_dispatch_mode from torch.utils._pytree import tree_flatten