Refactor _apply_replace_quant_nodes function signature

Updated the _apply_replace_quant_nodes function to accept a generic edge argument instead of EdgeProgramManager.

Author: psiddh

examples/arm/aot_arm_compiler.py

@@ -199,12 +199,9 @@ def _load_serialized_model(
     return model, example_inputs
 
 
-def _apply_replace_quant_nodes(edge: "EdgeProgramManager", args: argparse.Namespace):
-    """Apply the replace_quant_nodes pass to an EdgeProgramManager.
+def _apply_replace_quant_nodes(edge, args):
+    """Apply the replace_quant_nodes pass to the edge graph module."""
 
-    The `edge` argument is expected to be an EdgeProgramManager instance as
-    returned by `to_edge_transform_and_lower`.
-    """
     if args.target != "vgf" and not args.direct_drive:
         edge = edge.transform([ReplaceQuantNodesPass()])
     return edge