Skip to content

torch_to_nnef.op.fragment

List of fragment directly written in NNEF to allow composition.

Most reimplement aten:: operators.

Classes:

Name Description
Fragment

Extract definitions and extensions from our custom NNEF files.

Fragment dataclass

Fragment(name: str, extensions: Tuple[str, ...], definition: str)

Extract definitions and extensions from our custom NNEF files.