Searched refs:SourcePartition (Results 1 – 9 of 9) sorted by relevance
| /external/pytorch/torch/fx/passes/utils/ |
| D | source_matcher_utils.py | 32 class SourcePartition: class
55 ) -> Dict[Any, List[SourcePartition]]:
97 def make_partition(nodes: List[Node], module_type: Type) -> SourcePartition:
113 return SourcePartition(
121 ret: Dict[Type[Any], List[SourcePartition]] = {}
143 def check_subgraphs_connected(subgraph1: SourcePartition, subgraph2: SourcePartition) -> bool: argument
|
| /external/executorch/backends/cadence/aot/quantizer/ |
| D | utils.py | 22 SourcePartition,
176 ) -> SourcePartition:
192 return SourcePartition(
200 ret: Dict[Type[Any], List[SourcePartition]] = {}
208 def _partitions_sequential(partitions: Tuple[SourcePartition]) -> bool: argument
223 typed_partitions: OrderedDict[Any, List[SourcePartition]] = OrderedDict()
|
| /external/pytorch/torch/ao/quantization/pt2e/ |
| D | graph_utils.py | 11 SourcePartition,
61 def _partitions_sequential(partitions: List[SourcePartition]): argument
101 typed_partitions: OrderedDict[Any, List[SourcePartition]] = OrderedDict()
|
| /external/executorch/backends/xnnpack/_passes/ |
| D | convert_to_linear.py | 22 SourcePartition,
73 def find_bias_for_mm(self, src_partition: SourcePartition, mm_node: torch.fx.Node): argument
96 src_partition: SourcePartition, argument
|
| /external/executorch/backends/qualcomm/_passes/ |
| D | convert_to_linear.py | 21 SourcePartition,
69 src_partition: SourcePartition, argument
|
| /external/executorch/exir/backend/ |
| D | utils.py | 25 from torch.fx.passes.utils.source_matcher_utils import SourcePartition
103 partition: SourcePartition, argument
|
| /external/executorch/backends/xnnpack/partition/config/ |
| D | gemm_configs.py | 39 SourcePartition,
370 self, node: torch.fx.Node, ep: ExportedProgram, src_partition: SourcePartition argument
|
| /external/pytorch/torch/ao/quantization/quantizer/ |
| D | x86_inductor_quantizer.py | 55 SourcePartition,
688 partition_list: List[SourcePartition], argument
|
| /external/executorch/docs/source/ |
| D | compiler-custom-compiler-passes.md | 472 …_source_partitions(graph: torch.fx.Graph, wanted_sources: List[Any]) -> Dict[Any, SourcePartition]`
487 class SourcePartition():
|