Plugin Development (PyTorch)
For details about the adaptation plugin development process and sample in single-operator mode of the PyTorch framework, see "Adapting OpPlugin to a Single Operator" in . For details about how to develop custom operators in graph mode, see "Importing Custom Operators into a Graph as Plugins" in PyTorch Graph Mode User Guide (TorchAir).
Parent topic: Operator Adaptation