| /aosp_15_r20/external/executorch/backends/xnnpack/test/tester/ |
| H A D | tester.py | 81 def graph_module(self): member in Stage
178 def graph_module(self) -> str: member in Quantize
205 def graph_module(self) -> str: member in Export
227 def graph_module(self) -> str: member in ToEdge
279 def graph_module(self) -> str: member in RunPasses
312 def graph_module(self) -> str: member in ToEdgeTransformAndLower
332 def graph_module(self) -> str: member in Partition
356 def graph_module(self) -> str: member in ToExecutorch
389 def graph_module(self) -> None: member in Serialize
|
| /aosp_15_r20/external/pytorch/test/fx/ |
| H A D | test_pass_infra.py | 111 def check_div_target(graph_module): argument
126 def check_bad_args(graph_module, i): argument
196 def pass_fail(graph_module): argument
|
| /aosp_15_r20/external/executorch/backends/arm/_passes/ |
| H A D | annotate_channels_last_dim_order_pass.py | 86 def insert_input_transpose(self, node, input_node, graph_module): argument
103 def insert_output_transpose(self, node, graph_module): argument
|
| H A D | decompose_linear_pass.py | 30 def call(self, graph_module): argument
|
| H A D | match_arg_ranks_pass.py | 40 def _match_op_rank(self, graph_module, node, arg, max_rank): argument
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps_utils.py | 294 graph_module, argument
316 graph_module, argument
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | exported_program.py | 704 def graph_module(self): member in ExportedProgram
1130 self, graph_module, graph_signature, *, state_dict=None, verifiers=None argument
|
| /aosp_15_r20/external/pytorch/torch/_export/passes/ |
| H A D | remove_runtime_assertions.py | 12 def call(self, graph_module) -> PassResult: argument
|
| H A D | add_runtime_assertions_for_constraints_pass.py | 82 def call(self, graph_module) -> PassResult: argument
|
| /aosp_15_r20/external/executorch/exir/program/test/ |
| H A D | test_program.py | 665 def count_nodes(graph_module, target): argument
726 def count_nodes(graph_module, target): argument
|
| /aosp_15_r20/external/executorch/exir/passes/ |
| H A D | debug_handle_generator_pass.py | 41 def get_control_flow_submodules_list(graph_module): argument
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/infra/ |
| H A D | pass_base.py | 19 def __new__(cls, graph_module, modified): argument
|
| /aosp_15_r20/external/executorch/exir/backend/test/ |
| H A D | hta_partitioner_demo.py | 159 def generate_partition_list(self, graph_module) -> List[Partition]: argument
|
| /aosp_15_r20/external/executorch/backends/arm/ |
| H A D | arm_backend.py | 193 def _get_first_delegation_tag(graph_module) -> str | None: argument
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/passes/ |
| H A D | test_activation_fusion.py | 19 def check_node_has_tag(self, graph_module, node_target, tag): argument
|
| /aosp_15_r20/external/executorch/backends/qualcomm/_passes/ |
| H A D | layout_transform.py | 163 def insert_node(self, graph_module, node, revert_layout: bool) -> None: argument
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/ |
| H A D | test_xnnpack_utils.py | 254 graph_module, argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/ |
| H A D | _tp_transform.py | 91 def call(self, graph_module) -> PassResult: argument
|
| /aosp_15_r20/external/executorch/backends/arm/test/tester/ |
| H A D | arm_tester.py | 154 def graph_module(self) -> None: member in InitialModel
|
| /aosp_15_r20/external/executorch/exir/program/ |
| H A D | _program.py | 536 def graph_module(self) -> torch.fx.GraphModule: member in ExecutorchProgram
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 786 self, graph_module, argument
849 def printGraphModule(self, graph_module, print_str=True): argument
|