Skip to content

fragment

torch_to_nnef.op.fragment

List of fragment directly written in NNEF to allow composition.

Most reimplement aten:: operators.

Fragment dataclass

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

Extract definitions and extensions from our custom NNEF files.