get_fusion_pattern_to_root_node_getter

class torch.ao.quantization.backend_config.utils.get_fusion_pattern_to_root_node_getter(backend_config) [исходный код]

Получить отображение шаблонов слияния в функцию, которая возвращает корневой узел шаблона слияния. Например, чаще всего используется следующий вариант:

def get_root_node(node_pattern):
    while not isinstance(node_pattern[-1], Node):
        node_pattern = node_pattern[-1]
    return node_pattern[-1]

Это работает для всех шаблонов, у которых корневой узел является «последним узлом» в шаблоне, например (torch.add, MatchAllNode, (torch.ReLU, torch.Conv2d)).

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

dict[Pattern, Callable]