Move RemoveGetItemPass (remove_getitem_op.py) to backends.transform

Multiple backends are using this pass - Arm, Neutron, XNNPACK.

Author: robert-kalmar

backends/arm/_passes/arm_pass_manager.py

@@ -57,7 +57,7 @@
 
 from executorch.backends.arm.tosa_specification import Tosa_0_80, TosaSpecification
 from executorch.backends.transforms.fuse_view_copy import FuseViewCopyTransform
-from executorch.backends.xnnpack._passes.remove_getitem_op import RemoveGetItemPass
+from executorch.backends.transforms.remove_getitem_op import RemoveGetItemPass
 from executorch.exir import ExportedProgram
 from executorch.exir.pass_manager import PassManager
 from torch.fx import GraphModule

backends/nxp/nxp_backend.py

@@ -28,7 +28,7 @@
     NeutronNodeArtifacts,
 )
 from executorch.backends.nxp.neutron_pass_manager import NeutronPassManager
-from executorch.backends.xnnpack._passes import RemoveGetItemPass
+from executorch.backends.transforms.remove_getitem_op import RemoveGetItemPass
 from executorch.exir.backend.backend_details import BackendDetails, PreprocessResult
 from executorch.exir.backend.compile_spec_schema import CompileSpec
 from executorch.exir.verification.verifier import EXIREdgeDialectVerifier

backends/nxp/tests/ir/converter/node_converter/test_max_pool_2d_converter.py

@@ -10,6 +10,7 @@
 from executorch.backends.nxp.backend.edge_program_converter import (
     EdgeProgramToIRConverter,
 )
+from executorch.backends.nxp.neutron_pass_manager import NeutronPassManager
 from executorch.backends.nxp.tests.executorch_pipeline import (
     to_edge_program,
     to_quantized_edge_program,
@@ -20,7 +21,6 @@
     ToNHWCPreprocess,
 )
 from executorch.backends.nxp.tests.models import MaxPool2dConvModule, MaxPool2dModule
-from executorch.backends.nxp.neutron_pass_manager import NeutronPassManager
 from executorch.backends.xnnpack._passes import RemoveGetItemPass
 from executorch.exir.verification.verifier import EXIREdgeDialectVerifier
 from torch.export import ExportedProgram

backends/nxp/tests/test_node_format_inference.py

@@ -10,12 +10,12 @@
     NodeFormat,
     NodeFormatInference,
 )
+from executorch.backends.nxp.neutron_pass_manager import NeutronPassManager
 from executorch.backends.nxp.tests.models import (
     Conv2dModule,
     MaxPool2dModule,
     SoftmaxModule,
 )
-from executorch.backends.nxp.neutron_pass_manager import NeutronPassManager
 from executorch.backends.xnnpack._passes import RemoveGetItemPass
 from executorch.exir.verification.verifier import EXIREdgeDialectVerifier
 

backends/xnnpack/_passes/remove_getitem_op.py renamed to backends/transforms/remove_getitem_op.py

@@ -6,6 +6,8 @@
 
 from typing import List, Optional, Type
 
+from executorch.backends.transforms.remove_getitem_op import RemoveGetItemPass
+
 from executorch.backends.xnnpack._passes.channels_last_tagged_reshape_pass import (
     ChannelsLastTaggedReshapePass,
 )
@@ -23,7 +25,6 @@
     FuseBatchNormWithConvPass,
 )
 from executorch.backends.xnnpack._passes.prelu_reshape_pass import PReLUReshapePass
-from executorch.backends.xnnpack._passes.remove_getitem_op import RemoveGetItemPass
 from executorch.backends.xnnpack._passes.tag_implicit_q_dq_pass import (
     TagImplicitQDqPass,
 )

backends/xnnpack/_passes/__init__.py

@@ -7,7 +7,7 @@
 import unittest
 
 import torch
-from executorch.backends.xnnpack._passes.remove_getitem_op import RemoveGetItemPass
+from executorch.backends.transforms.remove_getitem_op import RemoveGetItemPass
 from executorch.backends.xnnpack.test.tester import RunPasses, Tester