Fix partitioner parameters recognition (#2777)

Summary:
Fixes parameters recognition for MPS backend

Testing:
```
python3 -m unittest backends.apple.mps.test.test_mps -k test_mps_backend_partitioner -v
./cmake-out/examples/apple/mps/mps_executor_runner --model_path mps_backend_partitioner.pte --bundled_program
```

cc cccclai, shoumikhin

Pull Request resolved: #2777

Reviewed By: shoumikhin

Differential Revision: D55566484

Pulled By: cccclai

fbshipit-source-id: 21048f1eb605a65f3dc9560efad7c97d7f17d85e

Author: DenisVieriu97

backends/apple/mps/partition/mps_partitioner.py

@@ -9,6 +9,7 @@
 import torch
 from executorch.backends.apple.mps.mps_preprocess import MPSBackend
 from executorch.backends.apple.mps.operators.node_visitor import get_node_visitors
+from executorch.backends.apple.mps.utils.mps_utils import is_parameter
 from executorch.exir.backend.backend_details import CompileSpec
 from executorch.exir.backend.canonical_partitioners.pattern_op_partitioner import (
     generate_partitions_from_list_of_nodes,
@@ -30,8 +31,15 @@
 class MPSOperatorSupport(OperatorSupportBase):
     def __init__(self, edge_program: torch.export.ExportedProgram, compiler_specs):
         self.node_visitors = get_node_visitors(edge_program)
+        self.edge_program = edge_program
 
     def is_node_supported(self, submodules, node: torch.fx.Node) -> bool:
+        # Parameters are supported if any of their users are supported
+        if is_parameter(self.edge_program, node):
+            return any(
+                self.is_node_supported(submodules, user) for user in node.users.keys()
+            )
+
         if node.op != "call_function":
             return False