We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1c1fdef commit 90c4b1dCopy full SHA for 90c4b1d
examples/onboarding_guide/customop/example_custom_op.py
@@ -41,6 +41,8 @@ def CustomOpBluePrint(input: onnxscript.FLOAT):
41
42
# Layer 2: PyTorch Autograd Function
43
# Connects PyTorch execution to ONNX export
44
+# Pytorch forward function is called during PyTorch execution (CPU/GPU).
45
+# When running on ONNX runtime, the CustomOpBluePrint function (Layer 1) is called instead.
46
47
48
class CustomOpBluePrintFunc(torch.autograd.Function):
0 commit comments