Home
last modified time | relevance | path

Searched defs:ExportGraphSignature (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/executorch/exir/
H A Dmemory_planning.py46 graph_signature: Optional[ExportGraphSignature] = None,
296 node: Node, graph_signature: Optional[ExportGraphSignature] = None
315 nodes: Iterable[Node], graph_signature: Optional[ExportGraphSignature] = None
338 graph_signature: Optional[ExportGraphSignature] = None,
430 graph_signature: Optional[ExportGraphSignature] = None,
551 graph_signature: Optional[ExportGraphSignature] = None,
607 graph_signature: Optional[ExportGraphSignature] = None,
725 [torch.fx.GraphModule, int, Optional[ExportGraphSignature], bool, bool],
730 graph_signature: Optional[ExportGraphSignature] = None,
/aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/
H A D_tp_transform.py81 graph_signature: ExportGraphSignature,
131 graph_signature: ExportGraphSignature,
167 def _get_input_node_fqn(input_name: str, graph_signature: ExportGraphSignature) -> str:
183 graph_signature: ExportGraphSignature,
524 graph_signature: ExportGraphSignature,
/aosp_15_r20/external/pytorch/torch/_export/
H A Dutils.py79 old_sig: "ExportGraphSignature", new_sig: "ExportGraphSignature"
151 new_sig: "ExportGraphSignature",
706 export_graph_signature: "ExportGraphSignature",
/aosp_15_r20/external/pytorch/torch/export/
H A D_trace.py348 sig: ExportGraphSignature,
387 graph_signature: ExportGraphSignature,
414 graph_signature: ExportGraphSignature,
839 graph_signature: ExportGraphSignature,
860 graph_signature: ExportGraphSignature,
942 def _verify_placeholder_names(gm: torch.fx.GraphModule, sig: ExportGraphSignature):
H A Dexported_program.py557 gm: torch.fx.GraphModule, new_graph_signature: ExportGraphSignature
667 graph_signature: ExportGraphSignature,
1026 old_signature: ExportGraphSignature,
H A D_unlift.py195 graph_signature: ExportGraphSignature,
261 graph_signature: Optional[ExportGraphSignature] = None,
H A Dgraph_signature.py134 class ExportGraphSignature: class
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dmemory_planning.py57 graph_signature: ExportGraphSignature,
114 graph_signature: ExportGraphSignature,
/aosp_15_r20/external/pytorch/torch/export/experimental/
H A D__init__.py33 gm: torch.fx.GraphModule, sig: torch.export.graph_signature.ExportGraphSignature
/aosp_15_r20/external/executorch/exir/passes/
H A Dmemory_planning_pass.py92 graph_signature: Optional[ExportGraphSignature] = None,
H A Dspec_prop_pass.py31 node: Node, graph_signature: Optional[ExportGraphSignature] = None
/aosp_15_r20/external/pytorch/torch/_export/passes/
H A Dlift_constants_pass.py113 graph_signature: ExportGraphSignature,
/aosp_15_r20/external/executorch/exir/serde/
H A Dexport_serialize.py429 graph_signature: ep.ExportGraphSignature,
1047 def serialize_signature(self, sig: ep.ExportGraphSignature) -> GraphSignature:
H A Dserialize.py50 graph_signature: ep.ExportGraphSignature,
/aosp_15_r20/external/executorch/exir/tests/
H A Dtest_memory_planning.py189 graph_signature: Optional[ExportGraphSignature] = None,
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Dserialize.py426 graph_signature: ep.ExportGraphSignature,
1077 def serialize_signature(self, sig: ep.ExportGraphSignature) -> GraphSignature:
/aosp_15_r20/external/executorch/backends/arm/test/tester/
H A Darm_tester.py571 def _format_export_graph_signature(signature: ExportGraphSignature) -> str:
/aosp_15_r20/external/executorch/exir/program/
H A D_program.py124 old_signature: ExportGraphSignature,