Home
last modified time | relevance | path

Searched refs:ExportedProgram (Results 1 – 25 of 159) sorted by relevance

1234567

/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A Dio_adapter.py42 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
63 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
99 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
120 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
190 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
227 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
252 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
280 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
314 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
373 model: torch.nn.Module | Callable | torch_export.ExportedProgram | None = None,
[all …]
/aosp_15_r20/external/executorch/exir/program/
H A D_program.py59 from torch.export import ExportedProgram
190 transformed_ep = ExportedProgram(
289 def transform_exported_program(ep, *passes: PassType) -> ExportedProgram:
296 class HackedUpExportedProgramDONOTUSE(ExportedProgram):
384 exported_program: ExportedProgram, argument
543 def dump_exported_program(self) -> ExportedProgram:
587 ExportedProgram(
625 new_ep.exported_program = ExportedProgram(
705 program: ExportedProgram, argument
707 ) -> ExportedProgram:
[all …]
H A D_fake_program.py15 from torch.export import ExportedProgram
18 def get_fake_program(real_exported_program: ExportedProgram) -> ExportedProgram: argument
47 fake_exported_program = ExportedProgram(
61 fake_exported_program: ExportedProgram, real_exported_program: ExportedProgram argument
/aosp_15_r20/external/executorch/docs/source/tutorials_source/
H A Dexport-to-executorch-tutorial.py52 from torch.export import export, ExportedProgram
69 aten_dialect: ExportedProgram = export(SimpleConv(), example_args)
104 aten_dialect: ExportedProgram = export(Basic(), example_args)
133 aten_dialect: ExportedProgram = export(
216 aten_dialect: ExportedProgram = export(converted_graph, example_args)
246 aten_dialect: ExportedProgram = export(SimpleConv(), example_args)
270 aten_encode: ExportedProgram = export(Encode(), encode_args)
273 aten_decode: ExportedProgram = export(Decode(), decode_args)
294 aten_dialect: ExportedProgram = export(SimpleConv(), example_args)
360 aten_dialect: ExportedProgram = export(LowerableModule(), example_args)
[all …]
/aosp_15_r20/external/executorch/exir/passes/
H A Dconstant_prop_pass.py24 from torch.export import ExportedProgram
47 exported_program: ExportedProgram, argument
67 exported_program: ExportedProgram, argument
82 exported_program: ExportedProgram, argument
108 exported_program: ExportedProgram, argument
152 def get_first_user_input(exported_program: ExportedProgram) -> torch.fx.Node: argument
170 exported_program: ExportedProgram, argument
197 def get_fake_mode(exported_program: ExportedProgram): argument
210 exported_program: ExportedProgram, argument
229 exported_program: ExportedProgram, argument
[all …]
/aosp_15_r20/external/executorch/exir/backend/
H A Dbackend_api.py38 from torch.export import ExportedProgram
69 edge_program: ExportedProgram, argument
107 assert isinstance(edge_program, ExportedProgram)
159 owning_program: ExportedProgram, argument
194 owning_program: ExportedProgram, argument
264 def generate_debug_handle(ep: ExportedProgram) -> int: argument
312 owning_program: ExportedProgram, argument
335 edge_program: ExportedProgram, argument
337 ) -> ExportedProgram:
402 return ExportedProgram(
H A Dpartitioner.py16 from torch.export import ExportedProgram
32 tagged_exported_program: ExportedProgram
65 def __call__(self, exported_program: ExportedProgram) -> PartitionResult: argument
74 def partition(self, exported_program: ExportedProgram) -> PartitionResult: argument
99 ep: ExportedProgram, argument
/aosp_15_r20/external/executorch/devtools/etrecord/
H A D_etrecord.py25 ExportedProgram,
56 edge_dialect_program: Optional[ExportedProgram] = None
57 graph_map: Optional[Dict[str, ExportedProgram]] = None
66 etrecord_zip: ZipFile, module_name: str, method_name: str, ep: ExportedProgram argument
68 assert isinstance(ep, ExportedProgram)
91 ExportedProgram, argument
99 elif isinstance(export_module, ExportedProgram):
117 etrecord_zip: ZipFile, edge_dialect_exported_program: ExportedProgram argument
172 ExportedProgram, argument
288 graph_map: Dict[str, ExportedProgram] = {}
/aosp_15_r20/external/executorch/backends/xnnpack/partition/config/
H A Dgemm_configs.py36 from torch.export import ExportedProgram
62 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
73 self, node: torch.fx.Node, ep: ExportedProgram argument
99 ep: ExportedProgram, argument
122 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType argument
153 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType argument
193 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType argument
206 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType argument
266 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType argument
295 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
[all …]
H A Dgeneric_node_configs.py21 from torch.export import ExportedProgram
36 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
40 self, node: torch.fx.Node, ep: ExportedProgram argument
125 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
173 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
229 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
283 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
337 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
384 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
409 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
[all …]
H A Dnode_configs.py24 from torch.export import ExportedProgram
33 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
57 self, node: torch.fx.Node, ep: ExportedProgram argument
83 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
112 self, node: torch.fx.Node, ep: ExportedProgram argument
128 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
143 self, node: torch.fx.Node, ep: ExportedProgram argument
/aosp_15_r20/external/executorch/exir/backend/canonical_partitioners/
H A Dconfig_partitioner.py11 from executorch.exir.backend.backend_details import ExportedProgram
69 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool: argument
92 self, node: torch.fx.Node, ep: ExportedProgram argument
131 ep: ExportedProgram, argument
163 self, ep: ExportedProgram argument
178 def generate_partitions(self, ep: ExportedProgram) -> List[Partition]: argument
187 def partition(self, exported_program: ExportedProgram) -> PartitionResult: argument
/aosp_15_r20/external/executorch/exir/emit/
H A D_emit_program.py23 from torch.export.exported_program import ExportedProgram, OutputKind
51 def _remove_non_user_outputs(exported_program: ExportedProgram) -> torch.fx.GraphModule: argument
84 def _get_training_metadata(methods: Dict[str, ExportedProgram]) -> Dict[str, int]: argument
111 methods: Union[ExportedProgram, Dict[str, ExportedProgram]], argument
130 if isinstance(methods, ExportedProgram):
136 if not isinstance(exported_program, ExportedProgram):
/aosp_15_r20/external/executorch/exir/serde/
H A Dserialize.py268 assert isinstance(serialized_artifact.exported_program, schema.ExportedProgram)
290 self, exported_program: ep.ExportedProgram argument
297 assert isinstance(exported_program, ep.ExportedProgram)
324 serialized_ep = schema.ExportedProgram(
572 exported_program: export_serialize.ExportedProgram, argument
578 ) -> ep.ExportedProgram:
579 assert isinstance(exported_program, export_serialize.ExportedProgram)
627 exported_program = ep.ExportedProgram(
652 exported_program: ep.ExportedProgram, argument
658 assert isinstance(serialized_artifact.exported_program, schema.ExportedProgram)
[all …]
/aosp_15_r20/external/executorch/exir/
H A Dlowered_backend_module.py33 ExportedProgram,
64 _original_exported_program: ExportedProgram # The original EXIR module
68 edge_program: ExportedProgram, argument
82 copied_program = ExportedProgram(
131 def original_module(self) -> ExportedProgram:
310 exported_program = ExportedProgram(
369 gm: torch.fx.GraphModule, owning_program: ExportedProgram argument
435 original_program: ExportedProgram, argument
658 owning_program: ExportedProgram, argument
662 ) -> Tuple[ExportedProgram, Dict[str, InputSpec], Dict[str, OutputSpec]]:
[all …]
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dutils.py21 node: torch.fx.Node, edge_program: torch.export.ExportedProgram argument
31 node: torch.fx.Node, edge_program: torch.export.ExportedProgram argument
48 param: torch.Tensor, node: torch.fx.Node, edge_program: torch.export.ExportedProgram argument
67 tensor: torch.fx.Node, edge_program: torch.export.ExportedProgram argument
95 tensor: torch.fx.Node, edge_program: torch.export.ExportedProgram argument
/aosp_15_r20/external/pytorch/torch/export/
H A D__init__.py65 from .exported_program import ExportedProgram, ModuleCallEntry, ModuleCallSignature
81 ) -> ExportedProgram:
172 ) -> ExportedProgram:
282 ep: ExportedProgram, argument
333 if not isinstance(ep, ExportedProgram):
368 ) -> ExportedProgram:
/aosp_15_r20/external/executorch/exir/backend/test/
H A Dtest_backends_nested.py36 from torch.export import ExportedProgram
78 edge_program: ExportedProgram, argument
120 def partition(self, exported_program: ExportedProgram) -> PartitionResult: argument
131 edge_program: ExportedProgram, argument
134 assert isinstance(edge_program, ExportedProgram)
206 def partition(self, exported_program: ExportedProgram) -> PartitionResult: argument
H A Dop_partitioner_demo.py30 from torch.export import ExportedProgram
87 def partition(self, exported_program: ExportedProgram) -> PartitionResult: argument
105 def partition(self, edge_exported_program: ExportedProgram) -> PartitionResult: argument
147 def partition(self, edge_exported_program: ExportedProgram) -> PartitionResult: argument
213 self, ep: ExportedProgram argument
262 def partition(self, exported_program: ExportedProgram) -> PartitionResult: argument
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_analysis.py52 exported_program: torch.export.ExportedProgram, argument
143 def _get_io_specs(exported_program: torch.export.ExportedProgram) -> tuple[dict, dict]: argument
176 exported_program: torch.export.ExportedProgram, argument
187 exported_program: torch.export.ExportedProgram, argument
228 program_a: torch.export.ExportedProgram, program_b: torch.export.ExportedProgram argument
H A D_capture_strategies.py39 exported_program: torch.export.ExportedProgram | None
104 ) -> torch.export.ExportedProgram:
122 ) -> torch.export.ExportedProgram:
163 ) -> torch.export.ExportedProgram:
204 ) -> torch.export.ExportedProgram:
297 ) -> torch.export.ExportedProgram:
/aosp_15_r20/external/executorch/backends/apple/mps/
H A Dmps_preprocess.py35 from torch.export.exported_program import ExportedProgram
68 edge_program: ExportedProgram, argument
168 _: ExportedProgram, argument
194 edge_program: ExportedProgram, argument
204 edge_program: ExportedProgram, argument
215 edge_program: ExportedProgram, argument
/aosp_15_r20/external/executorch/backends/xnnpack/_passes/
H A D__init__.py39 from torch.export import ExportedProgram
45 exported_program: ExportedProgram, argument
76 def exported_program(self) -> ExportedProgram:
79 def transform(self) -> ExportedProgram:
/aosp_15_r20/external/executorch/backends/arm/test/
H A Drunner_utils.py22 from torch.export import ExportedProgram
50 def _get_input_names(program: ExportedProgram) -> list[str]: argument
72 program: ExportedProgram, argument
110 def _get_output_node(program: ExportedProgram) -> Node: argument
127 program: ExportedProgram, output_node: Node argument
188 exported_program: ExportedProgram, argument
189 edge_program: ExportedProgram, argument
/aosp_15_r20/external/executorch/examples/apple/mps/scripts/
H A Dbench_utils.py12 from torch.export.exported_program import ExportedProgram
50 def get_executorch_model(executorch_program: ExportedProgram): argument
64 def bench_torch(executorch_program: ExportedProgram, model, inputs, model_name): argument
82 executorch_program: ExportedProgram, argument

1234567