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 |
|
extras |
Dispatch for custom PyTorch ops registered under the |
fragment |
List of fragment directly written in NNEF to allow composition. |
gated_delta |
Shared surface for the gated-delta-net (GDN) linear-attention recurrence. |
helper |
|
quantized |
PyTorch quantized::* operators translation. |