torch.fx.experimental.optimization.modules_to_mkldnn

torch.fx.experimental.optimization.modules_to_mkldnn(nodes, modules) [исходный код]

Для каждого узла, если это модуль, который можно предварительно преобразовать в MKLDNN, мы выполняем такое преобразование и создаём отображение, позволяющее преобразовать версию модуля для MKLDNN обратно в исходную.

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

dict[Module, Module]