Home
last modified time | relevance | path

Searched defs:ExportedProgram (Results 1 – 25 of 117) sorted by relevance

12345

/aosp_15_r20/external/executorch/exir/passes/
H A Dconstant_prop_pass.py47 exported_program: ExportedProgram,
67 exported_program: ExportedProgram,
82 exported_program: ExportedProgram,
108 exported_program: ExportedProgram,
152 def get_first_user_input(exported_program: ExportedProgram) -> torch.fx.Node:
170 exported_program: ExportedProgram,
197 def get_fake_mode(exported_program: ExportedProgram):
210 exported_program: ExportedProgram,
229 exported_program: ExportedProgram,
264 exported_program: ExportedProgram,
/aosp_15_r20/external/executorch/backends/xnnpack/partition/config/
H A Dgemm_configs.py62 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
73 self, node: torch.fx.Node, ep: ExportedProgram
99 ep: ExportedProgram,
122 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType
153 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType
193 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType
206 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType
266 self, node: torch.fx.Node, ep: ExportedProgram, precision: ConfigPrecisionType
295 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
343 ep: ExportedProgram,
[all …]
H A Dgeneric_node_configs.py36 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
40 self, node: torch.fx.Node, ep: ExportedProgram
125 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
173 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
229 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
283 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
337 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
384 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
409 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
488 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
H A Dnode_configs.py33 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
57 self, node: torch.fx.Node, ep: ExportedProgram
83 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
112 self, node: torch.fx.Node, ep: ExportedProgram
128 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
143 self, node: torch.fx.Node, ep: ExportedProgram
/aosp_15_r20/external/executorch/exir/program/
H A D_program.py384 exported_program: ExportedProgram,
705 program: ExportedProgram,
763 program: ExportedProgram,
805 def _restore_transformed_ops_to_aten_ops(program: ExportedProgram):
861 program: ExportedProgram,
899 aten_programs: Dict[str, ExportedProgram], argument
961 programs: Union[ExportedProgram, Dict[str, ExportedProgram]], argument
1071 programs: Union[ExportedProgram, Dict[str, ExportedProgram]], argument
1114 programs: Union[ExportedProgram, Dict[str, ExportedProgram]], argument
1162 edge_programs: Union[ExportedProgram, Dict[str, ExportedProgram]], argument
[all …]
H A D_fake_program.py18 def get_fake_program(real_exported_program: ExportedProgram) -> ExportedProgram:
61 fake_exported_program: ExportedProgram, real_exported_program: ExportedProgram
/aosp_15_r20/external/executorch/exir/backend/test/
H A Dtest_partitioner.py60 self, edge_exported_program: ExportedProgram
124 self, edge_exported_program: ExportedProgram
161 self, edge_exported_program: ExportedProgram
216 self, edge_exported_program: ExportedProgram
286 self, edge_exported_program: ExportedProgram
363 self, edge_exported_program: ExportedProgram
449 self, edge_exported_program: ExportedProgram
508 self, edge_exported_program: ExportedProgram
581 self, edge_exported_program: ExportedProgram
H A Dop_partitioner_demo.py87 def partition(self, exported_program: ExportedProgram) -> PartitionResult:
105 def partition(self, edge_exported_program: ExportedProgram) -> PartitionResult:
147 def partition(self, edge_exported_program: ExportedProgram) -> PartitionResult:
213 self, ep: ExportedProgram
262 def partition(self, exported_program: ExportedProgram) -> PartitionResult:
H A Dtest_backends_nested.py78 edge_program: ExportedProgram,
120 def partition(self, exported_program: ExportedProgram) -> PartitionResult:
131 edge_program: ExportedProgram,
206 def partition(self, exported_program: ExportedProgram) -> PartitionResult:
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_analysis.py52 exported_program: torch.export.ExportedProgram,
143 def _get_io_specs(exported_program: torch.export.ExportedProgram) -> tuple[dict, dict]:
176 exported_program: torch.export.ExportedProgram,
187 exported_program: torch.export.ExportedProgram,
228 program_a: torch.export.ExportedProgram, program_b: torch.export.ExportedProgram
/aosp_15_r20/external/executorch/exir/backend/
H A Dbackend_api.py69 edge_program: ExportedProgram,
159 owning_program: ExportedProgram,
194 owning_program: ExportedProgram,
264 def generate_debug_handle(ep: ExportedProgram) -> int:
312 owning_program: ExportedProgram,
335 edge_program: ExportedProgram,
H A Dutils.py179 tagged_exported_program: ExportedProgram,
206 tagged_exported_program: ExportedProgram,
340 def tag_constant_data(edge_program: ExportedProgram) -> None:
386 def tag_mutated_buffer(edge_program: ExportedProgram) -> None:
H A Dpartitioner.py65 def __call__(self, exported_program: ExportedProgram) -> PartitionResult:
74 def partition(self, exported_program: ExportedProgram) -> PartitionResult:
99 ep: ExportedProgram,
/aosp_15_r20/external/executorch/exir/backend/canonical_partitioners/
H A Dconfig_partitioner.py69 def check_constraints(self, node: torch.fx.Node, ep: ExportedProgram) -> bool:
92 self, node: torch.fx.Node, ep: ExportedProgram
131 ep: ExportedProgram,
163 self, ep: ExportedProgram
178 def generate_partitions(self, ep: ExportedProgram) -> List[Partition]:
187 def partition(self, exported_program: ExportedProgram) -> PartitionResult:
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dutils.py21 node: torch.fx.Node, edge_program: torch.export.ExportedProgram
31 node: torch.fx.Node, edge_program: torch.export.ExportedProgram
48 param: torch.Tensor, node: torch.fx.Node, edge_program: torch.export.ExportedProgram
67 tensor: torch.fx.Node, edge_program: torch.export.ExportedProgram
95 tensor: torch.fx.Node, edge_program: torch.export.ExportedProgram
/aosp_15_r20/external/executorch/backends/arm/test/
H A Drunner_utils.py50 def _get_input_names(program: ExportedProgram) -> list[str]:
72 program: ExportedProgram,
110 def _get_output_node(program: ExportedProgram) -> Node:
127 program: ExportedProgram, output_node: Node
188 exported_program: ExportedProgram,
189 edge_program: ExportedProgram,
/aosp_15_r20/external/executorch/backends/apple/mps/
H A Dmps_preprocess.py68 edge_program: ExportedProgram,
168 _: ExportedProgram,
194 edge_program: ExportedProgram,
204 edge_program: ExportedProgram,
215 edge_program: ExportedProgram,
/aosp_15_r20/external/pytorch/torch/_export/
H A Dutils.py405 def is_param(program: "ExportedProgram", node: torch.fx.Node) -> bool:
414 program: "ExportedProgram",
429 def is_buffer(program: "ExportedProgram", node: torch.fx.Node) -> bool:
438 program: "ExportedProgram",
457 program: "ExportedProgram",
468 program: "ExportedProgram",
/aosp_15_r20/external/executorch/exir/emit/
H A D_emit_program.py51 def _remove_non_user_outputs(exported_program: ExportedProgram) -> torch.fx.GraphModule:
84 def _get_training_metadata(methods: Dict[str, ExportedProgram]) -> Dict[str, int]: argument
111 methods: Union[ExportedProgram, Dict[str, ExportedProgram]], argument
/aosp_15_r20/external/executorch/exir/
H A Dlowered_backend_module.py68 edge_program: ExportedProgram,
369 gm: torch.fx.GraphModule, owning_program: ExportedProgram
435 original_program: ExportedProgram,
658 owning_program: ExportedProgram,
845 original_program: ExportedProgram,
/aosp_15_r20/external/executorch/backends/qualcomm/partition/
H A Dqnn_partitioner.py41 edge_program: torch.export.ExportedProgram,
122 self, edge_program: torch.export.ExportedProgram
136 self, partitions: List[Partition], edge_program: torch.export.ExportedProgram
160 def partition(self, edge_program: torch.export.ExportedProgram) -> PartitionResult:
/aosp_15_r20/external/executorch/backends/arm/
H A Dprocess_node.py132 edge_program: ExportedProgram,
162 edge_program: ExportedProgram,
185 edge_program: ExportedProgram,
200 edge_program: ExportedProgram,
/aosp_15_r20/external/executorch/devtools/etrecord/
H A D_etrecord.py66 etrecord_zip: ZipFile, module_name: str, method_name: str, ep: ExportedProgram
91 ExportedProgram, argument
117 etrecord_zip: ZipFile, edge_dialect_exported_program: ExportedProgram
172 ExportedProgram,
/aosp_15_r20/external/executorch/examples/apple/mps/scripts/
H A Dbench_utils.py50 def get_executorch_model(executorch_program: ExportedProgram):
64 def bench_torch(executorch_program: ExportedProgram, model, inputs, model_name):
82 executorch_program: ExportedProgram,
/aosp_15_r20/external/executorch/backends/apple/mps/partition/
H A Dmps_partitioner.py39 def __init__(self, edge_program: torch.export.ExportedProgram, compiler_specs):
60 def generate_partitions(self, edge_program: ExportedProgram) -> List[Any]:
123 def partition(self, edge_program: ExportedProgram) -> PartitionResult:

12345