Home
last modified time | relevance | path

Searched defs:NodeLatency (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dpartitioner_utils.py66 class NodeLatency(NamedTuple): class
130 partition: Partition, node_to_latency_mapping: Dict[Node, NodeLatency] argument
219 partitions: List[Partition], node_to_latency_mapping: Dict[Node, NodeLatency] argument
H A Daccelerator_partitioner.py797 node_to_latency_mapping: Dict[Node, NodeLatency], argument
918 node_to_latency_mapping: Dict[Node, NodeLatency], argument