Skip to content

Commit 90c4b1d

Browse files
committed
Added code comment
Signed-off-by: Rishin Raj <rishinr@qti.qualcomm.com>
1 parent 1c1fdef commit 90c4b1d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/onboarding_guide/customop/example_custom_op.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@ def CustomOpBluePrint(input: onnxscript.FLOAT):
4141

4242
# Layer 2: PyTorch Autograd Function
4343
# 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.
4446

4547

4648
class CustomOpBluePrintFunc(torch.autograd.Function):

0 commit comments

Comments
 (0)