maybe_get_next_module

class torch.ao.quantization.fx.utils.maybe_get_next_module(node, modules, target_module_type=None, target_functional_type=None) [исходный код]

Получает следующий модуль, соответствующий требуемому типу is_target_module_type, если он существует

Параметры:

• node (Node) – Узел, пользователей которого нужно проверить
• target_module_type (type[Module] | None) – Тип модуля, который нужно проверить
• target_functional_type (Any | None) – Функциональный тип, который нужно проверить

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

Node | None