Skip to content

Instantly share code, notes, and snippets.

@dhruvbird

dhruvbird/PyTorchModelForwardGraph.py

Created November 12, 2021 19:37
Show Gist options
  • Save dhruvbird/b696b5e8537a58560d40e0c1182e1b2e to your computer and use it in GitHub Desktop.
Save dhruvbird/b696b5e8537a58560d40e0c1182e1b2e to your computer and use it in GitHub Desktop.
Download ZIP
Result of running scripted.forward.graph
Raw
PyTorchModelForwardGraph.py
graph(%self : __torch__.___torch_mangle_41.AddtensorsModel,
%x.1 : Tensor,
%y.1 : Tensor):
%5 : Tensor = prim::CallMethod[name="helper"](%self, %x.1, %y.1) # <ipython-input-98-b27af2abc3c5>:12:11
return (%5)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment