torch.fx.experimental.accelerator_partitioner.set_parents_and_children

torch.fx.experimental.accelerator_partitioner.set_parents_and_children(partitions) [исходный код]

Для заданного списка разделов помечает родительские и дочерние разделы для каждого раздела