Skip to content

torch_to_nnef.op

Module that translate all internal torch_to_nnef IR operators into NNEF.

Modules:

Name Description
aten

PyTorch Aten::* operators translation.

custom_extractors

op.custom_extractors provides mechanism to control extraction to NNEF.

extras

Dispatch for custom PyTorch ops registered under the t2n_extra:: namespace.

fragment

List of fragment directly written in NNEF to allow composition.

helper
quantized

PyTorch quantized::* operators translation.