Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.accelerator_partitioner.reorganize_partitions#

torch.fx.experimental.accelerator_partitioner.reorganize_partitions(partitions)[source]#

Given a list of partitions, reorganize partition id, its parents and its children for each partition

On this page
Show Source
PyTorch Libraries