Move is_param and get_param out of exir and into export (#67)

Summary:
X-link: pytorch/pytorch#107264

Pull Request resolved: #67

These doesn't feel edge specific so moving out of exir.

Differential Revision: D48361384

fbshipit-source-id: bc30f25fd33dc0ad0f8c1c1c27cd50352c5cd9c8

Author: JacobSzwejbka

exir/program/__init__.py

@@ -11,8 +11,6 @@
     edge_to_executorch_passes,
     ExecutorchProgram,
     ExirExportedProgram,
-    get_param,
-    is_param,
     multi_method_program_to_executorch,
     MultiMethodExecutorchProgram,
     MultiMethodExirExportedProgram,
@@ -23,8 +21,6 @@
     "ExecutorchProgram",
     "_to_edge",
     "edge_to_executorch_passes",
-    "is_param",
-    "get_param",
     "MultiMethodExirExportedProgram",
     "MultiMethodExecutorchProgram",
     "multi_method_program_to_executorch",

exir/program/_program.py

@@ -203,28 +203,6 @@ def edge_to_executorch_passes(config: ExecutorchBackendConfig) -> List[PassType]
     return passes
 
 
-def is_param(edge_program: ExportedProgram, node: torch.fx.Node) -> bool:
-    """
-    Checks if the given node is a parameter within the edge_program
-    """
-    return node.name in edge_program.graph_signature.inputs_to_parameters
-
-
-def get_param(
-    edge_program: ExportedProgram,
-    node: torch.fx.Node,
-) -> Optional[torch.nn.Parameter]:
-    """
-    Returns the parameter associated with the given node in the edge program.
-    Returns None if the node is not a parameter within the edge_program
-    """
-    if is_param(edge_program, node):
-        parameter_name = edge_program.graph_signature.inputs_to_parameters[node.name]
-        return edge_program.state_dict[parameter_name]
-
-    return None
-
-
 ######## MULTI METHOD STUFF BELOW HERE. TO BE MERGED INTO ExirExportedProgram and ExecutorchProgram AND THEN DELETED ##########