Skip to content

Instantly share code, notes, and snippets.

@dhruvbird

dhruvbird/PyTorchModelHelperGraph.py

Created November 12, 2021 19:38
Show Gist options
  • Save dhruvbird/13cb659e4a3b49abb97bde67aa3a8002 to your computer and use it in GitHub Desktop.
Save dhruvbird/13cb659e4a3b49abb97bde67aa3a8002 to your computer and use it in GitHub Desktop.
Download ZIP
Result of running optimized_model.helper.graph
Raw
PyTorchModelHelperGraph.py
graph(%self : __torch__.___torch_mangle_41.AddtensorsModel,
%x.1 : Tensor,
%y.1 : Tensor):
%6 : int = prim::Constant[value=1]()
%z.1 : Tensor = prim::GetAttr[name="t1"](%self)
%7 : Tensor = aten::add(%z.1, %x.1, %6) # <ipython-input-98-b27af2abc3c5>:8:8
%z.5 : Tensor = aten::add(%7, %y.1, %6) # <ipython-input-98-b27af2abc3c5>:8:8
return (%z.5)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment