Fix ._transform naming (#162)

Summary: Pull Request resolved: #162

Reviewed By: manuelcandales

Differential Revision: D48784931

Author: angelayi

exir/capture/_capture.py

@@ -15,7 +15,6 @@
 from executorch.exir.capture._config import CaptureConfig
 from executorch.exir.error import ExportError, ExportErrorType, InternalError
 from executorch.exir.program import ExirExportedProgram, MultiMethodExirExportedProgram
-from executorch.exir.program._program import transform_exported_program
 from executorch.exir.tracer import (
     _default_decomposition_table,
     dispatch_trace,
@@ -83,7 +82,7 @@ def capture(
             # TODO remove this later
             with patch("torch._export.DECOMP_TABLE", _default_decomposition_table()):
                 ep = export(f, args, constraints=constraints)
-            ep = transform_exported_program(ep, ReplaceViewOpsWithViewCopyOpsPass())
+            ep = ep._transform(ReplaceViewOpsWithViewCopyOpsPass())
             if not config._unlift:
                 return ExirExportedProgram(ep, False)
             graph_module = ep.module()

exir/lowered_backend_module.py

@@ -242,7 +242,7 @@ def program(self, emit_stacktrace: bool = False) -> Program:
             equality_constraints=lowered_exported_program.equality_constraints,
             module_call_graph=lowered_exported_program.module_call_graph,
         )
-        exported_program = exported_program.transform(
+        exported_program = exported_program._transform(
             SpecPropPass(), MemoryPlanningPass("greedy")
         )
         emitted_program = emit_program(

exir/program/_program.py

@@ -13,7 +13,7 @@
 from executorch.exir.capture._config import EdgeCompileConfig, ExecutorchBackendConfig
 from executorch.exir.emit import emit_program, EmitterOutput
 from executorch.exir.error import ExportError
-from executorch.exir.pass_manager import PassManager, PassType
+from executorch.exir.pass_manager import PassType
 from executorch.exir.passes import (
     aten_to_edge_passes,
     EdgeToBackendOpsPass,
@@ -34,14 +34,6 @@
 Val = Any
 
 
-# Stub to ease migration from `transform` to private `_transform`
-def transform_exported_program(ep, *passes: PassType) -> ExportedProgram:
-    if hasattr(ep, "_transform"):
-        return ep._transform(*passes)
-    else:
-        return ep.transform(*passes)
-
-
 @compatibility(is_backward_compatible=False)
 class ExirExportedProgram:
     def __init__(
@@ -56,9 +48,7 @@ def __init__(
         self.after_to_edge_passes = after_to_edge_passes
 
     def transform(self, *passes: PassType) -> "ExirExportedProgram":
-        self.exported_program = transform_exported_program(
-            self.exported_program, *passes
-        )
+        self.exported_program = self.exported_program._transform(*passes)
         return self
 
     def __call__(self, *args: Any) -> Any:
@@ -86,7 +76,7 @@ def to_executorch(
             raise RuntimeError("Must run to_edge before to_executorch.")
         config = config or ExecutorchBackendConfig()
         ep = self.exported_program
-        new_prog = transform_exported_program(ep, *edge_to_executorch_passes(config))
+        new_prog = ep._transform(*edge_to_executorch_passes(config))
         new_prog = ExirExportedProgram(new_prog, self.after_to_edge_passes)
         executorch_prog = ExecutorchProgram(
             new_prog,