Searched defs:node_to_latency_mapping (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | accelerator_partitioner.py | 854 transfer_rate_bytes_per_sec, node_to_latency_mapping argument
951 n0, n1, p0, p1, node_to_latency_mapping, transfer_rate_per_sec argument
983 node, p0, p1, node_to_latency_mapping, transfer_rate_per_sec argument
|
| H A D | partitioner_utils.py | 94 node_to_latency_mapping: Dict[Node, NodeLatency] = {} variable in PartitionerConfig
|