Expand description
Tensor computation graph partitioner for distributed execution.
Partitions a tensor computation graph into balanced subgraphs, minimizing cross-partition communication (cut edges).
Structsยง
- Graph
Edge - A directed edge between two nodes in the tensor computation graph.
- Node
Weight - Weight associated with a single computation node in the tensor graph.
- Partition
- A single partition produced by the partitioner.
- Partition
Stats - Statistics describing a completed partitioning.
- Tensor
Graph Partitioner - Partitions a tensor computation graph into balanced subgraphs for distributed execution, minimising cross-partition edge traffic.