Searched refs:node_to_latency_mapping (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | accelerator_partitioner.py | 361 partitioner_config.node_to_latency_mapping,
367 partitioner_config.node_to_latency_mapping,
797 node_to_latency_mapping: Dict[Node, NodeLatency],
842 partitions, node_to_latency_mapping
854 transfer_rate_bytes_per_sec, node_to_latency_mapping argument
866 self.partitions, node_to_latency_mapping
907 transfer_rate_bytes_per_sec, node_to_latency_mapping
918 node_to_latency_mapping: Dict[Node, NodeLatency],
951 n0, n1, p0, p1, node_to_latency_mapping, transfer_rate_per_sec argument
961 self.partitions, node_to_latency_mapping
[all …]
|
| H A D | partitioner_utils.py | 94 node_to_latency_mapping: Dict[Node, NodeLatency] = {} variable in PartitionerConfig
130 partition: Partition, node_to_latency_mapping: Dict[Node, NodeLatency]
158 node_latency = node_to_latency_mapping[node]
219 partitions: List[Partition], node_to_latency_mapping: Dict[Node, NodeLatency]
228 partition, node_to_latency_mapping
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx_experimental.py | 322 node_to_latency_mapping: Dict[Node, NodeLatency] = {}
326 node_to_latency_mapping[node] = NodeLatency(
330 node_to_latency_mapping[node] = NodeLatency(
333 return node_to_latency_mapping
339 node_to_latency_mapping = get_node_to_latency_mapping(traced)
348 partitions, node_to_latency_mapping
377 node_to_latency_mapping: Dict[Node, NodeLatency] = {}
381 node_to_latency_mapping[node] = NodeLatency(
385 node_to_latency_mapping[node] = NodeLatency(
388 return node_to_latency_mapping
[all …]
|