Home
last modified time | relevance | path

Searched refs:get_node_to_partition_mapping (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Daccelerator_partitioner.py161 def get_node_to_partition_mapping(partitions: List[Partition]) -> Dict[Node, int]: function
399 self.node_to_partition = get_node_to_partition_mapping(self.partitions)
489 self.node_to_partition = get_node_to_partition_mapping(self.partitions)
791 self.node_to_partition = get_node_to_partition_mapping(self.partitions)
912 self.node_to_partition = get_node_to_partition_mapping(self.partitions)