Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.accelerator_partitioner.set_parents_and_children#

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

Given a list of partitions, mark parents and children for each partition

On this page
Show Source
PyTorch Libraries