Skip to content

Instantly share code, notes, and snippets.

@pashu123
Created May 4, 2022 16:20
Show Gist options
  • Save pashu123/03dab8e4f2506b8e155300b2953e87bd to your computer and use it in GitHub Desktop.
Save pashu123/03dab8e4f2506b8e155300b2953e87bd to your computer and use it in GitHub Desktop.
graph(%self : __torch__.torch.fx.graph_module.f,
%params_1.1 : Tensor,
%params_2.1 : Tensor,
%args_1.1 : Tensor):
%90 : float = prim::Constant[value=-0.01]() # <eval_with_key>.2:26:59
%57 : bool = prim::Constant[value=1]() # <eval_with_key>.2:17:46
%26 : bool = prim::Constant[value=0]() # <eval_with_key>.2:9:132
%115 : Device = prim::Constant[value="cpu"]()
%17 : NoneType = prim::Constant()
%23 : int = prim::Constant[value=6]() # <eval_with_key>.2:9:85
%24 : int = prim::Constant[value=0]() # <eval_with_key>.2:9:97
%25 : int = prim::Constant[value=1]() # <eval_with_key>.2:9:116
%30 : int = prim::Constant[value=3]() # <eval_with_key>.2:10:47
%t.1 : Tensor = aten::t(%params_2.1) # <eval_with_key>.2:5:8
%detach.1 : Tensor = aten::detach(%t.1) # <eval_with_key>.2:6:13
%addmm.1 : Tensor = aten::addmm(%params_1.1, %args_1.1, %detach.1, %25, %25) # <eval_with_key>.2:7:12
%sum_1.1 : Tensor = aten::sum(%addmm.1, %17) # <eval_with_key>.2:8:12
%ones_like.1 : Tensor = aten::ones_like(%sum_1.1, %23, %24, %115, %26, %25) # <eval_with_key>.2:9:16
%31 : int[] = prim::ListConstruct(%30, %30)
%expand.1 : Tensor = aten::expand(%ones_like.1, %31, %26) # <eval_with_key>.2:10:13
%detach_1.1 : Tensor = aten::detach(%expand.1) # <eval_with_key>.2:11:15
%t_1.1 : Tensor = aten::t(%detach_1.1) # <eval_with_key>.2:12:10
%detach_2.1 : Tensor = aten::detach(%t_1.1) # <eval_with_key>.2:13:15
%mm.1 : Tensor = aten::mm(%detach_2.1, %args_1.1) # <eval_with_key>.2:14:9
%t_2.1 : Tensor = aten::t(%mm.1) # <eval_with_key>.2:15:10
%detach_3.1 : Tensor = aten::detach(%t_2.1) # <eval_with_key>.2:16:15
%56 : int[] = prim::ListConstruct(%24)
%sum_2.1 : Tensor = aten::sum(%detach_1.1, %56, %57, %17) # <eval_with_key>.2:17:12
%62 : int[] = prim::ListConstruct(%30)
%view.1 : Tensor = aten::view(%sum_2.1, %62) # <eval_with_key>.2:18:11
%detach_4.1 : Tensor = aten::detach(%view.1) # <eval_with_key>.2:19:15
%detach_5.1 : Tensor = aten::detach(%detach_4.1) # <eval_with_key>.2:20:15
%detach_6.1 : Tensor = aten::detach(%detach_5.1) # <eval_with_key>.2:21:15
%t_3.1 : Tensor = aten::t(%detach_3.1) # <eval_with_key>.2:22:10
%detach_7.1 : Tensor = aten::detach(%t_3.1) # <eval_with_key>.2:23:15
%detach_8.1 : Tensor = aten::detach(%detach_7.1) # <eval_with_key>.2:24:15
%detach_9.1 : Tensor = aten::detach(%detach_8.1) # <eval_with_key>.2:25:15
%add_.1 : Tensor = aten::add_(%params_1.1, %detach_6.1, %90) # <eval_with_key>.2:26:11
%add__1.1 : Tensor = aten::add_(%params_2.1, %detach_9.1, %90) # <eval_with_key>.2:27:13
%105 : Tensor[] = prim::ListConstruct(%add_.1, %add__1.1)
return (%105)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment