Skip to content

Instantly share code, notes, and snippets.

@dhruvbird

dhruvbird/OptimizedModelForwardGraph.py

Created November 12, 2021 19:39
Show Gist options
  • Save dhruvbird/30af73058a30e3df3691c177cc982b1e to your computer and use it in GitHub Desktop.
Save dhruvbird/30af73058a30e3df3691c177cc982b1e to your computer and use it in GitHub Desktop.
Download ZIP
Result of running optimized_model.forward.graph
Raw
OptimizedModelForwardGraph.py
graph(%self : __torch__.___torch_mangle_47.AddtensorsModel,
%x.1 : Tensor,
%y.1 : Tensor):
%3 : int = prim::Constant[value=1]()
%self.t1 : Tensor = prim::Constant[value= 0 5 10 [ CPUFloatType{3} ]]()
%5 : Tensor = aten::add(%self.t1, %x.1, %3) # <ipython-input-98-b27af2abc3c5>:8:8
%z.5 : Tensor = aten::add(%5, %y.1, %3) # <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