NodeEventTracker

class torch.fx.passes.splitter_base.NodeEventTracker(tracker_mode, dump_prefix) [исходный код]

Отслеживает события узлов во время выполнения разделителя.

Предупреждение

Этот API является экспериментальным и НЕ обеспечивает обратную совместимость.

add(node, desc, dep=None) [исходный код]

Добавляет новое событие в средство отслеживания.

dump() [исходный код]

Функция, вызываемая в конце выполнения средства поиска для вывода отслеживаемых событий, заданных режимом.

print_all(writer=None) [исходный код]

Выводит все узлы в списке. @param writer: функция для записи в файл. Если None, используется print.

print_node(node_name, recursive=False, tab='', writer=None) [исходный код]

Выводит узел и его события. @param recursive: если True, выводит узлы, вызвавшие события для текущего узла. @param tab: отступ для зависимостей. @param writer: функция для записи в файл. Если None, используется print.

to_dict() [исходный код]

Создает словарь со всеми событиями.

Тип возвращаемого значения:

dict[str, list[str]]