Skip to content

Instantly share code, notes, and snippets.

@pashu123
Created March 14, 2022 07:56
Show Gist options
  • Save pashu123/d13778fc9c4ef6b158b773070aa062fd to your computer and use it in GitHub Desktop.
Save pashu123/d13778fc9c4ef6b158b773070aa062fd to your computer and use it in GitHub Desktop.
graph(%self : __torch__.torch.fx.graph_module.___torch_mangle_0.GraphModule,
%primals_1.1 : Tensor,
%primals_2.1 : Tensor,
%primals_3.1 : Tensor,
%primals_4.1 : Tensor,
%primals_5.1 : Tensor,
%primals_6.1 : Tensor,
%primals_7.1 : Tensor,
%primals_8.1 : Tensor,
%primals_9.1 : Tensor,
%primals_10.1 : Tensor,
%primals_11.1 : Tensor,
%primals_12.1 : Tensor,
%primals_13.1 : Tensor,
%primals_14.1 : Tensor,
%primals_15.1 : Tensor,
%primals_16.1 : Tensor,
%primals_17.1 : Tensor,
%primals_18.1 : Tensor,
%primals_19.1 : Tensor,
%primals_20.1 : Tensor,
%primals_21.1 : Tensor,
%primals_22.1 : Tensor,
%primals_23.1 : Tensor,
%primals_24.1 : Tensor,
%primals_25.1 : Tensor,
%primals_26.1 : Tensor,
%primals_27.1 : Tensor,
%primals_28.1 : Tensor,
%primals_29.1 : Tensor,
%primals_30.1 : Tensor,
%primals_31.1 : Tensor,
%primals_32.1 : Tensor,
%primals_33.1 : Tensor,
%primals_34.1 : Tensor,
%primals_35.1 : Tensor,
%primals_36.1 : Tensor,
%primals_37.1 : Tensor,
%primals_38.1 : Tensor,
%primals_39.1 : Tensor,
%primals_40.1 : Tensor,
%primals_41.1 : Tensor,
%primals_42.1 : Tensor,
%primals_43.1 : Tensor,
%primals_44.1 : Tensor,
%primals_45.1 : Tensor,
%primals_46.1 : Tensor,
%primals_47.1 : Tensor,
%primals_48.1 : Tensor,
%primals_49.1 : Tensor,
%primals_50.1 : Tensor,
%primals_51.1 : Tensor,
%primals_52.1 : Tensor,
%primals_53.1 : Tensor,
%primals_54.1 : Tensor,
%primals_55.1 : Tensor,
%primals_56.1 : Tensor,
%primals_57.1 : Tensor,
%primals_58.1 : Tensor,
%primals_59.1 : Tensor,
%primals_60.1 : Tensor,
%primals_61.1 : Tensor,
%primals_62.1 : Tensor,
%primals_63.1 : Tensor,
%primals_64.1 : Tensor,
%primals_65.1 : Tensor,
%primals_66.1 : Tensor,
%primals_67.1 : Tensor,
%primals_68.1 : Tensor,
%primals_69.1 : Tensor,
%primals_70.1 : Tensor,
%primals_71.1 : Tensor,
%primals_72.1 : Tensor,
%primals_73.1 : Tensor,
%primals_74.1 : Tensor,
%primals_75.1 : Tensor,
%primals_76.1 : Tensor,
%primals_77.1 : Tensor,
%primals_78.1 : Tensor,
%primals_79.1 : Tensor,
%primals_80.1 : Tensor,
%primals_81.1 : Tensor,
%primals_82.1 : Tensor,
%primals_83.1 : Tensor,
%primals_84.1 : Tensor,
%primals_85.1 : Tensor,
%primals_86.1 : Tensor,
%primals_87.1 : Tensor,
%primals_88.1 : Tensor,
%primals_89.1 : Tensor,
%primals_90.1 : Tensor,
%primals_91.1 : Tensor,
%primals_92.1 : Tensor,
%primals_93.1 : Tensor,
%primals_94.1 : Tensor,
%primals_95.1 : Tensor,
%primals_96.1 : Tensor,
%primals_97.1 : Tensor,
%primals_98.1 : Tensor,
%primals_99.1 : Tensor,
%primals_100.1 : Tensor,
%primals_101.1 : Tensor,
%primals_102.1 : Tensor,
%primals_103.1 : Tensor,
%primals_104.1 : Tensor,
%primals_105.1 : Tensor,
%primals_106.1 : Tensor,
%primals_107.1 : Tensor,
%primals_108.1 : Tensor):
%1392 : Tensor = prim::Constant[value={6}]() # :0:0
%1347 : Tensor = prim::Constant[value={13}]() # :0:0
%1305 : Tensor = prim::Constant[value={1}]() # :0:0
%1302 : Tensor = prim::Constant[value={27}]() # :0:0
%1299 : Tensor = prim::Constant[value={2}]() # :0:0
%1288 : Tensor = prim::Constant[value=<Tensor>]() # :0:0
%1279 : Tensor = prim::Constant[value={8}]() # :0:0
%1263 : Tensor = prim::Constant[value=<Tensor>]() # :0:0
%1144 : Tensor = prim::Constant[value=<Tensor>]() # :0:0
%1115 : Tensor = prim::Constant[value=<Tensor>]() # :0:0
%1108 : Tensor = prim::Constant[value=<Tensor>]() # :0:0
%1101 : Tensor = prim::Constant[value=<Tensor>]() # :0:0
%1093 : Tensor = prim::Constant[value=<Tensor>]() # :0:0
%788 : int[] = prim::Constant[value=[0, 0]]()
%780 : Tensor = prim::Constant[value={11.3137}]() # :0:0
%137 : bool = prim::Constant[value=1]() # <eval_with_key>.3:9:89
%133 : NoneType = prim::Constant() # <eval_with_key>.3:9:65
%123 : bool = prim::Constant[value=0]() # <eval_with_key>.3:6:98
%117 : int = prim::Constant[value=2]() # :0:0
%119 : int = prim::Constant[value=3]() # :0:0
%121 : int = prim::Constant[value=1]() # :0:0
%124 : int = prim::Constant[value=0]() # :0:0
%128 : int = prim::Constant[value=64]() # :0:0
%129 : int = prim::Constant[value=112]() # :0:0
%138 : float = prim::Constant[value=0.10000000000000001]() # <eval_with_key>.3:9:95
%139 : float = prim::Constant[value=1.0000000000000001e-05]() # <eval_with_key>.3:9:100
%145 : int = prim::Constant[value=32]() # :0:0
%160 : int = prim::Constant[value=16]() # :0:0
%174 : int = prim::Constant[value=8]() # :0:0
%363 : int = prim::Constant[value=56]() # :0:0
%391 : int = prim::Constant[value=128]() # :0:0
%537 : int = prim::Constant[value=48]() # :0:0
%551 : int = prim::Constant[value=24]() # :0:0
%566 : int = prim::Constant[value=28]() # :0:0
%594 : int = prim::Constant[value=192]() # :0:0
%608 : int = prim::Constant[value=96]() # :0:0
%748 : int = prim::Constant[value=784]() # :0:0
%1097 : int = prim::Constant[value=-1]() # :0:0
%1272 : int = prim::Constant[value=224]() # :0:0
%111 : Tensor[] = prim::ListConstruct(%primals_107.1, %primals_108.1)
%cat.1 : Tensor = aten::cat(%111, %124) # <eval_with_key>.3:5:10
%118 : int[] = prim::ListConstruct(%117, %117)
%120 : int[] = prim::ListConstruct(%119, %119)
%122 : int[] = prim::ListConstruct(%121, %121)
%125 : int[] = prim::ListConstruct(%124, %124)
%convolution.1 : Tensor = aten::convolution(%cat.1, %primals_48.1, %primals_47.1, %118, %120, %122, %123, %125, %121) # <eval_with_key>.3:6:18
%130 : int[] = prim::ListConstruct(%121, %128, %129, %129)
%view.1 : Tensor = aten::view(%convolution.1, %130) # <eval_with_key>.3:7:11
%140 : Tensor, %141 : Tensor, %142 : Tensor = aten::native_batch_norm(%view.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:9:24
%146 : int[] = prim::ListConstruct(%117, %145, %129, %129)
%view_1.1 : Tensor = aten::view(%140, %146) # <eval_with_key>.3:11:13
%relu_.1 : Tensor = aten::relu_(%view_1.1) # <eval_with_key>.3:13:12
%153 : int[] = prim::ListConstruct(%121, %121)
%154 : int[] = prim::ListConstruct(%124, %124)
%155 : int[] = prim::ListConstruct(%121, %121)
%157 : int[] = prim::ListConstruct(%124, %124)
%convolution_1.1 : Tensor = aten::convolution(%relu_.1, %primals_50.1, %primals_49.1, %153, %154, %155, %123, %157, %121) # <eval_with_key>.3:14:20
%161 : int[] = prim::ListConstruct(%121, %160, %129, %129)
%view_2.1 : Tensor = aten::view(%convolution_1.1, %161) # <eval_with_key>.3:15:13
%169 : Tensor, %170 : Tensor, %171 : Tensor = aten::native_batch_norm(%view_2.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:17:26
%175 : int[] = prim::ListConstruct(%117, %174, %129, %129)
%view_3.1 : Tensor = aten::view(%169, %175) # <eval_with_key>.3:19:13
%relu__1.1 : Tensor = aten::relu_(%view_3.1) # <eval_with_key>.3:21:14
%182 : int[] = prim::ListConstruct(%121, %121)
%183 : int[] = prim::ListConstruct(%121, %121)
%184 : int[] = prim::ListConstruct(%121, %121)
%186 : int[] = prim::ListConstruct(%124, %124)
%convolution_2.1 : Tensor = aten::convolution(%relu__1.1, %primals_52.1, %primals_51.1, %182, %183, %184, %123, %186, %121) # <eval_with_key>.3:22:20
%189 : int[] = prim::ListConstruct(%121, %160, %129, %129)
%view_4.1 : Tensor = aten::view(%convolution_2.1, %189) # <eval_with_key>.3:23:13
%197 : Tensor, %198 : Tensor, %199 : Tensor = aten::native_batch_norm(%view_4.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:25:26
%202 : int[] = prim::ListConstruct(%117, %174, %129, %129)
%view_5.1 : Tensor = aten::view(%197, %202) # <eval_with_key>.3:27:13
%relu__2.1 : Tensor = aten::relu_(%view_5.1) # <eval_with_key>.3:29:14
%209 : int[] = prim::ListConstruct(%121, %121)
%210 : int[] = prim::ListConstruct(%124, %124)
%211 : int[] = prim::ListConstruct(%121, %121)
%213 : int[] = prim::ListConstruct(%124, %124)
%convolution_3.1 : Tensor = aten::convolution(%relu__2.1, %primals_54.1, %primals_53.1, %209, %210, %211, %123, %213, %121) # <eval_with_key>.3:30:20
%216 : int[] = prim::ListConstruct(%121, %128, %129, %129)
%view_6.1 : Tensor = aten::view(%convolution_3.1, %216) # <eval_with_key>.3:31:13
%224 : Tensor, %225 : Tensor, %226 : Tensor = aten::native_batch_norm(%view_6.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:33:26
%229 : int[] = prim::ListConstruct(%117, %145, %129, %129)
%view_7.1 : Tensor = aten::view(%224, %229) # <eval_with_key>.3:35:13
%relu__3.1 : Tensor = aten::relu_(%view_7.1) # <eval_with_key>.3:37:14
%add.1 : Tensor = aten::add(%relu_.1, %relu__3.1, %121) # <eval_with_key>.3:38:10
%relu__4.1 : Tensor = aten::relu_(%add.1) # <eval_with_key>.3:39:14
%242 : int[] = prim::ListConstruct(%121, %121)
%243 : int[] = prim::ListConstruct(%124, %124)
%244 : int[] = prim::ListConstruct(%121, %121)
%246 : int[] = prim::ListConstruct(%124, %124)
%convolution_4.1 : Tensor = aten::convolution(%relu__4.1, %primals_56.1, %primals_55.1, %242, %243, %244, %123, %246, %121) # <eval_with_key>.3:40:20
%249 : int[] = prim::ListConstruct(%121, %160, %129, %129)
%view_8.1 : Tensor = aten::view(%convolution_4.1, %249) # <eval_with_key>.3:41:13
%257 : Tensor, %258 : Tensor, %259 : Tensor = aten::native_batch_norm(%view_8.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:43:26
%262 : int[] = prim::ListConstruct(%117, %174, %129, %129)
%view_9.1 : Tensor = aten::view(%257, %262) # <eval_with_key>.3:45:13
%relu__5.1 : Tensor = aten::relu_(%view_9.1) # <eval_with_key>.3:47:14
%269 : int[] = prim::ListConstruct(%121, %121)
%270 : int[] = prim::ListConstruct(%121, %121)
%271 : int[] = prim::ListConstruct(%121, %121)
%273 : int[] = prim::ListConstruct(%124, %124)
%convolution_5.1 : Tensor = aten::convolution(%relu__5.1, %primals_58.1, %primals_57.1, %269, %270, %271, %123, %273, %121) # <eval_with_key>.3:48:20
%276 : int[] = prim::ListConstruct(%121, %160, %129, %129)
%view_10.1 : Tensor = aten::view(%convolution_5.1, %276) # <eval_with_key>.3:49:14
%284 : Tensor, %285 : Tensor, %286 : Tensor = aten::native_batch_norm(%view_10.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:51:26
%289 : int[] = prim::ListConstruct(%117, %174, %129, %129)
%view_11.1 : Tensor = aten::view(%284, %289) # <eval_with_key>.3:53:14
%relu__6.1 : Tensor = aten::relu_(%view_11.1) # <eval_with_key>.3:55:14
%296 : int[] = prim::ListConstruct(%121, %121)
%297 : int[] = prim::ListConstruct(%124, %124)
%298 : int[] = prim::ListConstruct(%121, %121)
%300 : int[] = prim::ListConstruct(%124, %124)
%convolution_6.1 : Tensor = aten::convolution(%relu__6.1, %primals_60.1, %primals_59.1, %296, %297, %298, %123, %300, %121) # <eval_with_key>.3:56:20
%303 : int[] = prim::ListConstruct(%121, %128, %129, %129)
%view_12.1 : Tensor = aten::view(%convolution_6.1, %303) # <eval_with_key>.3:57:14
%311 : Tensor, %312 : Tensor, %313 : Tensor = aten::native_batch_norm(%view_12.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:59:26
%316 : int[] = prim::ListConstruct(%117, %145, %129, %129)
%view_13.1 : Tensor = aten::view(%311, %316) # <eval_with_key>.3:61:14
%relu__7.1 : Tensor = aten::relu_(%view_13.1) # <eval_with_key>.3:63:14
%add_1.1 : Tensor = aten::add(%relu__4.1, %relu__7.1, %121) # <eval_with_key>.3:64:12
%relu__8.1 : Tensor = aten::relu_(%add_1.1) # <eval_with_key>.3:65:14
%329 : int[] = prim::ListConstruct(%121, %121)
%330 : int[] = prim::ListConstruct(%124, %124)
%331 : int[] = prim::ListConstruct(%121, %121)
%333 : int[] = prim::ListConstruct(%124, %124)
%convolution_7.1 : Tensor = aten::convolution(%relu__8.1, %primals_62.1, %primals_61.1, %329, %330, %331, %123, %333, %121) # <eval_with_key>.3:66:20
%336 : int[] = prim::ListConstruct(%121, %145, %129, %129)
%view_14.1 : Tensor = aten::view(%convolution_7.1, %336) # <eval_with_key>.3:67:14
%344 : Tensor, %345 : Tensor, %346 : Tensor = aten::native_batch_norm(%view_14.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:69:26
%349 : int[] = prim::ListConstruct(%117, %160, %129, %129)
%view_15.1 : Tensor = aten::view(%344, %349) # <eval_with_key>.3:71:14
%relu__9.1 : Tensor = aten::relu_(%view_15.1) # <eval_with_key>.3:73:14
%356 : int[] = prim::ListConstruct(%117, %117)
%357 : int[] = prim::ListConstruct(%121, %121)
%358 : int[] = prim::ListConstruct(%121, %121)
%360 : int[] = prim::ListConstruct(%124, %124)
%convolution_8.1 : Tensor = aten::convolution(%relu__9.1, %primals_64.1, %primals_63.1, %356, %357, %358, %123, %360, %121) # <eval_with_key>.3:74:20
%364 : int[] = prim::ListConstruct(%121, %145, %363, %363)
%view_16.1 : Tensor = aten::view(%convolution_8.1, %364) # <eval_with_key>.3:75:14
%372 : Tensor, %373 : Tensor, %374 : Tensor = aten::native_batch_norm(%view_16.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:77:26
%377 : int[] = prim::ListConstruct(%117, %160, %363, %363)
%view_17.1 : Tensor = aten::view(%372, %377) # <eval_with_key>.3:79:14
%relu__10.1 : Tensor = aten::relu_(%view_17.1) # <eval_with_key>.3:81:15
%384 : int[] = prim::ListConstruct(%121, %121)
%385 : int[] = prim::ListConstruct(%124, %124)
%386 : int[] = prim::ListConstruct(%121, %121)
%388 : int[] = prim::ListConstruct(%124, %124)
%convolution_9.1 : Tensor = aten::convolution(%relu__10.1, %primals_66.1, %primals_65.1, %384, %385, %386, %123, %388, %121) # <eval_with_key>.3:82:20
%392 : int[] = prim::ListConstruct(%121, %391, %363, %363)
%view_18.1 : Tensor = aten::view(%convolution_9.1, %392) # <eval_with_key>.3:83:14
%400 : Tensor, %401 : Tensor, %402 : Tensor = aten::native_batch_norm(%view_18.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:85:26
%405 : int[] = prim::ListConstruct(%117, %128, %363, %363)
%view_19.1 : Tensor = aten::view(%400, %405) # <eval_with_key>.3:87:14
%relu__11.1 : Tensor = aten::relu_(%view_19.1) # <eval_with_key>.3:89:15
%412 : int[] = prim::ListConstruct(%117, %117)
%413 : int[] = prim::ListConstruct(%124, %124)
%414 : int[] = prim::ListConstruct(%121, %121)
%416 : int[] = prim::ListConstruct(%124, %124)
%convolution_10.1 : Tensor = aten::convolution(%relu__8.1, %primals_68.1, %primals_67.1, %412, %413, %414, %123, %416, %121) # <eval_with_key>.3:90:21
%419 : int[] = prim::ListConstruct(%121, %391, %363, %363)
%view_20.1 : Tensor = aten::view(%convolution_10.1, %419) # <eval_with_key>.3:91:14
%427 : Tensor, %428 : Tensor, %429 : Tensor = aten::native_batch_norm(%view_20.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:93:27
%432 : int[] = prim::ListConstruct(%117, %128, %363, %363)
%view_21.1 : Tensor = aten::view(%427, %432) # <eval_with_key>.3:95:14
%add_2.1 : Tensor = aten::add(%view_21.1, %relu__11.1, %121) # <eval_with_key>.3:97:12
%relu__12.1 : Tensor = aten::relu_(%add_2.1) # <eval_with_key>.3:98:15
%443 : int[] = prim::ListConstruct(%121, %121)
%444 : int[] = prim::ListConstruct(%124, %124)
%445 : int[] = prim::ListConstruct(%121, %121)
%447 : int[] = prim::ListConstruct(%124, %124)
%convolution_11.1 : Tensor = aten::convolution(%relu__12.1, %primals_70.1, %primals_69.1, %443, %444, %445, %123, %447, %121) # <eval_with_key>.3:99:21
%450 : int[] = prim::ListConstruct(%121, %145, %363, %363)
%view_22.1 : Tensor = aten::view(%convolution_11.1, %450) # <eval_with_key>.3:100:14
%458 : Tensor, %459 : Tensor, %460 : Tensor = aten::native_batch_norm(%view_22.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:102:27
%463 : int[] = prim::ListConstruct(%117, %160, %363, %363)
%view_23.1 : Tensor = aten::view(%458, %463) # <eval_with_key>.3:104:14
%relu__13.1 : Tensor = aten::relu_(%view_23.1) # <eval_with_key>.3:106:15
%470 : int[] = prim::ListConstruct(%121, %121)
%471 : int[] = prim::ListConstruct(%121, %121)
%472 : int[] = prim::ListConstruct(%121, %121)
%474 : int[] = prim::ListConstruct(%124, %124)
%convolution_12.1 : Tensor = aten::convolution(%relu__13.1, %primals_72.1, %primals_71.1, %470, %471, %472, %123, %474, %121) # <eval_with_key>.3:107:21
%477 : int[] = prim::ListConstruct(%121, %145, %363, %363)
%view_24.1 : Tensor = aten::view(%convolution_12.1, %477) # <eval_with_key>.3:108:14
%485 : Tensor, %486 : Tensor, %487 : Tensor = aten::native_batch_norm(%view_24.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:110:27
%490 : int[] = prim::ListConstruct(%117, %160, %363, %363)
%view_25.1 : Tensor = aten::view(%485, %490) # <eval_with_key>.3:112:14
%relu__14.1 : Tensor = aten::relu_(%view_25.1) # <eval_with_key>.3:114:15
%497 : int[] = prim::ListConstruct(%121, %121)
%498 : int[] = prim::ListConstruct(%124, %124)
%499 : int[] = prim::ListConstruct(%121, %121)
%501 : int[] = prim::ListConstruct(%124, %124)
%convolution_13.1 : Tensor = aten::convolution(%relu__14.1, %primals_74.1, %primals_73.1, %497, %498, %499, %123, %501, %121) # <eval_with_key>.3:115:21
%504 : int[] = prim::ListConstruct(%121, %391, %363, %363)
%view_26.1 : Tensor = aten::view(%convolution_13.1, %504) # <eval_with_key>.3:116:14
%512 : Tensor, %513 : Tensor, %514 : Tensor = aten::native_batch_norm(%view_26.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:118:27
%517 : int[] = prim::ListConstruct(%117, %128, %363, %363)
%view_27.1 : Tensor = aten::view(%512, %517) # <eval_with_key>.3:120:14
%relu__15.1 : Tensor = aten::relu_(%view_27.1) # <eval_with_key>.3:122:15
%add_3.1 : Tensor = aten::add(%relu__12.1, %relu__15.1, %121) # <eval_with_key>.3:123:12
%relu__16.1 : Tensor = aten::relu_(%add_3.1) # <eval_with_key>.3:124:15
%530 : int[] = prim::ListConstruct(%121, %121)
%531 : int[] = prim::ListConstruct(%124, %124)
%532 : int[] = prim::ListConstruct(%121, %121)
%534 : int[] = prim::ListConstruct(%124, %124)
%convolution_14.1 : Tensor = aten::convolution(%relu__16.1, %primals_76.1, %primals_75.1, %530, %531, %532, %123, %534, %121) # <eval_with_key>.3:125:21
%538 : int[] = prim::ListConstruct(%121, %537, %363, %363)
%view_28.1 : Tensor = aten::view(%convolution_14.1, %538) # <eval_with_key>.3:126:14
%546 : Tensor, %547 : Tensor, %548 : Tensor = aten::native_batch_norm(%view_28.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:128:27
%552 : int[] = prim::ListConstruct(%117, %551, %363, %363)
%view_29.1 : Tensor = aten::view(%546, %552) # <eval_with_key>.3:130:14
%relu__17.1 : Tensor = aten::relu_(%view_29.1) # <eval_with_key>.3:132:15
%559 : int[] = prim::ListConstruct(%117, %117)
%560 : int[] = prim::ListConstruct(%121, %121)
%561 : int[] = prim::ListConstruct(%121, %121)
%563 : int[] = prim::ListConstruct(%124, %124)
%convolution_15.1 : Tensor = aten::convolution(%relu__17.1, %primals_78.1, %primals_77.1, %559, %560, %561, %123, %563, %121) # <eval_with_key>.3:133:21
%567 : int[] = prim::ListConstruct(%121, %537, %566, %566)
%view_30.1 : Tensor = aten::view(%convolution_15.1, %567) # <eval_with_key>.3:134:14
%575 : Tensor, %576 : Tensor, %577 : Tensor = aten::native_batch_norm(%view_30.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:136:27
%580 : int[] = prim::ListConstruct(%117, %551, %566, %566)
%view_31.1 : Tensor = aten::view(%575, %580) # <eval_with_key>.3:138:14
%relu__18.1 : Tensor = aten::relu_(%view_31.1) # <eval_with_key>.3:140:15
%587 : int[] = prim::ListConstruct(%121, %121)
%588 : int[] = prim::ListConstruct(%124, %124)
%589 : int[] = prim::ListConstruct(%121, %121)
%591 : int[] = prim::ListConstruct(%124, %124)
%convolution_16.1 : Tensor = aten::convolution(%relu__18.1, %primals_80.1, %primals_79.1, %587, %588, %589, %123, %591, %121) # <eval_with_key>.3:141:21
%595 : int[] = prim::ListConstruct(%121, %594, %566, %566)
%view_32.1 : Tensor = aten::view(%convolution_16.1, %595) # <eval_with_key>.3:142:14
%603 : Tensor, %604 : Tensor, %605 : Tensor = aten::native_batch_norm(%view_32.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:144:27
%609 : int[] = prim::ListConstruct(%117, %608, %566, %566)
%view_33.1 : Tensor = aten::view(%603, %609) # <eval_with_key>.3:146:14
%relu__19.1 : Tensor = aten::relu_(%view_33.1) # <eval_with_key>.3:148:15
%616 : int[] = prim::ListConstruct(%117, %117)
%617 : int[] = prim::ListConstruct(%124, %124)
%618 : int[] = prim::ListConstruct(%121, %121)
%620 : int[] = prim::ListConstruct(%124, %124)
%convolution_17.1 : Tensor = aten::convolution(%relu__16.1, %primals_82.1, %primals_81.1, %616, %617, %618, %123, %620, %121) # <eval_with_key>.3:149:21
%623 : int[] = prim::ListConstruct(%121, %594, %566, %566)
%view_34.1 : Tensor = aten::view(%convolution_17.1, %623) # <eval_with_key>.3:150:14
%631 : Tensor, %632 : Tensor, %633 : Tensor = aten::native_batch_norm(%view_34.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:152:27
%636 : int[] = prim::ListConstruct(%117, %608, %566, %566)
%view_35.1 : Tensor = aten::view(%631, %636) # <eval_with_key>.3:154:14
%add_4.1 : Tensor = aten::add(%view_35.1, %relu__19.1, %121) # <eval_with_key>.3:156:12
%relu__20.1 : Tensor = aten::relu_(%add_4.1) # <eval_with_key>.3:157:15
%647 : int[] = prim::ListConstruct(%121, %121)
%648 : int[] = prim::ListConstruct(%124, %124)
%649 : int[] = prim::ListConstruct(%121, %121)
%651 : int[] = prim::ListConstruct(%124, %124)
%convolution_18.1 : Tensor = aten::convolution(%relu__20.1, %primals_84.1, %primals_83.1, %647, %648, %649, %123, %651, %121) # <eval_with_key>.3:158:21
%654 : int[] = prim::ListConstruct(%121, %537, %566, %566)
%view_36.1 : Tensor = aten::view(%convolution_18.1, %654) # <eval_with_key>.3:159:14
%662 : Tensor, %663 : Tensor, %664 : Tensor = aten::native_batch_norm(%view_36.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:161:27
%667 : int[] = prim::ListConstruct(%117, %551, %566, %566)
%view_37.1 : Tensor = aten::view(%662, %667) # <eval_with_key>.3:163:14
%relu__21.1 : Tensor = aten::relu_(%view_37.1) # <eval_with_key>.3:165:15
%674 : int[] = prim::ListConstruct(%121, %121)
%675 : int[] = prim::ListConstruct(%121, %121)
%676 : int[] = prim::ListConstruct(%121, %121)
%678 : int[] = prim::ListConstruct(%124, %124)
%convolution_19.1 : Tensor = aten::convolution(%relu__21.1, %primals_86.1, %primals_85.1, %674, %675, %676, %123, %678, %121) # <eval_with_key>.3:166:21
%681 : int[] = prim::ListConstruct(%121, %537, %566, %566)
%view_38.1 : Tensor = aten::view(%convolution_19.1, %681) # <eval_with_key>.3:167:14
%689 : Tensor, %690 : Tensor, %691 : Tensor = aten::native_batch_norm(%view_38.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:169:27
%694 : int[] = prim::ListConstruct(%117, %551, %566, %566)
%view_39.1 : Tensor = aten::view(%689, %694) # <eval_with_key>.3:171:14
%relu__22.1 : Tensor = aten::relu_(%view_39.1) # <eval_with_key>.3:173:15
%701 : int[] = prim::ListConstruct(%121, %121)
%702 : int[] = prim::ListConstruct(%124, %124)
%703 : int[] = prim::ListConstruct(%121, %121)
%705 : int[] = prim::ListConstruct(%124, %124)
%convolution_20.1 : Tensor = aten::convolution(%relu__22.1, %primals_88.1, %primals_87.1, %701, %702, %703, %123, %705, %121) # <eval_with_key>.3:174:21
%708 : int[] = prim::ListConstruct(%121, %594, %566, %566)
%view_40.1 : Tensor = aten::view(%convolution_20.1, %708) # <eval_with_key>.3:175:14
%716 : Tensor, %717 : Tensor, %718 : Tensor = aten::native_batch_norm(%view_40.1, %133, %133, %133, %133, %137, %138, %139) # <eval_with_key>.3:177:27
%721 : int[] = prim::ListConstruct(%117, %608, %566, %566)
%view_41.1 : Tensor = aten::view(%716, %721) # <eval_with_key>.3:179:14
%relu__23.1 : Tensor = aten::relu_(%view_41.1) # <eval_with_key>.3:181:15
%add_5.1 : Tensor = aten::add(%relu__20.1, %relu__23.1, %121) # <eval_with_key>.3:182:12
%relu__24.1 : Tensor = aten::relu_(%add_5.1) # <eval_with_key>.3:183:15
%734 : int[] = prim::ListConstruct(%121, %121)
%735 : int[] = prim::ListConstruct(%124, %124)
%736 : int[] = prim::ListConstruct(%121, %121)
%738 : int[] = prim::ListConstruct(%124, %124)
%convolution_21.1 : Tensor = aten::convolution(%relu__24.1, %primals_46.1, %primals_45.1, %734, %735, %736, %123, %738, %121) # <eval_with_key>.3:184:21
%split.1 : Tensor[] = aten::split(%convolution_21.1, %121, %124) # <eval_with_key>.3:185:12
%getitem_63.1 : Tensor = aten::__getitem__(%split.1, %124) # <eval_with_key>.3:186:17
%getitem_64.1 : Tensor = aten::__getitem__(%split.1, %121) # <eval_with_key>.3:187:17
%749 : int[] = prim::ListConstruct(%121, %391, %748)
%view_42.1 : Tensor = aten::view(%getitem_63.1, %749) # <eval_with_key>.3:190:14
%752 : int[] = prim::ListConstruct(%121, %391, %748)
%view_43.1 : Tensor = aten::view(%getitem_64.1, %752) # <eval_with_key>.3:192:14
%transpose.1 : Tensor = aten::transpose(%view_42.1, %121, %117) # <eval_with_key>.3:194:16
%757 : int[] = prim::ListConstruct(%121, %748, %391)
%expand.1 : Tensor = aten::expand(%transpose.1, %757, %123) # <eval_with_key>.3:196:13
%761 : int[] = prim::ListConstruct(%121, %748, %391)
%view_44.1 : Tensor = aten::view(%expand.1, %761) # <eval_with_key>.3:198:14
%764 : int[] = prim::ListConstruct(%121, %391, %748)
%expand_1.1 : Tensor = aten::expand(%view_43.1, %764, %123) # <eval_with_key>.3:201:15
%768 : int[] = prim::ListConstruct(%121, %391, %748)
%view_45.1 : Tensor = aten::view(%expand_1.1, %768) # <eval_with_key>.3:203:14
%bmm.1 : Tensor = aten::bmm(%view_44.1, %view_45.1) # <eval_with_key>.3:206:10
%774 : int[] = prim::ListConstruct(%121, %748, %748)
%_unsafe_view.1 : Tensor = aten::_unsafe_view(%bmm.1, %774) # <eval_with_key>.3:207:19
%777 : int[] = prim::ListConstruct(%121, %566, %566, %121, %566, %566)
%view_46.1 : Tensor = aten::view(%_unsafe_view.1, %777) # <eval_with_key>.3:208:14
%div.1 : Tensor = aten::div(%view_46.1, %780) # <eval_with_key>.3:211:10
%783 : int[] = prim::ListConstruct(%748, %121, %566, %566)
%view_47.1 : Tensor = aten::view(%div.1, %783) # <eval_with_key>.3:212:14
%786 : int[] = prim::ListConstruct(%117, %117)
%787 : int[] = prim::ListConstruct(%117, %117)
%avg_pool2d.1 : Tensor = aten::avg_pool2d(%view_47.1, %786, %787, %788, %123, %137, %133) # <eval_with_key>.3:215:17
%794 : int[] = prim::ListConstruct(%117, %117)
%795 : int[] = prim::ListConstruct(%117, %117)
%avg_pool2d_1.1 : Tensor = aten::avg_pool2d(%avg_pool2d.1, %794, %795, %788, %123, %137, %133) # <eval_with_key>.3:216:19
%802 : int[] = prim::ListConstruct(%117, %117)
%803 : int[] = prim::ListConstruct(%117, %117)
%avg_pool2d_2.1 : Tensor = aten::avg_pool2d(%avg_pool2d_1.1, %802, %803, %788, %123, %137, %133) # <eval_with_key>.3:217:19
%812 : int[] = prim::ListConstruct(%117, %117)
%813 : int[] = prim::ListConstruct(%119, %119)
%814 : int[] = prim::ListConstruct(%121, %121)
%816 : int[] = prim::ListConstruct(%124, %124)
%convolution_22.1 : Tensor = aten::convolution(%primals_107.1, %primals_4.1, %primals_3.1, %812, %813, %814, %123, %816, %121) # <eval_with_key>.3:218:21
%relu__25.1 : Tensor = aten::relu_(%convolution_22.1) # <eval_with_key>.3:219:15
%823 : int[] = prim::ListConstruct(%121, %121)
%824 : int[] = prim::ListConstruct(%124, %124)
%825 : int[] = prim::ListConstruct(%121, %121)
%827 : int[] = prim::ListConstruct(%124, %124)
%convolution_23.1 : Tensor = aten::convolution(%relu__25.1, %primals_6.1, %primals_5.1, %823, %824, %825, %123, %827, %121) # <eval_with_key>.3:220:21
%relu__26.1 : Tensor = aten::relu_(%convolution_23.1) # <eval_with_key>.3:221:15
%834 : int[] = prim::ListConstruct(%121, %121)
%835 : int[] = prim::ListConstruct(%121, %121)
%836 : int[] = prim::ListConstruct(%121, %121)
%838 : int[] = prim::ListConstruct(%124, %124)
%convolution_24.1 : Tensor = aten::convolution(%relu__26.1, %primals_8.1, %primals_7.1, %834, %835, %836, %123, %838, %121) # <eval_with_key>.3:222:21
%relu__27.1 : Tensor = aten::relu_(%convolution_24.1) # <eval_with_key>.3:223:15
%845 : int[] = prim::ListConstruct(%121, %121)
%846 : int[] = prim::ListConstruct(%124, %124)
%847 : int[] = prim::ListConstruct(%121, %121)
%849 : int[] = prim::ListConstruct(%124, %124)
%convolution_25.1 : Tensor = aten::convolution(%relu__27.1, %primals_10.1, %primals_9.1, %845, %846, %847, %123, %849, %121) # <eval_with_key>.3:224:21
%relu__28.1 : Tensor = aten::relu_(%convolution_25.1) # <eval_with_key>.3:225:15
%add_6.1 : Tensor = aten::add(%relu__25.1, %relu__28.1, %121) # <eval_with_key>.3:226:12
%relu__29.1 : Tensor = aten::relu_(%add_6.1) # <eval_with_key>.3:227:15
%862 : int[] = prim::ListConstruct(%121, %121)
%863 : int[] = prim::ListConstruct(%124, %124)
%864 : int[] = prim::ListConstruct(%121, %121)
%866 : int[] = prim::ListConstruct(%124, %124)
%convolution_26.1 : Tensor = aten::convolution(%relu__29.1, %primals_12.1, %primals_11.1, %862, %863, %864, %123, %866, %121) # <eval_with_key>.3:228:21
%relu__30.1 : Tensor = aten::relu_(%convolution_26.1) # <eval_with_key>.3:229:15
%873 : int[] = prim::ListConstruct(%121, %121)
%874 : int[] = prim::ListConstruct(%121, %121)
%875 : int[] = prim::ListConstruct(%121, %121)
%877 : int[] = prim::ListConstruct(%124, %124)
%convolution_27.1 : Tensor = aten::convolution(%relu__30.1, %primals_14.1, %primals_13.1, %873, %874, %875, %123, %877, %121) # <eval_with_key>.3:230:21
%relu__31.1 : Tensor = aten::relu_(%convolution_27.1) # <eval_with_key>.3:231:15
%884 : int[] = prim::ListConstruct(%121, %121)
%885 : int[] = prim::ListConstruct(%124, %124)
%886 : int[] = prim::ListConstruct(%121, %121)
%888 : int[] = prim::ListConstruct(%124, %124)
%convolution_28.1 : Tensor = aten::convolution(%relu__31.1, %primals_16.1, %primals_15.1, %884, %885, %886, %123, %888, %121) # <eval_with_key>.3:232:21
%relu__32.1 : Tensor = aten::relu_(%convolution_28.1) # <eval_with_key>.3:233:15
%add_7.1 : Tensor = aten::add(%relu__29.1, %relu__32.1, %121) # <eval_with_key>.3:234:12
%relu__33.1 : Tensor = aten::relu_(%add_7.1) # <eval_with_key>.3:235:15
%901 : int[] = prim::ListConstruct(%121, %121)
%902 : int[] = prim::ListConstruct(%124, %124)
%903 : int[] = prim::ListConstruct(%121, %121)
%905 : int[] = prim::ListConstruct(%124, %124)
%convolution_29.1 : Tensor = aten::convolution(%relu__33.1, %primals_18.1, %primals_17.1, %901, %902, %903, %123, %905, %121) # <eval_with_key>.3:236:21
%relu__34.1 : Tensor = aten::relu_(%convolution_29.1) # <eval_with_key>.3:237:15
%912 : int[] = prim::ListConstruct(%117, %117)
%913 : int[] = prim::ListConstruct(%121, %121)
%914 : int[] = prim::ListConstruct(%121, %121)
%916 : int[] = prim::ListConstruct(%124, %124)
%convolution_30.1 : Tensor = aten::convolution(%relu__34.1, %primals_20.1, %primals_19.1, %912, %913, %914, %123, %916, %121) # <eval_with_key>.3:238:21
%relu__35.1 : Tensor = aten::relu_(%convolution_30.1) # <eval_with_key>.3:239:15
%923 : int[] = prim::ListConstruct(%121, %121)
%924 : int[] = prim::ListConstruct(%124, %124)
%925 : int[] = prim::ListConstruct(%121, %121)
%927 : int[] = prim::ListConstruct(%124, %124)
%convolution_31.1 : Tensor = aten::convolution(%relu__35.1, %primals_22.1, %primals_21.1, %923, %924, %925, %123, %927, %121) # <eval_with_key>.3:240:21
%relu__36.1 : Tensor = aten::relu_(%convolution_31.1) # <eval_with_key>.3:241:15
%934 : int[] = prim::ListConstruct(%117, %117)
%935 : int[] = prim::ListConstruct(%124, %124)
%936 : int[] = prim::ListConstruct(%121, %121)
%938 : int[] = prim::ListConstruct(%124, %124)
%convolution_32.1 : Tensor = aten::convolution(%relu__33.1, %primals_24.1, %primals_23.1, %934, %935, %936, %123, %938, %121) # <eval_with_key>.3:242:21
%add_8.1 : Tensor = aten::add(%convolution_32.1, %relu__36.1, %121) # <eval_with_key>.3:243:12
%relu__37.1 : Tensor = aten::relu_(%add_8.1) # <eval_with_key>.3:244:15
%949 : int[] = prim::ListConstruct(%121, %121)
%950 : int[] = prim::ListConstruct(%124, %124)
%951 : int[] = prim::ListConstruct(%121, %121)
%953 : int[] = prim::ListConstruct(%124, %124)
%convolution_33.1 : Tensor = aten::convolution(%relu__37.1, %primals_26.1, %primals_25.1, %949, %950, %951, %123, %953, %121) # <eval_with_key>.3:245:21
%relu__38.1 : Tensor = aten::relu_(%convolution_33.1) # <eval_with_key>.3:246:15
%960 : int[] = prim::ListConstruct(%121, %121)
%961 : int[] = prim::ListConstruct(%121, %121)
%962 : int[] = prim::ListConstruct(%121, %121)
%964 : int[] = prim::ListConstruct(%124, %124)
%convolution_34.1 : Tensor = aten::convolution(%relu__38.1, %primals_28.1, %primals_27.1, %960, %961, %962, %123, %964, %121) # <eval_with_key>.3:247:21
%relu__39.1 : Tensor = aten::relu_(%convolution_34.1) # <eval_with_key>.3:248:15
%971 : int[] = prim::ListConstruct(%121, %121)
%972 : int[] = prim::ListConstruct(%124, %124)
%973 : int[] = prim::ListConstruct(%121, %121)
%975 : int[] = prim::ListConstruct(%124, %124)
%convolution_35.1 : Tensor = aten::convolution(%relu__39.1, %primals_30.1, %primals_29.1, %971, %972, %973, %123, %975, %121) # <eval_with_key>.3:249:21
%relu__40.1 : Tensor = aten::relu_(%convolution_35.1) # <eval_with_key>.3:250:15
%add_9.1 : Tensor = aten::add(%relu__37.1, %relu__40.1, %121) # <eval_with_key>.3:251:12
%relu__41.1 : Tensor = aten::relu_(%add_9.1) # <eval_with_key>.3:252:15
%988 : int[] = prim::ListConstruct(%121, %121)
%989 : int[] = prim::ListConstruct(%124, %124)
%990 : int[] = prim::ListConstruct(%121, %121)
%992 : int[] = prim::ListConstruct(%124, %124)
%convolution_36.1 : Tensor = aten::convolution(%relu__41.1, %primals_32.1, %primals_31.1, %988, %989, %990, %123, %992, %121) # <eval_with_key>.3:253:21
%relu__42.1 : Tensor = aten::relu_(%convolution_36.1) # <eval_with_key>.3:254:15
%999 : int[] = prim::ListConstruct(%117, %117)
%1000 : int[] = prim::ListConstruct(%121, %121)
%1001 : int[] = prim::ListConstruct(%121, %121)
%1003 : int[] = prim::ListConstruct(%124, %124)
%convolution_37.1 : Tensor = aten::convolution(%relu__42.1, %primals_34.1, %primals_33.1, %999, %1000, %1001, %123, %1003, %121) # <eval_with_key>.3:255:21
%relu__43.1 : Tensor = aten::relu_(%convolution_37.1) # <eval_with_key>.3:256:15
%1010 : int[] = prim::ListConstruct(%121, %121)
%1011 : int[] = prim::ListConstruct(%124, %124)
%1012 : int[] = prim::ListConstruct(%121, %121)
%1014 : int[] = prim::ListConstruct(%124, %124)
%convolution_38.1 : Tensor = aten::convolution(%relu__43.1, %primals_36.1, %primals_35.1, %1010, %1011, %1012, %123, %1014, %121) # <eval_with_key>.3:257:21
%relu__44.1 : Tensor = aten::relu_(%convolution_38.1) # <eval_with_key>.3:258:15
%1021 : int[] = prim::ListConstruct(%117, %117)
%1022 : int[] = prim::ListConstruct(%124, %124)
%1023 : int[] = prim::ListConstruct(%121, %121)
%1025 : int[] = prim::ListConstruct(%124, %124)
%convolution_39.1 : Tensor = aten::convolution(%relu__41.1, %primals_38.1, %primals_37.1, %1021, %1022, %1023, %123, %1025, %121) # <eval_with_key>.3:259:21
%add_10.1 : Tensor = aten::add(%convolution_39.1, %relu__44.1, %121) # <eval_with_key>.3:260:13
%relu__45.1 : Tensor = aten::relu_(%add_10.1) # <eval_with_key>.3:261:15
%1036 : int[] = prim::ListConstruct(%121, %121)
%1037 : int[] = prim::ListConstruct(%124, %124)
%1038 : int[] = prim::ListConstruct(%121, %121)
%1040 : int[] = prim::ListConstruct(%124, %124)
%convolution_40.1 : Tensor = aten::convolution(%relu__45.1, %primals_40.1, %primals_39.1, %1036, %1037, %1038, %123, %1040, %121) # <eval_with_key>.3:262:21
%relu__46.1 : Tensor = aten::relu_(%convolution_40.1) # <eval_with_key>.3:263:15
%1047 : int[] = prim::ListConstruct(%121, %121)
%1048 : int[] = prim::ListConstruct(%121, %121)
%1049 : int[] = prim::ListConstruct(%121, %121)
%1051 : int[] = prim::ListConstruct(%124, %124)
%convolution_41.1 : Tensor = aten::convolution(%relu__46.1, %primals_42.1, %primals_41.1, %1047, %1048, %1049, %123, %1051, %121) # <eval_with_key>.3:264:21
%relu__47.1 : Tensor = aten::relu_(%convolution_41.1) # <eval_with_key>.3:265:15
%1058 : int[] = prim::ListConstruct(%121, %121)
%1059 : int[] = prim::ListConstruct(%124, %124)
%1060 : int[] = prim::ListConstruct(%121, %121)
%1062 : int[] = prim::ListConstruct(%124, %124)
%convolution_42.1 : Tensor = aten::convolution(%relu__47.1, %primals_44.1, %primals_43.1, %1058, %1059, %1060, %123, %1062, %121) # <eval_with_key>.3:266:21
%relu__48.1 : Tensor = aten::relu_(%convolution_42.1) # <eval_with_key>.3:267:15
%add_11.1 : Tensor = aten::add(%relu__45.1, %relu__48.1, %121) # <eval_with_key>.3:268:13
%relu__49.1 : Tensor = aten::relu_(%add_11.1) # <eval_with_key>.3:269:15
%1075 : int[] = prim::ListConstruct(%121, %121)
%1076 : int[] = prim::ListConstruct(%124, %124)
%1077 : int[] = prim::ListConstruct(%121, %121)
%1079 : int[] = prim::ListConstruct(%124, %124)
%convolution_43.1 : Tensor = aten::convolution(%relu__49.1, %primals_2.1, %primals_1.1, %1075, %1076, %1077, %123, %1079, %121) # <eval_with_key>.3:270:21
%1082 : int[] = prim::ListConstruct(%608, %128)
%split_with_sizes.1 : Tensor[] = aten::split_with_sizes(%convolution_43.1, %1082, %121) # <eval_with_key>.3:271:23
%getitem_65.1 : Tensor = aten::__getitem__(%split_with_sizes.1, %124) # <eval_with_key>.3:272:17
%getitem_66.1 : Tensor = aten::__getitem__(%split_with_sizes.1, %121) # <eval_with_key>.3:273:17
%tanh.1 : Tensor = aten::tanh(%getitem_65.1) # <eval_with_key>.3:276:11
%relu.1 : Tensor = aten::relu(%getitem_66.1) # <eval_with_key>.3:277:11
%grid_sampler_2d.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %1093, %124, %124, %137) # <eval_with_key>.3:279:22
%1098 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_48.1 : Tensor = aten::view(%grid_sampler_2d.1, %1098) # <eval_with_key>.3:280:14
%grid_sampler_2d_1.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %1101, %124, %124, %137) # <eval_with_key>.3:283:24
%1105 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_49.1 : Tensor = aten::view(%grid_sampler_2d_1.1, %1105) # <eval_with_key>.3:284:14
%grid_sampler_2d_2.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %1108, %124, %124, %137) # <eval_with_key>.3:287:24
%1112 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_50.1 : Tensor = aten::view(%grid_sampler_2d_2.1, %1112) # <eval_with_key>.3:288:14
%grid_sampler_2d_3.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %1115, %124, %124, %137) # <eval_with_key>.3:291:24
%1119 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_51.1 : Tensor = aten::view(%grid_sampler_2d_3.1, %1119) # <eval_with_key>.3:292:14
%1125 : Tensor[] = prim::ListConstruct(%view_48.1, %view_49.1, %view_50.1, %view_51.1)
%cat_1.1 : Tensor = aten::cat(%1125, %1097) # <eval_with_key>.3:294:12
%1129 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute.1 : Tensor = aten::permute(%cat_1.1, %1129) # <eval_with_key>.3:295:14
%clone.1 : Tensor = aten::clone(%permute.1, %124) # <eval_with_key>.3:297:12
%1136 : int[] = prim::ListConstruct(%121, %121)
%1137 : int[] = prim::ListConstruct(%124, %124)
%1138 : int[] = prim::ListConstruct(%121, %121)
%1140 : int[] = prim::ListConstruct(%124, %124)
%convolution_44.1 : Tensor = aten::convolution(%clone.1, %primals_96.1, %primals_95.1, %1136, %1137, %1138, %123, %1140, %121) # <eval_with_key>.3:298:21
%relu__50.1 : Tensor = aten::relu_(%convolution_44.1) # <eval_with_key>.3:299:15
%1147 : int[] = prim::ListConstruct(%121, %121)
%1148 : int[] = prim::ListConstruct(%119, %119)
%1149 : int[] = prim::ListConstruct(%121, %121)
%1151 : int[] = prim::ListConstruct(%124, %124)
%convolution_45.1 : Tensor = aten::convolution(%1144, %primals_98.1, %primals_97.1, %1147, %1148, %1149, %123, %1151, %121) # <eval_with_key>.3:301:21
%relu__51.1 : Tensor = aten::relu_(%convolution_45.1) # <eval_with_key>.3:302:15
%1158 : int[] = prim::ListConstruct(%121, %121)
%1159 : int[] = prim::ListConstruct(%121, %121)
%1160 : int[] = prim::ListConstruct(%121, %121)
%1162 : int[] = prim::ListConstruct(%124, %124)
%convolution_46.1 : Tensor = aten::convolution(%relu__51.1, %primals_100.1, %primals_99.1, %1158, %1159, %1160, %123, %1162, %121) # <eval_with_key>.3:303:21
%relu__52.1 : Tensor = aten::relu_(%convolution_46.1) # <eval_with_key>.3:304:15
%1168 : Tensor[] = prim::ListConstruct(%relu__50.1, %relu__52.1)
%cat_2.1 : Tensor = aten::cat(%1168, %121) # <eval_with_key>.3:305:12
%1173 : int[] = prim::ListConstruct(%121, %121)
%1174 : int[] = prim::ListConstruct(%121, %121)
%1175 : int[] = prim::ListConstruct(%121, %121)
%1177 : int[] = prim::ListConstruct(%124, %124)
%convolution_47.1 : Tensor = aten::convolution(%cat_2.1, %primals_94.1, %primals_93.1, %1173, %1174, %1175, %123, %1177, %121) # <eval_with_key>.3:306:21
%relu__53.1 : Tensor = aten::relu_(%convolution_47.1) # <eval_with_key>.3:307:15
%1184 : Tensor[] = prim::ListConstruct(%relu.1, %relu__53.1, %1144)
%cat_4.1 : Tensor = aten::cat(%1184, %121) # <eval_with_key>.3:310:12
%1188 : Tensor[] = prim::ListConstruct(%tanh.1, %cat_4.1)
%cat_5.1 : Tensor = aten::cat(%1188, %121) # <eval_with_key>.3:311:12
%1193 : int[] = prim::ListConstruct(%121, %121)
%1194 : int[] = prim::ListConstruct(%121, %121)
%1195 : int[] = prim::ListConstruct(%121, %121)
%1197 : int[] = prim::ListConstruct(%124, %124)
%convolution_48.1 : Tensor = aten::convolution(%cat_5.1, %primals_106.1, %primals_105.1, %1193, %1194, %1195, %123, %1197, %121) # <eval_with_key>.3:312:21
%sigmoid.1 : Tensor = aten::sigmoid(%convolution_48.1) # <eval_with_key>.3:313:14
%1204 : int[] = prim::ListConstruct(%121, %121)
%1205 : int[] = prim::ListConstruct(%121, %121)
%1206 : int[] = prim::ListConstruct(%121, %121)
%1208 : int[] = prim::ListConstruct(%124, %124)
%convolution_49.1 : Tensor = aten::convolution(%cat_5.1, %primals_104.1, %primals_103.1, %1204, %1205, %1206, %123, %1208, %121) # <eval_with_key>.3:314:21
%sigmoid_1.1 : Tensor = aten::sigmoid(%convolution_49.1) # <eval_with_key>.3:315:16
%mul.1 : Tensor = aten::mul(%sigmoid_1.1, %tanh.1) # <eval_with_key>.3:316:10
%1217 : Tensor[] = prim::ListConstruct(%mul.1, %cat_4.1)
%cat_6.1 : Tensor = aten::cat(%1217, %121) # <eval_with_key>.3:317:12
%1222 : int[] = prim::ListConstruct(%121, %121)
%1223 : int[] = prim::ListConstruct(%121, %121)
%1224 : int[] = prim::ListConstruct(%121, %121)
%1226 : int[] = prim::ListConstruct(%124, %124)
%convolution_50.1 : Tensor = aten::convolution(%cat_6.1, %primals_102.1, %primals_101.1, %1222, %1223, %1224, %123, %1226, %121) # <eval_with_key>.3:318:21
%tanh_1.1 : Tensor = aten::tanh(%convolution_50.1) # <eval_with_key>.3:319:13
%rsub.1 : Tensor = aten::rsub(%sigmoid.1, %121, %121) # <eval_with_key>.3:320:11
%mul_1.1 : Tensor = aten::mul(%rsub.1, %tanh.1) # <eval_with_key>.3:321:12
%mul_2.1 : Tensor = aten::mul(%sigmoid.1, %tanh_1.1) # <eval_with_key>.3:322:12
%add_12.1 : Tensor = aten::add(%mul_1.1, %mul_2.1, %121) # <eval_with_key>.3:323:13
%1246 : int[] = prim::ListConstruct(%121, %121)
%1247 : int[] = prim::ListConstruct(%121, %121)
%1248 : int[] = prim::ListConstruct(%121, %121)
%1250 : int[] = prim::ListConstruct(%124, %124)
%convolution_51.1 : Tensor = aten::convolution(%add_12.1, %primals_90.1, %primals_89.1, %1246, %1247, %1248, %123, %1250, %121) # <eval_with_key>.3:324:21
%relu__54.1 : Tensor = aten::relu_(%convolution_51.1) # <eval_with_key>.3:325:15
%1257 : int[] = prim::ListConstruct(%121, %121)
%1258 : int[] = prim::ListConstruct(%121, %121)
%1259 : int[] = prim::ListConstruct(%121, %121)
%1261 : int[] = prim::ListConstruct(%124, %124)
%convolution_52.1 : Tensor = aten::convolution(%relu__54.1, %primals_92.1, %primals_91.1, %1257, %1258, %1259, %123, %1261, %121) # <eval_with_key>.3:326:21
%add_13.1 : Tensor = aten::add(%1263, %convolution_52.1, %121) # <eval_with_key>.3:328:13
%sub.1 : Tensor = aten::sub(%add_13.1, %1263, %121) # <eval_with_key>.3:330:10
%1273 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d.1 : Tensor = aten::upsample_bilinear2d(%sub.1, %1273, %137, %133, %133) # <eval_with_key>.3:331:26
%mul_3.1 : Tensor = aten::mul(%upsample_bilinear2d.1, %1279) # <eval_with_key>.3:333:12
%1282 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_1.1 : Tensor = aten::permute(%add_13.1, %1282) # <eval_with_key>.3:336:16
%1285 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_52.1 : Tensor = aten::view(%permute_1.1, %1285) # <eval_with_key>.3:338:14
%add_14.1 : Tensor = aten::add(%view_52.1, %1288, %121) # <eval_with_key>.3:342:13
%1292 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_1.1 : Tensor[] = aten::split_with_sizes(%add_14.1, %1292, %1097) # <eval_with_key>.3:343:25
%getitem_67.1 : Tensor = aten::__getitem__(%split_with_sizes_1.1, %124) # <eval_with_key>.3:344:17
%getitem_68.1 : Tensor = aten::__getitem__(%split_with_sizes_1.1, %121) # <eval_with_key>.3:345:17
%mul_4.1 : Tensor = aten::mul(%getitem_67.1, %1299) # <eval_with_key>.3:349:12
%div_1.1 : Tensor = aten::div(%mul_4.1, %1302) # <eval_with_key>.3:351:12
%sub_1.1 : Tensor = aten::sub(%div_1.1, %1305, %121) # <eval_with_key>.3:353:12
%mul_5.1 : Tensor = aten::mul(%getitem_68.1, %1299) # <eval_with_key>.3:355:12
%div_2.1 : Tensor = aten::div(%mul_5.1, %1302) # <eval_with_key>.3:357:12
%sub_2.1 : Tensor = aten::sub(%div_2.1, %1305, %121) # <eval_with_key>.3:359:12
%1320 : Tensor[] = prim::ListConstruct(%sub_1.1, %sub_2.1)
%cat_7.1 : Tensor = aten::cat(%1320, %1097) # <eval_with_key>.3:360:12
%grid_sampler_2d_4.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_7.1, %124, %124, %137) # <eval_with_key>.3:361:24
%1327 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_53.1 : Tensor = aten::view(%grid_sampler_2d_4.1, %1327) # <eval_with_key>.3:362:14
%div_3.1 : Tensor = aten::div(%view_52.1, %1299) # <eval_with_key>.3:365:12
%add_15.1 : Tensor = aten::add(%div_3.1, %1288, %121) # <eval_with_key>.3:367:13
%1337 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_2.1 : Tensor[] = aten::split_with_sizes(%add_15.1, %1337, %1097) # <eval_with_key>.3:368:25
%getitem_69.1 : Tensor = aten::__getitem__(%split_with_sizes_2.1, %124) # <eval_with_key>.3:369:17
%getitem_70.1 : Tensor = aten::__getitem__(%split_with_sizes_2.1, %121) # <eval_with_key>.3:370:17
%mul_6.1 : Tensor = aten::mul(%getitem_69.1, %1299) # <eval_with_key>.3:374:12
%div_4.1 : Tensor = aten::div(%mul_6.1, %1347) # <eval_with_key>.3:376:12
%sub_3.1 : Tensor = aten::sub(%div_4.1, %1305, %121) # <eval_with_key>.3:378:12
%mul_7.1 : Tensor = aten::mul(%getitem_70.1, %1299) # <eval_with_key>.3:380:12
%div_5.1 : Tensor = aten::div(%mul_7.1, %1347) # <eval_with_key>.3:382:12
%sub_4.1 : Tensor = aten::sub(%div_5.1, %1305, %121) # <eval_with_key>.3:384:12
%1365 : Tensor[] = prim::ListConstruct(%sub_3.1, %sub_4.1)
%cat_8.1 : Tensor = aten::cat(%1365, %1097) # <eval_with_key>.3:385:12
%grid_sampler_2d_5.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_8.1, %124, %124, %137) # <eval_with_key>.3:386:24
%1372 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_54.1 : Tensor = aten::view(%grid_sampler_2d_5.1, %1372) # <eval_with_key>.3:387:14
%div_6.1 : Tensor = aten::div(%div_3.1, %1299) # <eval_with_key>.3:390:12
%add_16.1 : Tensor = aten::add(%div_6.1, %1288, %121) # <eval_with_key>.3:392:13
%1382 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_3.1 : Tensor[] = aten::split_with_sizes(%add_16.1, %1382, %1097) # <eval_with_key>.3:393:25
%getitem_71.1 : Tensor = aten::__getitem__(%split_with_sizes_3.1, %124) # <eval_with_key>.3:394:17
%getitem_72.1 : Tensor = aten::__getitem__(%split_with_sizes_3.1, %121) # <eval_with_key>.3:395:17
%mul_8.1 : Tensor = aten::mul(%getitem_71.1, %1299) # <eval_with_key>.3:399:12
%div_7.1 : Tensor = aten::div(%mul_8.1, %1392) # <eval_with_key>.3:401:12
%sub_5.1 : Tensor = aten::sub(%div_7.1, %1305, %121) # <eval_with_key>.3:403:12
%mul_9.1 : Tensor = aten::mul(%getitem_72.1, %1299) # <eval_with_key>.3:405:12
%div_8.1 : Tensor = aten::div(%mul_9.1, %1392) # <eval_with_key>.3:407:12
%sub_6.1 : Tensor = aten::sub(%div_8.1, %1305, %121) # <eval_with_key>.3:409:12
%1410 : Tensor[] = prim::ListConstruct(%sub_5.1, %sub_6.1)
%cat_9.1 : Tensor = aten::cat(%1410, %1097) # <eval_with_key>.3:410:12
%grid_sampler_2d_6.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_9.1, %124, %124, %137) # <eval_with_key>.3:411:24
%1417 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_55.1 : Tensor = aten::view(%grid_sampler_2d_6.1, %1417) # <eval_with_key>.3:412:14
%div_9.1 : Tensor = aten::div(%div_6.1, %1299) # <eval_with_key>.3:415:12
%add_17.1 : Tensor = aten::add(%div_9.1, %1288, %121) # <eval_with_key>.3:417:13
%1427 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_4.1 : Tensor[] = aten::split_with_sizes(%add_17.1, %1427, %1097) # <eval_with_key>.3:418:25
%getitem_73.1 : Tensor = aten::__getitem__(%split_with_sizes_4.1, %124) # <eval_with_key>.3:419:17
%getitem_74.1 : Tensor = aten::__getitem__(%split_with_sizes_4.1, %121) # <eval_with_key>.3:420:17
%mul_10.1 : Tensor = aten::mul(%getitem_73.1, %1299) # <eval_with_key>.3:424:13
%div_10.1 : Tensor = aten::div(%mul_10.1, %1299) # <eval_with_key>.3:426:13
%sub_7.1 : Tensor = aten::sub(%div_10.1, %1305, %121) # <eval_with_key>.3:428:12
%mul_11.1 : Tensor = aten::mul(%getitem_74.1, %1299) # <eval_with_key>.3:430:13
%div_11.1 : Tensor = aten::div(%mul_11.1, %1299) # <eval_with_key>.3:432:13
%sub_8.1 : Tensor = aten::sub(%div_11.1, %1305, %121) # <eval_with_key>.3:434:12
%1455 : Tensor[] = prim::ListConstruct(%sub_7.1, %sub_8.1)
%cat_10.1 : Tensor = aten::cat(%1455, %1097) # <eval_with_key>.3:435:13
%grid_sampler_2d_7.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_10.1, %124, %124, %137) # <eval_with_key>.3:436:24
%1462 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_56.1 : Tensor = aten::view(%grid_sampler_2d_7.1, %1462) # <eval_with_key>.3:437:14
%1468 : Tensor[] = prim::ListConstruct(%view_53.1, %view_54.1, %view_55.1, %view_56.1)
%cat_11.1 : Tensor = aten::cat(%1468, %1097) # <eval_with_key>.3:439:13
%1472 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_2.1 : Tensor = aten::permute(%cat_11.1, %1472) # <eval_with_key>.3:440:16
%clone_1.1 : Tensor = aten::clone(%permute_2.1, %124) # <eval_with_key>.3:442:14
%sub_9.1 : Tensor = aten::sub(%add_13.1, %1263, %121) # <eval_with_key>.3:444:12
%1483 : int[] = prim::ListConstruct(%121, %121)
%1484 : int[] = prim::ListConstruct(%124, %124)
%1485 : int[] = prim::ListConstruct(%121, %121)
%1487 : int[] = prim::ListConstruct(%124, %124)
%convolution_53.1 : Tensor = aten::convolution(%clone_1.1, %primals_96.1, %primals_95.1, %1483, %1484, %1485, %123, %1487, %121) # <eval_with_key>.3:445:21
%relu__55.1 : Tensor = aten::relu_(%convolution_53.1) # <eval_with_key>.3:446:15
%1494 : int[] = prim::ListConstruct(%121, %121)
%1495 : int[] = prim::ListConstruct(%119, %119)
%1496 : int[] = prim::ListConstruct(%121, %121)
%1498 : int[] = prim::ListConstruct(%124, %124)
%convolution_54.1 : Tensor = aten::convolution(%sub_9.1, %primals_98.1, %primals_97.1, %1494, %1495, %1496, %123, %1498, %121) # <eval_with_key>.3:447:21
%relu__56.1 : Tensor = aten::relu_(%convolution_54.1) # <eval_with_key>.3:448:15
%1505 : int[] = prim::ListConstruct(%121, %121)
%1506 : int[] = prim::ListConstruct(%121, %121)
%1507 : int[] = prim::ListConstruct(%121, %121)
%1509 : int[] = prim::ListConstruct(%124, %124)
%convolution_55.1 : Tensor = aten::convolution(%relu__56.1, %primals_100.1, %primals_99.1, %1505, %1506, %1507, %123, %1509, %121) # <eval_with_key>.3:449:21
%relu__57.1 : Tensor = aten::relu_(%convolution_55.1) # <eval_with_key>.3:450:15
%1515 : Tensor[] = prim::ListConstruct(%relu__55.1, %relu__57.1)
%cat_12.1 : Tensor = aten::cat(%1515, %121) # <eval_with_key>.3:451:13
%1520 : int[] = prim::ListConstruct(%121, %121)
%1521 : int[] = prim::ListConstruct(%121, %121)
%1522 : int[] = prim::ListConstruct(%121, %121)
%1524 : int[] = prim::ListConstruct(%124, %124)
%convolution_56.1 : Tensor = aten::convolution(%cat_12.1, %primals_94.1, %primals_93.1, %1520, %1521, %1522, %123, %1524, %121) # <eval_with_key>.3:452:21
%relu__58.1 : Tensor = aten::relu_(%convolution_56.1) # <eval_with_key>.3:453:15
%1531 : Tensor[] = prim::ListConstruct(%relu.1, %relu__58.1, %sub_9.1)
%cat_14.1 : Tensor = aten::cat(%1531, %121) # <eval_with_key>.3:455:13
%1535 : Tensor[] = prim::ListConstruct(%add_12.1, %cat_14.1)
%cat_15.1 : Tensor = aten::cat(%1535, %121) # <eval_with_key>.3:456:13
%1540 : int[] = prim::ListConstruct(%121, %121)
%1541 : int[] = prim::ListConstruct(%121, %121)
%1542 : int[] = prim::ListConstruct(%121, %121)
%1544 : int[] = prim::ListConstruct(%124, %124)
%convolution_57.1 : Tensor = aten::convolution(%cat_15.1, %primals_106.1, %primals_105.1, %1540, %1541, %1542, %123, %1544, %121) # <eval_with_key>.3:457:21
%sigmoid_2.1 : Tensor = aten::sigmoid(%convolution_57.1) # <eval_with_key>.3:458:16
%1551 : int[] = prim::ListConstruct(%121, %121)
%1552 : int[] = prim::ListConstruct(%121, %121)
%1553 : int[] = prim::ListConstruct(%121, %121)
%1555 : int[] = prim::ListConstruct(%124, %124)
%convolution_58.1 : Tensor = aten::convolution(%cat_15.1, %primals_104.1, %primals_103.1, %1551, %1552, %1553, %123, %1555, %121) # <eval_with_key>.3:459:21
%sigmoid_3.1 : Tensor = aten::sigmoid(%convolution_58.1) # <eval_with_key>.3:460:16
%mul_12.1 : Tensor = aten::mul(%sigmoid_3.1, %add_12.1) # <eval_with_key>.3:461:13
%1564 : Tensor[] = prim::ListConstruct(%mul_12.1, %cat_14.1)
%cat_16.1 : Tensor = aten::cat(%1564, %121) # <eval_with_key>.3:462:13
%1569 : int[] = prim::ListConstruct(%121, %121)
%1570 : int[] = prim::ListConstruct(%121, %121)
%1571 : int[] = prim::ListConstruct(%121, %121)
%1573 : int[] = prim::ListConstruct(%124, %124)
%convolution_59.1 : Tensor = aten::convolution(%cat_16.1, %primals_102.1, %primals_101.1, %1569, %1570, %1571, %123, %1573, %121) # <eval_with_key>.3:463:21
%tanh_2.1 : Tensor = aten::tanh(%convolution_59.1) # <eval_with_key>.3:464:13
%rsub_1.1 : Tensor = aten::rsub(%sigmoid_2.1, %121, %121) # <eval_with_key>.3:465:13
%mul_13.1 : Tensor = aten::mul(%rsub_1.1, %add_12.1) # <eval_with_key>.3:466:13
%mul_14.1 : Tensor = aten::mul(%sigmoid_2.1, %tanh_2.1) # <eval_with_key>.3:467:13
%add_18.1 : Tensor = aten::add(%mul_13.1, %mul_14.1, %121) # <eval_with_key>.3:468:13
%1593 : int[] = prim::ListConstruct(%121, %121)
%1594 : int[] = prim::ListConstruct(%121, %121)
%1595 : int[] = prim::ListConstruct(%121, %121)
%1597 : int[] = prim::ListConstruct(%124, %124)
%convolution_60.1 : Tensor = aten::convolution(%add_18.1, %primals_90.1, %primals_89.1, %1593, %1594, %1595, %123, %1597, %121) # <eval_with_key>.3:469:21
%relu__59.1 : Tensor = aten::relu_(%convolution_60.1) # <eval_with_key>.3:470:15
%1604 : int[] = prim::ListConstruct(%121, %121)
%1605 : int[] = prim::ListConstruct(%121, %121)
%1606 : int[] = prim::ListConstruct(%121, %121)
%1608 : int[] = prim::ListConstruct(%124, %124)
%convolution_61.1 : Tensor = aten::convolution(%relu__59.1, %primals_92.1, %primals_91.1, %1604, %1605, %1606, %123, %1608, %121) # <eval_with_key>.3:471:21
%add_19.1 : Tensor = aten::add(%add_13.1, %convolution_61.1, %121) # <eval_with_key>.3:472:13
%sub_10.1 : Tensor = aten::sub(%add_19.1, %1263, %121) # <eval_with_key>.3:474:13
%1619 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_1.1 : Tensor = aten::upsample_bilinear2d(%sub_10.1, %1619, %137, %133, %133) # <eval_with_key>.3:475:28
%mul_15.1 : Tensor = aten::mul(%upsample_bilinear2d_1.1, %1279) # <eval_with_key>.3:477:13
%1628 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_3.1 : Tensor = aten::permute(%add_19.1, %1628) # <eval_with_key>.3:480:16
%1631 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_57.1 : Tensor = aten::view(%permute_3.1, %1631) # <eval_with_key>.3:482:14
%add_20.1 : Tensor = aten::add(%view_57.1, %1288, %121) # <eval_with_key>.3:486:13
%1638 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_5.1 : Tensor[] = aten::split_with_sizes(%add_20.1, %1638, %1097) # <eval_with_key>.3:487:25
%getitem_75.1 : Tensor = aten::__getitem__(%split_with_sizes_5.1, %124) # <eval_with_key>.3:488:17
%getitem_76.1 : Tensor = aten::__getitem__(%split_with_sizes_5.1, %121) # <eval_with_key>.3:489:17
%mul_16.1 : Tensor = aten::mul(%getitem_75.1, %1299) # <eval_with_key>.3:493:13
%div_13.1 : Tensor = aten::div(%mul_16.1, %1302) # <eval_with_key>.3:495:13
%sub_11.1 : Tensor = aten::sub(%div_13.1, %1305, %121) # <eval_with_key>.3:497:13
%mul_17.1 : Tensor = aten::mul(%getitem_76.1, %1299) # <eval_with_key>.3:499:13
%div_14.1 : Tensor = aten::div(%mul_17.1, %1302) # <eval_with_key>.3:501:13
%sub_12.1 : Tensor = aten::sub(%div_14.1, %1305, %121) # <eval_with_key>.3:503:13
%1666 : Tensor[] = prim::ListConstruct(%sub_11.1, %sub_12.1)
%cat_17.1 : Tensor = aten::cat(%1666, %1097) # <eval_with_key>.3:504:13
%grid_sampler_2d_8.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_17.1, %124, %124, %137) # <eval_with_key>.3:505:24
%1673 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_58.1 : Tensor = aten::view(%grid_sampler_2d_8.1, %1673) # <eval_with_key>.3:506:14
%div_15.1 : Tensor = aten::div(%view_57.1, %1299) # <eval_with_key>.3:509:13
%add_21.1 : Tensor = aten::add(%div_15.1, %1288, %121) # <eval_with_key>.3:511:13
%1683 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_6.1 : Tensor[] = aten::split_with_sizes(%add_21.1, %1683, %1097) # <eval_with_key>.3:512:25
%getitem_77.1 : Tensor = aten::__getitem__(%split_with_sizes_6.1, %124) # <eval_with_key>.3:513:17
%getitem_78.1 : Tensor = aten::__getitem__(%split_with_sizes_6.1, %121) # <eval_with_key>.3:514:17
%mul_18.1 : Tensor = aten::mul(%getitem_77.1, %1299) # <eval_with_key>.3:518:13
%div_16.1 : Tensor = aten::div(%mul_18.1, %1347) # <eval_with_key>.3:520:13
%sub_13.1 : Tensor = aten::sub(%div_16.1, %1305, %121) # <eval_with_key>.3:522:13
%mul_19.1 : Tensor = aten::mul(%getitem_78.1, %1299) # <eval_with_key>.3:524:13
%div_17.1 : Tensor = aten::div(%mul_19.1, %1347) # <eval_with_key>.3:526:13
%sub_14.1 : Tensor = aten::sub(%div_17.1, %1305, %121) # <eval_with_key>.3:528:13
%1711 : Tensor[] = prim::ListConstruct(%sub_13.1, %sub_14.1)
%cat_18.1 : Tensor = aten::cat(%1711, %1097) # <eval_with_key>.3:529:13
%grid_sampler_2d_9.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_18.1, %124, %124, %137) # <eval_with_key>.3:530:24
%1718 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_59.1 : Tensor = aten::view(%grid_sampler_2d_9.1, %1718) # <eval_with_key>.3:531:14
%div_18.1 : Tensor = aten::div(%div_15.1, %1299) # <eval_with_key>.3:534:13
%add_22.1 : Tensor = aten::add(%div_18.1, %1288, %121) # <eval_with_key>.3:536:13
%1728 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_7.1 : Tensor[] = aten::split_with_sizes(%add_22.1, %1728, %1097) # <eval_with_key>.3:537:25
%getitem_79.1 : Tensor = aten::__getitem__(%split_with_sizes_7.1, %124) # <eval_with_key>.3:538:17
%getitem_80.1 : Tensor = aten::__getitem__(%split_with_sizes_7.1, %121) # <eval_with_key>.3:539:17
%mul_20.1 : Tensor = aten::mul(%getitem_79.1, %1299) # <eval_with_key>.3:543:13
%div_19.1 : Tensor = aten::div(%mul_20.1, %1392) # <eval_with_key>.3:545:13
%sub_15.1 : Tensor = aten::sub(%div_19.1, %1305, %121) # <eval_with_key>.3:547:13
%mul_21.1 : Tensor = aten::mul(%getitem_80.1, %1299) # <eval_with_key>.3:549:13
%div_20.1 : Tensor = aten::div(%mul_21.1, %1392) # <eval_with_key>.3:551:13
%sub_16.1 : Tensor = aten::sub(%div_20.1, %1305, %121) # <eval_with_key>.3:553:13
%1756 : Tensor[] = prim::ListConstruct(%sub_15.1, %sub_16.1)
%cat_19.1 : Tensor = aten::cat(%1756, %1097) # <eval_with_key>.3:554:13
%grid_sampler_2d_10.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_19.1, %124, %124, %137) # <eval_with_key>.3:555:25
%1763 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_60.1 : Tensor = aten::view(%grid_sampler_2d_10.1, %1763) # <eval_with_key>.3:556:14
%div_21.1 : Tensor = aten::div(%div_18.1, %1299) # <eval_with_key>.3:559:13
%add_23.1 : Tensor = aten::add(%div_21.1, %1288, %121) # <eval_with_key>.3:561:13
%1773 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_8.1 : Tensor[] = aten::split_with_sizes(%add_23.1, %1773, %1097) # <eval_with_key>.3:562:25
%getitem_81.1 : Tensor = aten::__getitem__(%split_with_sizes_8.1, %124) # <eval_with_key>.3:563:17
%getitem_82.1 : Tensor = aten::__getitem__(%split_with_sizes_8.1, %121) # <eval_with_key>.3:564:17
%mul_22.1 : Tensor = aten::mul(%getitem_81.1, %1299) # <eval_with_key>.3:568:13
%div_22.1 : Tensor = aten::div(%mul_22.1, %1299) # <eval_with_key>.3:570:13
%sub_17.1 : Tensor = aten::sub(%div_22.1, %1305, %121) # <eval_with_key>.3:572:13
%mul_23.1 : Tensor = aten::mul(%getitem_82.1, %1299) # <eval_with_key>.3:574:13
%div_23.1 : Tensor = aten::div(%mul_23.1, %1299) # <eval_with_key>.3:576:13
%sub_18.1 : Tensor = aten::sub(%div_23.1, %1305, %121) # <eval_with_key>.3:578:13
%1801 : Tensor[] = prim::ListConstruct(%sub_17.1, %sub_18.1)
%cat_20.1 : Tensor = aten::cat(%1801, %1097) # <eval_with_key>.3:579:13
%grid_sampler_2d_11.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_20.1, %124, %124, %137) # <eval_with_key>.3:580:25
%1808 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_61.1 : Tensor = aten::view(%grid_sampler_2d_11.1, %1808) # <eval_with_key>.3:581:14
%1814 : Tensor[] = prim::ListConstruct(%view_58.1, %view_59.1, %view_60.1, %view_61.1)
%cat_21.1 : Tensor = aten::cat(%1814, %1097) # <eval_with_key>.3:583:13
%1818 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_4.1 : Tensor = aten::permute(%cat_21.1, %1818) # <eval_with_key>.3:584:16
%clone_2.1 : Tensor = aten::clone(%permute_4.1, %124) # <eval_with_key>.3:586:14
%sub_19.1 : Tensor = aten::sub(%add_19.1, %1263, %121) # <eval_with_key>.3:588:13
%1829 : int[] = prim::ListConstruct(%121, %121)
%1830 : int[] = prim::ListConstruct(%124, %124)
%1831 : int[] = prim::ListConstruct(%121, %121)
%1833 : int[] = prim::ListConstruct(%124, %124)
%convolution_62.1 : Tensor = aten::convolution(%clone_2.1, %primals_96.1, %primals_95.1, %1829, %1830, %1831, %123, %1833, %121) # <eval_with_key>.3:589:21
%relu__60.1 : Tensor = aten::relu_(%convolution_62.1) # <eval_with_key>.3:590:15
%1840 : int[] = prim::ListConstruct(%121, %121)
%1841 : int[] = prim::ListConstruct(%119, %119)
%1842 : int[] = prim::ListConstruct(%121, %121)
%1844 : int[] = prim::ListConstruct(%124, %124)
%convolution_63.1 : Tensor = aten::convolution(%sub_19.1, %primals_98.1, %primals_97.1, %1840, %1841, %1842, %123, %1844, %121) # <eval_with_key>.3:591:21
%relu__61.1 : Tensor = aten::relu_(%convolution_63.1) # <eval_with_key>.3:592:15
%1851 : int[] = prim::ListConstruct(%121, %121)
%1852 : int[] = prim::ListConstruct(%121, %121)
%1853 : int[] = prim::ListConstruct(%121, %121)
%1855 : int[] = prim::ListConstruct(%124, %124)
%convolution_64.1 : Tensor = aten::convolution(%relu__61.1, %primals_100.1, %primals_99.1, %1851, %1852, %1853, %123, %1855, %121) # <eval_with_key>.3:593:21
%relu__62.1 : Tensor = aten::relu_(%convolution_64.1) # <eval_with_key>.3:594:15
%1861 : Tensor[] = prim::ListConstruct(%relu__60.1, %relu__62.1)
%cat_22.1 : Tensor = aten::cat(%1861, %121) # <eval_with_key>.3:595:13
%1866 : int[] = prim::ListConstruct(%121, %121)
%1867 : int[] = prim::ListConstruct(%121, %121)
%1868 : int[] = prim::ListConstruct(%121, %121)
%1870 : int[] = prim::ListConstruct(%124, %124)
%convolution_65.1 : Tensor = aten::convolution(%cat_22.1, %primals_94.1, %primals_93.1, %1866, %1867, %1868, %123, %1870, %121) # <eval_with_key>.3:596:21
%relu__63.1 : Tensor = aten::relu_(%convolution_65.1) # <eval_with_key>.3:597:15
%1877 : Tensor[] = prim::ListConstruct(%relu.1, %relu__63.1, %sub_19.1)
%cat_24.1 : Tensor = aten::cat(%1877, %121) # <eval_with_key>.3:599:13
%1881 : Tensor[] = prim::ListConstruct(%add_18.1, %cat_24.1)
%cat_25.1 : Tensor = aten::cat(%1881, %121) # <eval_with_key>.3:600:13
%1886 : int[] = prim::ListConstruct(%121, %121)
%1887 : int[] = prim::ListConstruct(%121, %121)
%1888 : int[] = prim::ListConstruct(%121, %121)
%1890 : int[] = prim::ListConstruct(%124, %124)
%convolution_66.1 : Tensor = aten::convolution(%cat_25.1, %primals_106.1, %primals_105.1, %1886, %1887, %1888, %123, %1890, %121) # <eval_with_key>.3:601:21
%sigmoid_4.1 : Tensor = aten::sigmoid(%convolution_66.1) # <eval_with_key>.3:602:16
%1897 : int[] = prim::ListConstruct(%121, %121)
%1898 : int[] = prim::ListConstruct(%121, %121)
%1899 : int[] = prim::ListConstruct(%121, %121)
%1901 : int[] = prim::ListConstruct(%124, %124)
%convolution_67.1 : Tensor = aten::convolution(%cat_25.1, %primals_104.1, %primals_103.1, %1897, %1898, %1899, %123, %1901, %121) # <eval_with_key>.3:603:21
%sigmoid_5.1 : Tensor = aten::sigmoid(%convolution_67.1) # <eval_with_key>.3:604:16
%mul_24.1 : Tensor = aten::mul(%sigmoid_5.1, %add_18.1) # <eval_with_key>.3:605:13
%1910 : Tensor[] = prim::ListConstruct(%mul_24.1, %cat_24.1)
%cat_26.1 : Tensor = aten::cat(%1910, %121) # <eval_with_key>.3:606:13
%1915 : int[] = prim::ListConstruct(%121, %121)
%1916 : int[] = prim::ListConstruct(%121, %121)
%1917 : int[] = prim::ListConstruct(%121, %121)
%1919 : int[] = prim::ListConstruct(%124, %124)
%convolution_68.1 : Tensor = aten::convolution(%cat_26.1, %primals_102.1, %primals_101.1, %1915, %1916, %1917, %123, %1919, %121) # <eval_with_key>.3:607:21
%tanh_3.1 : Tensor = aten::tanh(%convolution_68.1) # <eval_with_key>.3:608:13
%rsub_2.1 : Tensor = aten::rsub(%sigmoid_4.1, %121, %121) # <eval_with_key>.3:609:13
%mul_25.1 : Tensor = aten::mul(%rsub_2.1, %add_18.1) # <eval_with_key>.3:610:13
%mul_26.1 : Tensor = aten::mul(%sigmoid_4.1, %tanh_3.1) # <eval_with_key>.3:611:13
%add_24.1 : Tensor = aten::add(%mul_25.1, %mul_26.1, %121) # <eval_with_key>.3:612:13
%1939 : int[] = prim::ListConstruct(%121, %121)
%1940 : int[] = prim::ListConstruct(%121, %121)
%1941 : int[] = prim::ListConstruct(%121, %121)
%1943 : int[] = prim::ListConstruct(%124, %124)
%convolution_69.1 : Tensor = aten::convolution(%add_24.1, %primals_90.1, %primals_89.1, %1939, %1940, %1941, %123, %1943, %121) # <eval_with_key>.3:613:21
%relu__64.1 : Tensor = aten::relu_(%convolution_69.1) # <eval_with_key>.3:614:15
%1950 : int[] = prim::ListConstruct(%121, %121)
%1951 : int[] = prim::ListConstruct(%121, %121)
%1952 : int[] = prim::ListConstruct(%121, %121)
%1954 : int[] = prim::ListConstruct(%124, %124)
%convolution_70.1 : Tensor = aten::convolution(%relu__64.1, %primals_92.1, %primals_91.1, %1950, %1951, %1952, %123, %1954, %121) # <eval_with_key>.3:615:21
%add_25.1 : Tensor = aten::add(%add_19.1, %convolution_70.1, %121) # <eval_with_key>.3:616:13
%sub_20.1 : Tensor = aten::sub(%add_25.1, %1263, %121) # <eval_with_key>.3:618:13
%1965 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_2.1 : Tensor = aten::upsample_bilinear2d(%sub_20.1, %1965, %137, %133, %133) # <eval_with_key>.3:619:28
%mul_27.1 : Tensor = aten::mul(%upsample_bilinear2d_2.1, %1279) # <eval_with_key>.3:621:13
%1974 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_5.1 : Tensor = aten::permute(%add_25.1, %1974) # <eval_with_key>.3:624:16
%1977 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_62.1 : Tensor = aten::view(%permute_5.1, %1977) # <eval_with_key>.3:626:14
%add_26.1 : Tensor = aten::add(%view_62.1, %1288, %121) # <eval_with_key>.3:630:13
%1984 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_9.1 : Tensor[] = aten::split_with_sizes(%add_26.1, %1984, %1097) # <eval_with_key>.3:631:25
%getitem_83.1 : Tensor = aten::__getitem__(%split_with_sizes_9.1, %124) # <eval_with_key>.3:632:17
%getitem_84.1 : Tensor = aten::__getitem__(%split_with_sizes_9.1, %121) # <eval_with_key>.3:633:17
%mul_28.1 : Tensor = aten::mul(%getitem_83.1, %1299) # <eval_with_key>.3:637:13
%div_25.1 : Tensor = aten::div(%mul_28.1, %1302) # <eval_with_key>.3:639:13
%sub_21.1 : Tensor = aten::sub(%div_25.1, %1305, %121) # <eval_with_key>.3:641:13
%mul_29.1 : Tensor = aten::mul(%getitem_84.1, %1299) # <eval_with_key>.3:643:13
%div_26.1 : Tensor = aten::div(%mul_29.1, %1302) # <eval_with_key>.3:645:13
%sub_22.1 : Tensor = aten::sub(%div_26.1, %1305, %121) # <eval_with_key>.3:647:13
%2012 : Tensor[] = prim::ListConstruct(%sub_21.1, %sub_22.1)
%cat_27.1 : Tensor = aten::cat(%2012, %1097) # <eval_with_key>.3:648:13
%grid_sampler_2d_12.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_27.1, %124, %124, %137) # <eval_with_key>.3:649:25
%2019 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_63.1 : Tensor = aten::view(%grid_sampler_2d_12.1, %2019) # <eval_with_key>.3:650:14
%div_27.1 : Tensor = aten::div(%view_62.1, %1299) # <eval_with_key>.3:653:13
%add_27.1 : Tensor = aten::add(%div_27.1, %1288, %121) # <eval_with_key>.3:655:13
%2029 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_10.1 : Tensor[] = aten::split_with_sizes(%add_27.1, %2029, %1097) # <eval_with_key>.3:656:26
%getitem_85.1 : Tensor = aten::__getitem__(%split_with_sizes_10.1, %124) # <eval_with_key>.3:657:17
%getitem_86.1 : Tensor = aten::__getitem__(%split_with_sizes_10.1, %121) # <eval_with_key>.3:658:17
%mul_30.1 : Tensor = aten::mul(%getitem_85.1, %1299) # <eval_with_key>.3:662:13
%div_28.1 : Tensor = aten::div(%mul_30.1, %1347) # <eval_with_key>.3:664:13
%sub_23.1 : Tensor = aten::sub(%div_28.1, %1305, %121) # <eval_with_key>.3:666:13
%mul_31.1 : Tensor = aten::mul(%getitem_86.1, %1299) # <eval_with_key>.3:668:13
%div_29.1 : Tensor = aten::div(%mul_31.1, %1347) # <eval_with_key>.3:670:13
%sub_24.1 : Tensor = aten::sub(%div_29.1, %1305, %121) # <eval_with_key>.3:672:13
%2057 : Tensor[] = prim::ListConstruct(%sub_23.1, %sub_24.1)
%cat_28.1 : Tensor = aten::cat(%2057, %1097) # <eval_with_key>.3:673:13
%grid_sampler_2d_13.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_28.1, %124, %124, %137) # <eval_with_key>.3:674:25
%2064 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_64.1 : Tensor = aten::view(%grid_sampler_2d_13.1, %2064) # <eval_with_key>.3:675:14
%div_30.1 : Tensor = aten::div(%div_27.1, %1299) # <eval_with_key>.3:678:13
%add_28.1 : Tensor = aten::add(%div_30.1, %1288, %121) # <eval_with_key>.3:680:13
%2074 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_11.1 : Tensor[] = aten::split_with_sizes(%add_28.1, %2074, %1097) # <eval_with_key>.3:681:26
%getitem_87.1 : Tensor = aten::__getitem__(%split_with_sizes_11.1, %124) # <eval_with_key>.3:682:17
%getitem_88.1 : Tensor = aten::__getitem__(%split_with_sizes_11.1, %121) # <eval_with_key>.3:683:17
%mul_32.1 : Tensor = aten::mul(%getitem_87.1, %1299) # <eval_with_key>.3:687:13
%div_31.1 : Tensor = aten::div(%mul_32.1, %1392) # <eval_with_key>.3:689:13
%sub_25.1 : Tensor = aten::sub(%div_31.1, %1305, %121) # <eval_with_key>.3:691:13
%mul_33.1 : Tensor = aten::mul(%getitem_88.1, %1299) # <eval_with_key>.3:693:13
%div_32.1 : Tensor = aten::div(%mul_33.1, %1392) # <eval_with_key>.3:695:13
%sub_26.1 : Tensor = aten::sub(%div_32.1, %1305, %121) # <eval_with_key>.3:697:13
%2102 : Tensor[] = prim::ListConstruct(%sub_25.1, %sub_26.1)
%cat_29.1 : Tensor = aten::cat(%2102, %1097) # <eval_with_key>.3:698:13
%grid_sampler_2d_14.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_29.1, %124, %124, %137) # <eval_with_key>.3:699:25
%2109 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_65.1 : Tensor = aten::view(%grid_sampler_2d_14.1, %2109) # <eval_with_key>.3:700:14
%div_33.1 : Tensor = aten::div(%div_30.1, %1299) # <eval_with_key>.3:703:13
%add_29.1 : Tensor = aten::add(%div_33.1, %1288, %121) # <eval_with_key>.3:705:13
%2119 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_12.1 : Tensor[] = aten::split_with_sizes(%add_29.1, %2119, %1097) # <eval_with_key>.3:706:26
%getitem_89.1 : Tensor = aten::__getitem__(%split_with_sizes_12.1, %124) # <eval_with_key>.3:707:17
%getitem_90.1 : Tensor = aten::__getitem__(%split_with_sizes_12.1, %121) # <eval_with_key>.3:708:17
%mul_34.1 : Tensor = aten::mul(%getitem_89.1, %1299) # <eval_with_key>.3:712:13
%div_34.1 : Tensor = aten::div(%mul_34.1, %1299) # <eval_with_key>.3:714:13
%sub_27.1 : Tensor = aten::sub(%div_34.1, %1305, %121) # <eval_with_key>.3:716:13
%mul_35.1 : Tensor = aten::mul(%getitem_90.1, %1299) # <eval_with_key>.3:718:13
%div_35.1 : Tensor = aten::div(%mul_35.1, %1299) # <eval_with_key>.3:720:13
%sub_28.1 : Tensor = aten::sub(%div_35.1, %1305, %121) # <eval_with_key>.3:722:13
%2147 : Tensor[] = prim::ListConstruct(%sub_27.1, %sub_28.1)
%cat_30.1 : Tensor = aten::cat(%2147, %1097) # <eval_with_key>.3:723:13
%grid_sampler_2d_15.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_30.1, %124, %124, %137) # <eval_with_key>.3:724:25
%2154 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_66.1 : Tensor = aten::view(%grid_sampler_2d_15.1, %2154) # <eval_with_key>.3:725:14
%2160 : Tensor[] = prim::ListConstruct(%view_63.1, %view_64.1, %view_65.1, %view_66.1)
%cat_31.1 : Tensor = aten::cat(%2160, %1097) # <eval_with_key>.3:727:13
%2164 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_6.1 : Tensor = aten::permute(%cat_31.1, %2164) # <eval_with_key>.3:728:16
%clone_3.1 : Tensor = aten::clone(%permute_6.1, %124) # <eval_with_key>.3:730:14
%sub_29.1 : Tensor = aten::sub(%add_25.1, %1263, %121) # <eval_with_key>.3:732:13
%2175 : int[] = prim::ListConstruct(%121, %121)
%2176 : int[] = prim::ListConstruct(%124, %124)
%2177 : int[] = prim::ListConstruct(%121, %121)
%2179 : int[] = prim::ListConstruct(%124, %124)
%convolution_71.1 : Tensor = aten::convolution(%clone_3.1, %primals_96.1, %primals_95.1, %2175, %2176, %2177, %123, %2179, %121) # <eval_with_key>.3:733:21
%relu__65.1 : Tensor = aten::relu_(%convolution_71.1) # <eval_with_key>.3:734:15
%2186 : int[] = prim::ListConstruct(%121, %121)
%2187 : int[] = prim::ListConstruct(%119, %119)
%2188 : int[] = prim::ListConstruct(%121, %121)
%2190 : int[] = prim::ListConstruct(%124, %124)
%convolution_72.1 : Tensor = aten::convolution(%sub_29.1, %primals_98.1, %primals_97.1, %2186, %2187, %2188, %123, %2190, %121) # <eval_with_key>.3:735:21
%relu__66.1 : Tensor = aten::relu_(%convolution_72.1) # <eval_with_key>.3:736:15
%2197 : int[] = prim::ListConstruct(%121, %121)
%2198 : int[] = prim::ListConstruct(%121, %121)
%2199 : int[] = prim::ListConstruct(%121, %121)
%2201 : int[] = prim::ListConstruct(%124, %124)
%convolution_73.1 : Tensor = aten::convolution(%relu__66.1, %primals_100.1, %primals_99.1, %2197, %2198, %2199, %123, %2201, %121) # <eval_with_key>.3:737:21
%relu__67.1 : Tensor = aten::relu_(%convolution_73.1) # <eval_with_key>.3:738:15
%2207 : Tensor[] = prim::ListConstruct(%relu__65.1, %relu__67.1)
%cat_32.1 : Tensor = aten::cat(%2207, %121) # <eval_with_key>.3:739:13
%2212 : int[] = prim::ListConstruct(%121, %121)
%2213 : int[] = prim::ListConstruct(%121, %121)
%2214 : int[] = prim::ListConstruct(%121, %121)
%2216 : int[] = prim::ListConstruct(%124, %124)
%convolution_74.1 : Tensor = aten::convolution(%cat_32.1, %primals_94.1, %primals_93.1, %2212, %2213, %2214, %123, %2216, %121) # <eval_with_key>.3:740:21
%relu__68.1 : Tensor = aten::relu_(%convolution_74.1) # <eval_with_key>.3:741:15
%2223 : Tensor[] = prim::ListConstruct(%relu.1, %relu__68.1, %sub_29.1)
%cat_34.1 : Tensor = aten::cat(%2223, %121) # <eval_with_key>.3:743:13
%2227 : Tensor[] = prim::ListConstruct(%add_24.1, %cat_34.1)
%cat_35.1 : Tensor = aten::cat(%2227, %121) # <eval_with_key>.3:744:13
%2232 : int[] = prim::ListConstruct(%121, %121)
%2233 : int[] = prim::ListConstruct(%121, %121)
%2234 : int[] = prim::ListConstruct(%121, %121)
%2236 : int[] = prim::ListConstruct(%124, %124)
%convolution_75.1 : Tensor = aten::convolution(%cat_35.1, %primals_106.1, %primals_105.1, %2232, %2233, %2234, %123, %2236, %121) # <eval_with_key>.3:745:21
%sigmoid_6.1 : Tensor = aten::sigmoid(%convolution_75.1) # <eval_with_key>.3:746:16
%2243 : int[] = prim::ListConstruct(%121, %121)
%2244 : int[] = prim::ListConstruct(%121, %121)
%2245 : int[] = prim::ListConstruct(%121, %121)
%2247 : int[] = prim::ListConstruct(%124, %124)
%convolution_76.1 : Tensor = aten::convolution(%cat_35.1, %primals_104.1, %primals_103.1, %2243, %2244, %2245, %123, %2247, %121) # <eval_with_key>.3:747:21
%sigmoid_7.1 : Tensor = aten::sigmoid(%convolution_76.1) # <eval_with_key>.3:748:16
%mul_36.1 : Tensor = aten::mul(%sigmoid_7.1, %add_24.1) # <eval_with_key>.3:749:13
%2256 : Tensor[] = prim::ListConstruct(%mul_36.1, %cat_34.1)
%cat_36.1 : Tensor = aten::cat(%2256, %121) # <eval_with_key>.3:750:13
%2261 : int[] = prim::ListConstruct(%121, %121)
%2262 : int[] = prim::ListConstruct(%121, %121)
%2263 : int[] = prim::ListConstruct(%121, %121)
%2265 : int[] = prim::ListConstruct(%124, %124)
%convolution_77.1 : Tensor = aten::convolution(%cat_36.1, %primals_102.1, %primals_101.1, %2261, %2262, %2263, %123, %2265, %121) # <eval_with_key>.3:751:21
%tanh_4.1 : Tensor = aten::tanh(%convolution_77.1) # <eval_with_key>.3:752:13
%rsub_3.1 : Tensor = aten::rsub(%sigmoid_6.1, %121, %121) # <eval_with_key>.3:753:13
%mul_37.1 : Tensor = aten::mul(%rsub_3.1, %add_24.1) # <eval_with_key>.3:754:13
%mul_38.1 : Tensor = aten::mul(%sigmoid_6.1, %tanh_4.1) # <eval_with_key>.3:755:13
%add_30.1 : Tensor = aten::add(%mul_37.1, %mul_38.1, %121) # <eval_with_key>.3:756:13
%2285 : int[] = prim::ListConstruct(%121, %121)
%2286 : int[] = prim::ListConstruct(%121, %121)
%2287 : int[] = prim::ListConstruct(%121, %121)
%2289 : int[] = prim::ListConstruct(%124, %124)
%convolution_78.1 : Tensor = aten::convolution(%add_30.1, %primals_90.1, %primals_89.1, %2285, %2286, %2287, %123, %2289, %121) # <eval_with_key>.3:757:21
%relu__69.1 : Tensor = aten::relu_(%convolution_78.1) # <eval_with_key>.3:758:15
%2296 : int[] = prim::ListConstruct(%121, %121)
%2297 : int[] = prim::ListConstruct(%121, %121)
%2298 : int[] = prim::ListConstruct(%121, %121)
%2300 : int[] = prim::ListConstruct(%124, %124)
%convolution_79.1 : Tensor = aten::convolution(%relu__69.1, %primals_92.1, %primals_91.1, %2296, %2297, %2298, %123, %2300, %121) # <eval_with_key>.3:759:21
%add_31.1 : Tensor = aten::add(%add_25.1, %convolution_79.1, %121) # <eval_with_key>.3:760:13
%sub_30.1 : Tensor = aten::sub(%add_31.1, %1263, %121) # <eval_with_key>.3:762:13
%2311 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_3.1 : Tensor = aten::upsample_bilinear2d(%sub_30.1, %2311, %137, %133, %133) # <eval_with_key>.3:763:28
%mul_39.1 : Tensor = aten::mul(%upsample_bilinear2d_3.1, %1279) # <eval_with_key>.3:765:13
%2320 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_7.1 : Tensor = aten::permute(%add_31.1, %2320) # <eval_with_key>.3:768:16
%2323 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_67.1 : Tensor = aten::view(%permute_7.1, %2323) # <eval_with_key>.3:770:14
%add_32.1 : Tensor = aten::add(%view_67.1, %1288, %121) # <eval_with_key>.3:774:13
%2330 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_13.1 : Tensor[] = aten::split_with_sizes(%add_32.1, %2330, %1097) # <eval_with_key>.3:775:26
%getitem_91.1 : Tensor = aten::__getitem__(%split_with_sizes_13.1, %124) # <eval_with_key>.3:776:17
%getitem_92.1 : Tensor = aten::__getitem__(%split_with_sizes_13.1, %121) # <eval_with_key>.3:777:17
%mul_40.1 : Tensor = aten::mul(%getitem_91.1, %1299) # <eval_with_key>.3:781:13
%div_37.1 : Tensor = aten::div(%mul_40.1, %1302) # <eval_with_key>.3:783:13
%sub_31.1 : Tensor = aten::sub(%div_37.1, %1305, %121) # <eval_with_key>.3:785:13
%mul_41.1 : Tensor = aten::mul(%getitem_92.1, %1299) # <eval_with_key>.3:787:13
%div_38.1 : Tensor = aten::div(%mul_41.1, %1302) # <eval_with_key>.3:789:13
%sub_32.1 : Tensor = aten::sub(%div_38.1, %1305, %121) # <eval_with_key>.3:791:13
%2358 : Tensor[] = prim::ListConstruct(%sub_31.1, %sub_32.1)
%cat_37.1 : Tensor = aten::cat(%2358, %1097) # <eval_with_key>.3:792:13
%grid_sampler_2d_16.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_37.1, %124, %124, %137) # <eval_with_key>.3:793:25
%2365 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_68.1 : Tensor = aten::view(%grid_sampler_2d_16.1, %2365) # <eval_with_key>.3:794:14
%div_39.1 : Tensor = aten::div(%view_67.1, %1299) # <eval_with_key>.3:797:13
%add_33.1 : Tensor = aten::add(%div_39.1, %1288, %121) # <eval_with_key>.3:799:13
%2375 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_14.1 : Tensor[] = aten::split_with_sizes(%add_33.1, %2375, %1097) # <eval_with_key>.3:800:26
%getitem_93.1 : Tensor = aten::__getitem__(%split_with_sizes_14.1, %124) # <eval_with_key>.3:801:17
%getitem_94.1 : Tensor = aten::__getitem__(%split_with_sizes_14.1, %121) # <eval_with_key>.3:802:17
%mul_42.1 : Tensor = aten::mul(%getitem_93.1, %1299) # <eval_with_key>.3:806:13
%div_40.1 : Tensor = aten::div(%mul_42.1, %1347) # <eval_with_key>.3:808:13
%sub_33.1 : Tensor = aten::sub(%div_40.1, %1305, %121) # <eval_with_key>.3:810:13
%mul_43.1 : Tensor = aten::mul(%getitem_94.1, %1299) # <eval_with_key>.3:812:13
%div_41.1 : Tensor = aten::div(%mul_43.1, %1347) # <eval_with_key>.3:814:13
%sub_34.1 : Tensor = aten::sub(%div_41.1, %1305, %121) # <eval_with_key>.3:816:13
%2403 : Tensor[] = prim::ListConstruct(%sub_33.1, %sub_34.1)
%cat_38.1 : Tensor = aten::cat(%2403, %1097) # <eval_with_key>.3:817:13
%grid_sampler_2d_17.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_38.1, %124, %124, %137) # <eval_with_key>.3:818:25
%2410 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_69.1 : Tensor = aten::view(%grid_sampler_2d_17.1, %2410) # <eval_with_key>.3:819:14
%div_42.1 : Tensor = aten::div(%div_39.1, %1299) # <eval_with_key>.3:822:13
%add_34.1 : Tensor = aten::add(%div_42.1, %1288, %121) # <eval_with_key>.3:824:13
%2420 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_15.1 : Tensor[] = aten::split_with_sizes(%add_34.1, %2420, %1097) # <eval_with_key>.3:825:26
%getitem_95.1 : Tensor = aten::__getitem__(%split_with_sizes_15.1, %124) # <eval_with_key>.3:826:17
%getitem_96.1 : Tensor = aten::__getitem__(%split_with_sizes_15.1, %121) # <eval_with_key>.3:827:17
%mul_44.1 : Tensor = aten::mul(%getitem_95.1, %1299) # <eval_with_key>.3:831:13
%div_43.1 : Tensor = aten::div(%mul_44.1, %1392) # <eval_with_key>.3:833:13
%sub_35.1 : Tensor = aten::sub(%div_43.1, %1305, %121) # <eval_with_key>.3:835:13
%mul_45.1 : Tensor = aten::mul(%getitem_96.1, %1299) # <eval_with_key>.3:837:13
%div_44.1 : Tensor = aten::div(%mul_45.1, %1392) # <eval_with_key>.3:839:13
%sub_36.1 : Tensor = aten::sub(%div_44.1, %1305, %121) # <eval_with_key>.3:841:13
%2448 : Tensor[] = prim::ListConstruct(%sub_35.1, %sub_36.1)
%cat_39.1 : Tensor = aten::cat(%2448, %1097) # <eval_with_key>.3:842:13
%grid_sampler_2d_18.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_39.1, %124, %124, %137) # <eval_with_key>.3:843:25
%2455 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_70.1 : Tensor = aten::view(%grid_sampler_2d_18.1, %2455) # <eval_with_key>.3:844:14
%div_45.1 : Tensor = aten::div(%div_42.1, %1299) # <eval_with_key>.3:847:13
%add_35.1 : Tensor = aten::add(%div_45.1, %1288, %121) # <eval_with_key>.3:849:13
%2465 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_16.1 : Tensor[] = aten::split_with_sizes(%add_35.1, %2465, %1097) # <eval_with_key>.3:850:26
%getitem_97.1 : Tensor = aten::__getitem__(%split_with_sizes_16.1, %124) # <eval_with_key>.3:851:17
%getitem_98.1 : Tensor = aten::__getitem__(%split_with_sizes_16.1, %121) # <eval_with_key>.3:852:17
%mul_46.1 : Tensor = aten::mul(%getitem_97.1, %1299) # <eval_with_key>.3:856:13
%div_46.1 : Tensor = aten::div(%mul_46.1, %1299) # <eval_with_key>.3:858:13
%sub_37.1 : Tensor = aten::sub(%div_46.1, %1305, %121) # <eval_with_key>.3:860:13
%mul_47.1 : Tensor = aten::mul(%getitem_98.1, %1299) # <eval_with_key>.3:862:13
%div_47.1 : Tensor = aten::div(%mul_47.1, %1299) # <eval_with_key>.3:864:13
%sub_38.1 : Tensor = aten::sub(%div_47.1, %1305, %121) # <eval_with_key>.3:866:13
%2493 : Tensor[] = prim::ListConstruct(%sub_37.1, %sub_38.1)
%cat_40.1 : Tensor = aten::cat(%2493, %1097) # <eval_with_key>.3:867:13
%grid_sampler_2d_19.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_40.1, %124, %124, %137) # <eval_with_key>.3:868:25
%2500 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_71.1 : Tensor = aten::view(%grid_sampler_2d_19.1, %2500) # <eval_with_key>.3:869:14
%2506 : Tensor[] = prim::ListConstruct(%view_68.1, %view_69.1, %view_70.1, %view_71.1)
%cat_41.1 : Tensor = aten::cat(%2506, %1097) # <eval_with_key>.3:871:13
%2510 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_8.1 : Tensor = aten::permute(%cat_41.1, %2510) # <eval_with_key>.3:872:16
%clone_4.1 : Tensor = aten::clone(%permute_8.1, %124) # <eval_with_key>.3:874:14
%sub_39.1 : Tensor = aten::sub(%add_31.1, %1263, %121) # <eval_with_key>.3:876:13
%2521 : int[] = prim::ListConstruct(%121, %121)
%2522 : int[] = prim::ListConstruct(%124, %124)
%2523 : int[] = prim::ListConstruct(%121, %121)
%2525 : int[] = prim::ListConstruct(%124, %124)
%convolution_80.1 : Tensor = aten::convolution(%clone_4.1, %primals_96.1, %primals_95.1, %2521, %2522, %2523, %123, %2525, %121) # <eval_with_key>.3:877:21
%relu__70.1 : Tensor = aten::relu_(%convolution_80.1) # <eval_with_key>.3:878:15
%2532 : int[] = prim::ListConstruct(%121, %121)
%2533 : int[] = prim::ListConstruct(%119, %119)
%2534 : int[] = prim::ListConstruct(%121, %121)
%2536 : int[] = prim::ListConstruct(%124, %124)
%convolution_81.1 : Tensor = aten::convolution(%sub_39.1, %primals_98.1, %primals_97.1, %2532, %2533, %2534, %123, %2536, %121) # <eval_with_key>.3:879:21
%relu__71.1 : Tensor = aten::relu_(%convolution_81.1) # <eval_with_key>.3:880:15
%2543 : int[] = prim::ListConstruct(%121, %121)
%2544 : int[] = prim::ListConstruct(%121, %121)
%2545 : int[] = prim::ListConstruct(%121, %121)
%2547 : int[] = prim::ListConstruct(%124, %124)
%convolution_82.1 : Tensor = aten::convolution(%relu__71.1, %primals_100.1, %primals_99.1, %2543, %2544, %2545, %123, %2547, %121) # <eval_with_key>.3:881:21
%relu__72.1 : Tensor = aten::relu_(%convolution_82.1) # <eval_with_key>.3:882:15
%2553 : Tensor[] = prim::ListConstruct(%relu__70.1, %relu__72.1)
%cat_42.1 : Tensor = aten::cat(%2553, %121) # <eval_with_key>.3:883:13
%2558 : int[] = prim::ListConstruct(%121, %121)
%2559 : int[] = prim::ListConstruct(%121, %121)
%2560 : int[] = prim::ListConstruct(%121, %121)
%2562 : int[] = prim::ListConstruct(%124, %124)
%convolution_83.1 : Tensor = aten::convolution(%cat_42.1, %primals_94.1, %primals_93.1, %2558, %2559, %2560, %123, %2562, %121) # <eval_with_key>.3:884:21
%relu__73.1 : Tensor = aten::relu_(%convolution_83.1) # <eval_with_key>.3:885:15
%2569 : Tensor[] = prim::ListConstruct(%relu.1, %relu__73.1, %sub_39.1)
%cat_44.1 : Tensor = aten::cat(%2569, %121) # <eval_with_key>.3:887:13
%2573 : Tensor[] = prim::ListConstruct(%add_30.1, %cat_44.1)
%cat_45.1 : Tensor = aten::cat(%2573, %121) # <eval_with_key>.3:888:13
%2578 : int[] = prim::ListConstruct(%121, %121)
%2579 : int[] = prim::ListConstruct(%121, %121)
%2580 : int[] = prim::ListConstruct(%121, %121)
%2582 : int[] = prim::ListConstruct(%124, %124)
%convolution_84.1 : Tensor = aten::convolution(%cat_45.1, %primals_106.1, %primals_105.1, %2578, %2579, %2580, %123, %2582, %121) # <eval_with_key>.3:889:21
%sigmoid_8.1 : Tensor = aten::sigmoid(%convolution_84.1) # <eval_with_key>.3:890:16
%2589 : int[] = prim::ListConstruct(%121, %121)
%2590 : int[] = prim::ListConstruct(%121, %121)
%2591 : int[] = prim::ListConstruct(%121, %121)
%2593 : int[] = prim::ListConstruct(%124, %124)
%convolution_85.1 : Tensor = aten::convolution(%cat_45.1, %primals_104.1, %primals_103.1, %2589, %2590, %2591, %123, %2593, %121) # <eval_with_key>.3:891:21
%sigmoid_9.1 : Tensor = aten::sigmoid(%convolution_85.1) # <eval_with_key>.3:892:16
%mul_48.1 : Tensor = aten::mul(%sigmoid_9.1, %add_30.1) # <eval_with_key>.3:893:13
%2602 : Tensor[] = prim::ListConstruct(%mul_48.1, %cat_44.1)
%cat_46.1 : Tensor = aten::cat(%2602, %121) # <eval_with_key>.3:894:13
%2607 : int[] = prim::ListConstruct(%121, %121)
%2608 : int[] = prim::ListConstruct(%121, %121)
%2609 : int[] = prim::ListConstruct(%121, %121)
%2611 : int[] = prim::ListConstruct(%124, %124)
%convolution_86.1 : Tensor = aten::convolution(%cat_46.1, %primals_102.1, %primals_101.1, %2607, %2608, %2609, %123, %2611, %121) # <eval_with_key>.3:895:21
%tanh_5.1 : Tensor = aten::tanh(%convolution_86.1) # <eval_with_key>.3:896:13
%rsub_4.1 : Tensor = aten::rsub(%sigmoid_8.1, %121, %121) # <eval_with_key>.3:897:13
%mul_49.1 : Tensor = aten::mul(%rsub_4.1, %add_30.1) # <eval_with_key>.3:898:13
%mul_50.1 : Tensor = aten::mul(%sigmoid_8.1, %tanh_5.1) # <eval_with_key>.3:899:13
%add_36.1 : Tensor = aten::add(%mul_49.1, %mul_50.1, %121) # <eval_with_key>.3:900:13
%2631 : int[] = prim::ListConstruct(%121, %121)
%2632 : int[] = prim::ListConstruct(%121, %121)
%2633 : int[] = prim::ListConstruct(%121, %121)
%2635 : int[] = prim::ListConstruct(%124, %124)
%convolution_87.1 : Tensor = aten::convolution(%add_36.1, %primals_90.1, %primals_89.1, %2631, %2632, %2633, %123, %2635, %121) # <eval_with_key>.3:901:21
%relu__74.1 : Tensor = aten::relu_(%convolution_87.1) # <eval_with_key>.3:902:15
%2642 : int[] = prim::ListConstruct(%121, %121)
%2643 : int[] = prim::ListConstruct(%121, %121)
%2644 : int[] = prim::ListConstruct(%121, %121)
%2646 : int[] = prim::ListConstruct(%124, %124)
%convolution_88.1 : Tensor = aten::convolution(%relu__74.1, %primals_92.1, %primals_91.1, %2642, %2643, %2644, %123, %2646, %121) # <eval_with_key>.3:903:21
%add_37.1 : Tensor = aten::add(%add_31.1, %convolution_88.1, %121) # <eval_with_key>.3:904:13
%sub_40.1 : Tensor = aten::sub(%add_37.1, %1263, %121) # <eval_with_key>.3:906:13
%2657 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_4.1 : Tensor = aten::upsample_bilinear2d(%sub_40.1, %2657, %137, %133, %133) # <eval_with_key>.3:907:28
%mul_51.1 : Tensor = aten::mul(%upsample_bilinear2d_4.1, %1279) # <eval_with_key>.3:909:13
%2666 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_9.1 : Tensor = aten::permute(%add_37.1, %2666) # <eval_with_key>.3:912:16
%2669 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_72.1 : Tensor = aten::view(%permute_9.1, %2669) # <eval_with_key>.3:914:14
%add_38.1 : Tensor = aten::add(%view_72.1, %1288, %121) # <eval_with_key>.3:918:13
%2676 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_17.1 : Tensor[] = aten::split_with_sizes(%add_38.1, %2676, %1097) # <eval_with_key>.3:919:26
%getitem_99.1 : Tensor = aten::__getitem__(%split_with_sizes_17.1, %124) # <eval_with_key>.3:920:17
%getitem_100.1 : Tensor = aten::__getitem__(%split_with_sizes_17.1, %121) # <eval_with_key>.3:921:18
%mul_52.1 : Tensor = aten::mul(%getitem_99.1, %1299) # <eval_with_key>.3:925:13
%div_49.1 : Tensor = aten::div(%mul_52.1, %1302) # <eval_with_key>.3:927:13
%sub_41.1 : Tensor = aten::sub(%div_49.1, %1305, %121) # <eval_with_key>.3:929:13
%mul_53.1 : Tensor = aten::mul(%getitem_100.1, %1299) # <eval_with_key>.3:931:13
%div_50.1 : Tensor = aten::div(%mul_53.1, %1302) # <eval_with_key>.3:933:13
%sub_42.1 : Tensor = aten::sub(%div_50.1, %1305, %121) # <eval_with_key>.3:935:13
%2704 : Tensor[] = prim::ListConstruct(%sub_41.1, %sub_42.1)
%cat_47.1 : Tensor = aten::cat(%2704, %1097) # <eval_with_key>.3:936:13
%grid_sampler_2d_20.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_47.1, %124, %124, %137) # <eval_with_key>.3:937:25
%2711 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_73.1 : Tensor = aten::view(%grid_sampler_2d_20.1, %2711) # <eval_with_key>.3:938:14
%div_51.1 : Tensor = aten::div(%view_72.1, %1299) # <eval_with_key>.3:941:13
%add_39.1 : Tensor = aten::add(%div_51.1, %1288, %121) # <eval_with_key>.3:943:13
%2721 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_18.1 : Tensor[] = aten::split_with_sizes(%add_39.1, %2721, %1097) # <eval_with_key>.3:944:26
%getitem_101.1 : Tensor = aten::__getitem__(%split_with_sizes_18.1, %124) # <eval_with_key>.3:945:18
%getitem_102.1 : Tensor = aten::__getitem__(%split_with_sizes_18.1, %121) # <eval_with_key>.3:946:18
%mul_54.1 : Tensor = aten::mul(%getitem_101.1, %1299) # <eval_with_key>.3:950:13
%div_52.1 : Tensor = aten::div(%mul_54.1, %1347) # <eval_with_key>.3:952:13
%sub_43.1 : Tensor = aten::sub(%div_52.1, %1305, %121) # <eval_with_key>.3:954:13
%mul_55.1 : Tensor = aten::mul(%getitem_102.1, %1299) # <eval_with_key>.3:956:13
%div_53.1 : Tensor = aten::div(%mul_55.1, %1347) # <eval_with_key>.3:958:13
%sub_44.1 : Tensor = aten::sub(%div_53.1, %1305, %121) # <eval_with_key>.3:960:13
%2749 : Tensor[] = prim::ListConstruct(%sub_43.1, %sub_44.1)
%cat_48.1 : Tensor = aten::cat(%2749, %1097) # <eval_with_key>.3:961:13
%grid_sampler_2d_21.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_48.1, %124, %124, %137) # <eval_with_key>.3:962:25
%2756 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_74.1 : Tensor = aten::view(%grid_sampler_2d_21.1, %2756) # <eval_with_key>.3:963:14
%div_54.1 : Tensor = aten::div(%div_51.1, %1299) # <eval_with_key>.3:966:13
%add_40.1 : Tensor = aten::add(%div_54.1, %1288, %121) # <eval_with_key>.3:968:13
%2766 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_19.1 : Tensor[] = aten::split_with_sizes(%add_40.1, %2766, %1097) # <eval_with_key>.3:969:26
%getitem_103.1 : Tensor = aten::__getitem__(%split_with_sizes_19.1, %124) # <eval_with_key>.3:970:18
%getitem_104.1 : Tensor = aten::__getitem__(%split_with_sizes_19.1, %121) # <eval_with_key>.3:971:18
%mul_56.1 : Tensor = aten::mul(%getitem_103.1, %1299) # <eval_with_key>.3:975:13
%div_55.1 : Tensor = aten::div(%mul_56.1, %1392) # <eval_with_key>.3:977:13
%sub_45.1 : Tensor = aten::sub(%div_55.1, %1305, %121) # <eval_with_key>.3:979:13
%mul_57.1 : Tensor = aten::mul(%getitem_104.1, %1299) # <eval_with_key>.3:981:13
%div_56.1 : Tensor = aten::div(%mul_57.1, %1392) # <eval_with_key>.3:983:13
%sub_46.1 : Tensor = aten::sub(%div_56.1, %1305, %121) # <eval_with_key>.3:985:13
%2794 : Tensor[] = prim::ListConstruct(%sub_45.1, %sub_46.1)
%cat_49.1 : Tensor = aten::cat(%2794, %1097) # <eval_with_key>.3:986:13
%grid_sampler_2d_22.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_49.1, %124, %124, %137) # <eval_with_key>.3:987:25
%2801 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_75.1 : Tensor = aten::view(%grid_sampler_2d_22.1, %2801) # <eval_with_key>.3:988:14
%div_57.1 : Tensor = aten::div(%div_54.1, %1299) # <eval_with_key>.3:991:13
%add_41.1 : Tensor = aten::add(%div_57.1, %1288, %121) # <eval_with_key>.3:993:13
%2811 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_20.1 : Tensor[] = aten::split_with_sizes(%add_41.1, %2811, %1097) # <eval_with_key>.3:994:26
%getitem_105.1 : Tensor = aten::__getitem__(%split_with_sizes_20.1, %124) # <eval_with_key>.3:995:18
%getitem_106.1 : Tensor = aten::__getitem__(%split_with_sizes_20.1, %121) # <eval_with_key>.3:996:18
%mul_58.1 : Tensor = aten::mul(%getitem_105.1, %1299) # <eval_with_key>.3:1000:13
%div_58.1 : Tensor = aten::div(%mul_58.1, %1299) # <eval_with_key>.3:1002:13
%sub_47.1 : Tensor = aten::sub(%div_58.1, %1305, %121) # <eval_with_key>.3:1004:13
%mul_59.1 : Tensor = aten::mul(%getitem_106.1, %1299) # <eval_with_key>.3:1006:13
%div_59.1 : Tensor = aten::div(%mul_59.1, %1299) # <eval_with_key>.3:1008:13
%sub_48.1 : Tensor = aten::sub(%div_59.1, %1305, %121) # <eval_with_key>.3:1010:13
%2839 : Tensor[] = prim::ListConstruct(%sub_47.1, %sub_48.1)
%cat_50.1 : Tensor = aten::cat(%2839, %1097) # <eval_with_key>.3:1011:13
%grid_sampler_2d_23.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_50.1, %124, %124, %137) # <eval_with_key>.3:1012:25
%2846 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_76.1 : Tensor = aten::view(%grid_sampler_2d_23.1, %2846) # <eval_with_key>.3:1013:14
%2852 : Tensor[] = prim::ListConstruct(%view_73.1, %view_74.1, %view_75.1, %view_76.1)
%cat_51.1 : Tensor = aten::cat(%2852, %1097) # <eval_with_key>.3:1015:13
%2856 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_10.1 : Tensor = aten::permute(%cat_51.1, %2856) # <eval_with_key>.3:1016:17
%clone_5.1 : Tensor = aten::clone(%permute_10.1, %124) # <eval_with_key>.3:1018:14
%sub_49.1 : Tensor = aten::sub(%add_37.1, %1263, %121) # <eval_with_key>.3:1020:13
%2867 : int[] = prim::ListConstruct(%121, %121)
%2868 : int[] = prim::ListConstruct(%124, %124)
%2869 : int[] = prim::ListConstruct(%121, %121)
%2871 : int[] = prim::ListConstruct(%124, %124)
%convolution_89.1 : Tensor = aten::convolution(%clone_5.1, %primals_96.1, %primals_95.1, %2867, %2868, %2869, %123, %2871, %121) # <eval_with_key>.3:1021:21
%relu__75.1 : Tensor = aten::relu_(%convolution_89.1) # <eval_with_key>.3:1022:15
%2878 : int[] = prim::ListConstruct(%121, %121)
%2879 : int[] = prim::ListConstruct(%119, %119)
%2880 : int[] = prim::ListConstruct(%121, %121)
%2882 : int[] = prim::ListConstruct(%124, %124)
%convolution_90.1 : Tensor = aten::convolution(%sub_49.1, %primals_98.1, %primals_97.1, %2878, %2879, %2880, %123, %2882, %121) # <eval_with_key>.3:1023:21
%relu__76.1 : Tensor = aten::relu_(%convolution_90.1) # <eval_with_key>.3:1024:15
%2889 : int[] = prim::ListConstruct(%121, %121)
%2890 : int[] = prim::ListConstruct(%121, %121)
%2891 : int[] = prim::ListConstruct(%121, %121)
%2893 : int[] = prim::ListConstruct(%124, %124)
%convolution_91.1 : Tensor = aten::convolution(%relu__76.1, %primals_100.1, %primals_99.1, %2889, %2890, %2891, %123, %2893, %121) # <eval_with_key>.3:1025:21
%relu__77.1 : Tensor = aten::relu_(%convolution_91.1) # <eval_with_key>.3:1026:15
%2899 : Tensor[] = prim::ListConstruct(%relu__75.1, %relu__77.1)
%cat_52.1 : Tensor = aten::cat(%2899, %121) # <eval_with_key>.3:1027:13
%2904 : int[] = prim::ListConstruct(%121, %121)
%2905 : int[] = prim::ListConstruct(%121, %121)
%2906 : int[] = prim::ListConstruct(%121, %121)
%2908 : int[] = prim::ListConstruct(%124, %124)
%convolution_92.1 : Tensor = aten::convolution(%cat_52.1, %primals_94.1, %primals_93.1, %2904, %2905, %2906, %123, %2908, %121) # <eval_with_key>.3:1028:21
%relu__78.1 : Tensor = aten::relu_(%convolution_92.1) # <eval_with_key>.3:1029:15
%2915 : Tensor[] = prim::ListConstruct(%relu.1, %relu__78.1, %sub_49.1)
%cat_54.1 : Tensor = aten::cat(%2915, %121) # <eval_with_key>.3:1031:13
%2919 : Tensor[] = prim::ListConstruct(%add_36.1, %cat_54.1)
%cat_55.1 : Tensor = aten::cat(%2919, %121) # <eval_with_key>.3:1032:13
%2924 : int[] = prim::ListConstruct(%121, %121)
%2925 : int[] = prim::ListConstruct(%121, %121)
%2926 : int[] = prim::ListConstruct(%121, %121)
%2928 : int[] = prim::ListConstruct(%124, %124)
%convolution_93.1 : Tensor = aten::convolution(%cat_55.1, %primals_106.1, %primals_105.1, %2924, %2925, %2926, %123, %2928, %121) # <eval_with_key>.3:1033:21
%sigmoid_10.1 : Tensor = aten::sigmoid(%convolution_93.1) # <eval_with_key>.3:1034:17
%2935 : int[] = prim::ListConstruct(%121, %121)
%2936 : int[] = prim::ListConstruct(%121, %121)
%2937 : int[] = prim::ListConstruct(%121, %121)
%2939 : int[] = prim::ListConstruct(%124, %124)
%convolution_94.1 : Tensor = aten::convolution(%cat_55.1, %primals_104.1, %primals_103.1, %2935, %2936, %2937, %123, %2939, %121) # <eval_with_key>.3:1035:21
%sigmoid_11.1 : Tensor = aten::sigmoid(%convolution_94.1) # <eval_with_key>.3:1036:17
%mul_60.1 : Tensor = aten::mul(%sigmoid_11.1, %add_36.1) # <eval_with_key>.3:1037:13
%2948 : Tensor[] = prim::ListConstruct(%mul_60.1, %cat_54.1)
%cat_56.1 : Tensor = aten::cat(%2948, %121) # <eval_with_key>.3:1038:13
%2953 : int[] = prim::ListConstruct(%121, %121)
%2954 : int[] = prim::ListConstruct(%121, %121)
%2955 : int[] = prim::ListConstruct(%121, %121)
%2957 : int[] = prim::ListConstruct(%124, %124)
%convolution_95.1 : Tensor = aten::convolution(%cat_56.1, %primals_102.1, %primals_101.1, %2953, %2954, %2955, %123, %2957, %121) # <eval_with_key>.3:1039:21
%tanh_6.1 : Tensor = aten::tanh(%convolution_95.1) # <eval_with_key>.3:1040:13
%rsub_5.1 : Tensor = aten::rsub(%sigmoid_10.1, %121, %121) # <eval_with_key>.3:1041:13
%mul_61.1 : Tensor = aten::mul(%rsub_5.1, %add_36.1) # <eval_with_key>.3:1042:13
%mul_62.1 : Tensor = aten::mul(%sigmoid_10.1, %tanh_6.1) # <eval_with_key>.3:1043:13
%add_42.1 : Tensor = aten::add(%mul_61.1, %mul_62.1, %121) # <eval_with_key>.3:1044:13
%2977 : int[] = prim::ListConstruct(%121, %121)
%2978 : int[] = prim::ListConstruct(%121, %121)
%2979 : int[] = prim::ListConstruct(%121, %121)
%2981 : int[] = prim::ListConstruct(%124, %124)
%convolution_96.1 : Tensor = aten::convolution(%add_42.1, %primals_90.1, %primals_89.1, %2977, %2978, %2979, %123, %2981, %121) # <eval_with_key>.3:1045:21
%relu__79.1 : Tensor = aten::relu_(%convolution_96.1) # <eval_with_key>.3:1046:15
%2988 : int[] = prim::ListConstruct(%121, %121)
%2989 : int[] = prim::ListConstruct(%121, %121)
%2990 : int[] = prim::ListConstruct(%121, %121)
%2992 : int[] = prim::ListConstruct(%124, %124)
%convolution_97.1 : Tensor = aten::convolution(%relu__79.1, %primals_92.1, %primals_91.1, %2988, %2989, %2990, %123, %2992, %121) # <eval_with_key>.3:1047:21
%add_43.1 : Tensor = aten::add(%add_37.1, %convolution_97.1, %121) # <eval_with_key>.3:1048:13
%sub_50.1 : Tensor = aten::sub(%add_43.1, %1263, %121) # <eval_with_key>.3:1050:13
%3003 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_5.1 : Tensor = aten::upsample_bilinear2d(%sub_50.1, %3003, %137, %133, %133) # <eval_with_key>.3:1051:28
%mul_63.1 : Tensor = aten::mul(%upsample_bilinear2d_5.1, %1279) # <eval_with_key>.3:1053:13
%3012 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_11.1 : Tensor = aten::permute(%add_43.1, %3012) # <eval_with_key>.3:1056:17
%3015 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_77.1 : Tensor = aten::view(%permute_11.1, %3015) # <eval_with_key>.3:1058:14
%add_44.1 : Tensor = aten::add(%view_77.1, %1288, %121) # <eval_with_key>.3:1062:13
%3022 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_21.1 : Tensor[] = aten::split_with_sizes(%add_44.1, %3022, %1097) # <eval_with_key>.3:1063:26
%getitem_107.1 : Tensor = aten::__getitem__(%split_with_sizes_21.1, %124) # <eval_with_key>.3:1064:18
%getitem_108.1 : Tensor = aten::__getitem__(%split_with_sizes_21.1, %121) # <eval_with_key>.3:1065:18
%mul_64.1 : Tensor = aten::mul(%getitem_107.1, %1299) # <eval_with_key>.3:1069:13
%div_61.1 : Tensor = aten::div(%mul_64.1, %1302) # <eval_with_key>.3:1071:13
%sub_51.1 : Tensor = aten::sub(%div_61.1, %1305, %121) # <eval_with_key>.3:1073:13
%mul_65.1 : Tensor = aten::mul(%getitem_108.1, %1299) # <eval_with_key>.3:1075:13
%div_62.1 : Tensor = aten::div(%mul_65.1, %1302) # <eval_with_key>.3:1077:13
%sub_52.1 : Tensor = aten::sub(%div_62.1, %1305, %121) # <eval_with_key>.3:1079:13
%3050 : Tensor[] = prim::ListConstruct(%sub_51.1, %sub_52.1)
%cat_57.1 : Tensor = aten::cat(%3050, %1097) # <eval_with_key>.3:1080:13
%grid_sampler_2d_24.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_57.1, %124, %124, %137) # <eval_with_key>.3:1081:25
%3057 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_78.1 : Tensor = aten::view(%grid_sampler_2d_24.1, %3057) # <eval_with_key>.3:1082:14
%div_63.1 : Tensor = aten::div(%view_77.1, %1299) # <eval_with_key>.3:1085:13
%add_45.1 : Tensor = aten::add(%div_63.1, %1288, %121) # <eval_with_key>.3:1087:13
%3067 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_22.1 : Tensor[] = aten::split_with_sizes(%add_45.1, %3067, %1097) # <eval_with_key>.3:1088:26
%getitem_109.1 : Tensor = aten::__getitem__(%split_with_sizes_22.1, %124) # <eval_with_key>.3:1089:18
%getitem_110.1 : Tensor = aten::__getitem__(%split_with_sizes_22.1, %121) # <eval_with_key>.3:1090:18
%mul_66.1 : Tensor = aten::mul(%getitem_109.1, %1299) # <eval_with_key>.3:1094:13
%div_64.1 : Tensor = aten::div(%mul_66.1, %1347) # <eval_with_key>.3:1096:13
%sub_53.1 : Tensor = aten::sub(%div_64.1, %1305, %121) # <eval_with_key>.3:1098:13
%mul_67.1 : Tensor = aten::mul(%getitem_110.1, %1299) # <eval_with_key>.3:1100:13
%div_65.1 : Tensor = aten::div(%mul_67.1, %1347) # <eval_with_key>.3:1102:13
%sub_54.1 : Tensor = aten::sub(%div_65.1, %1305, %121) # <eval_with_key>.3:1104:13
%3095 : Tensor[] = prim::ListConstruct(%sub_53.1, %sub_54.1)
%cat_58.1 : Tensor = aten::cat(%3095, %1097) # <eval_with_key>.3:1105:13
%grid_sampler_2d_25.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_58.1, %124, %124, %137) # <eval_with_key>.3:1106:25
%3102 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_79.1 : Tensor = aten::view(%grid_sampler_2d_25.1, %3102) # <eval_with_key>.3:1107:14
%div_66.1 : Tensor = aten::div(%div_63.1, %1299) # <eval_with_key>.3:1110:13
%add_46.1 : Tensor = aten::add(%div_66.1, %1288, %121) # <eval_with_key>.3:1112:13
%3112 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_23.1 : Tensor[] = aten::split_with_sizes(%add_46.1, %3112, %1097) # <eval_with_key>.3:1113:26
%getitem_111.1 : Tensor = aten::__getitem__(%split_with_sizes_23.1, %124) # <eval_with_key>.3:1114:18
%getitem_112.1 : Tensor = aten::__getitem__(%split_with_sizes_23.1, %121) # <eval_with_key>.3:1115:18
%mul_68.1 : Tensor = aten::mul(%getitem_111.1, %1299) # <eval_with_key>.3:1119:13
%div_67.1 : Tensor = aten::div(%mul_68.1, %1392) # <eval_with_key>.3:1121:13
%sub_55.1 : Tensor = aten::sub(%div_67.1, %1305, %121) # <eval_with_key>.3:1123:13
%mul_69.1 : Tensor = aten::mul(%getitem_112.1, %1299) # <eval_with_key>.3:1125:13
%div_68.1 : Tensor = aten::div(%mul_69.1, %1392) # <eval_with_key>.3:1127:13
%sub_56.1 : Tensor = aten::sub(%div_68.1, %1305, %121) # <eval_with_key>.3:1129:13
%3140 : Tensor[] = prim::ListConstruct(%sub_55.1, %sub_56.1)
%cat_59.1 : Tensor = aten::cat(%3140, %1097) # <eval_with_key>.3:1130:13
%grid_sampler_2d_26.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_59.1, %124, %124, %137) # <eval_with_key>.3:1131:25
%3147 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_80.1 : Tensor = aten::view(%grid_sampler_2d_26.1, %3147) # <eval_with_key>.3:1132:14
%div_69.1 : Tensor = aten::div(%div_66.1, %1299) # <eval_with_key>.3:1135:13
%add_47.1 : Tensor = aten::add(%div_69.1, %1288, %121) # <eval_with_key>.3:1137:13
%3157 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_24.1 : Tensor[] = aten::split_with_sizes(%add_47.1, %3157, %1097) # <eval_with_key>.3:1138:26
%getitem_113.1 : Tensor = aten::__getitem__(%split_with_sizes_24.1, %124) # <eval_with_key>.3:1139:18
%getitem_114.1 : Tensor = aten::__getitem__(%split_with_sizes_24.1, %121) # <eval_with_key>.3:1140:18
%mul_70.1 : Tensor = aten::mul(%getitem_113.1, %1299) # <eval_with_key>.3:1144:13
%div_70.1 : Tensor = aten::div(%mul_70.1, %1299) # <eval_with_key>.3:1146:13
%sub_57.1 : Tensor = aten::sub(%div_70.1, %1305, %121) # <eval_with_key>.3:1148:13
%mul_71.1 : Tensor = aten::mul(%getitem_114.1, %1299) # <eval_with_key>.3:1150:13
%div_71.1 : Tensor = aten::div(%mul_71.1, %1299) # <eval_with_key>.3:1152:13
%sub_58.1 : Tensor = aten::sub(%div_71.1, %1305, %121) # <eval_with_key>.3:1154:13
%3185 : Tensor[] = prim::ListConstruct(%sub_57.1, %sub_58.1)
%cat_60.1 : Tensor = aten::cat(%3185, %1097) # <eval_with_key>.3:1155:13
%grid_sampler_2d_27.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_60.1, %124, %124, %137) # <eval_with_key>.3:1156:25
%3192 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_81.1 : Tensor = aten::view(%grid_sampler_2d_27.1, %3192) # <eval_with_key>.3:1157:14
%3198 : Tensor[] = prim::ListConstruct(%view_78.1, %view_79.1, %view_80.1, %view_81.1)
%cat_61.1 : Tensor = aten::cat(%3198, %1097) # <eval_with_key>.3:1159:13
%3202 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_12.1 : Tensor = aten::permute(%cat_61.1, %3202) # <eval_with_key>.3:1160:17
%clone_6.1 : Tensor = aten::clone(%permute_12.1, %124) # <eval_with_key>.3:1162:14
%sub_59.1 : Tensor = aten::sub(%add_43.1, %1263, %121) # <eval_with_key>.3:1164:13
%3213 : int[] = prim::ListConstruct(%121, %121)
%3214 : int[] = prim::ListConstruct(%124, %124)
%3215 : int[] = prim::ListConstruct(%121, %121)
%3217 : int[] = prim::ListConstruct(%124, %124)
%convolution_98.1 : Tensor = aten::convolution(%clone_6.1, %primals_96.1, %primals_95.1, %3213, %3214, %3215, %123, %3217, %121) # <eval_with_key>.3:1165:21
%relu__80.1 : Tensor = aten::relu_(%convolution_98.1) # <eval_with_key>.3:1166:15
%3224 : int[] = prim::ListConstruct(%121, %121)
%3225 : int[] = prim::ListConstruct(%119, %119)
%3226 : int[] = prim::ListConstruct(%121, %121)
%3228 : int[] = prim::ListConstruct(%124, %124)
%convolution_99.1 : Tensor = aten::convolution(%sub_59.1, %primals_98.1, %primals_97.1, %3224, %3225, %3226, %123, %3228, %121) # <eval_with_key>.3:1167:21
%relu__81.1 : Tensor = aten::relu_(%convolution_99.1) # <eval_with_key>.3:1168:15
%3235 : int[] = prim::ListConstruct(%121, %121)
%3236 : int[] = prim::ListConstruct(%121, %121)
%3237 : int[] = prim::ListConstruct(%121, %121)
%3239 : int[] = prim::ListConstruct(%124, %124)
%convolution_100.1 : Tensor = aten::convolution(%relu__81.1, %primals_100.1, %primals_99.1, %3235, %3236, %3237, %123, %3239, %121) # <eval_with_key>.3:1169:22
%relu__82.1 : Tensor = aten::relu_(%convolution_100.1) # <eval_with_key>.3:1170:15
%3245 : Tensor[] = prim::ListConstruct(%relu__80.1, %relu__82.1)
%cat_62.1 : Tensor = aten::cat(%3245, %121) # <eval_with_key>.3:1171:13
%3250 : int[] = prim::ListConstruct(%121, %121)
%3251 : int[] = prim::ListConstruct(%121, %121)
%3252 : int[] = prim::ListConstruct(%121, %121)
%3254 : int[] = prim::ListConstruct(%124, %124)
%convolution_101.1 : Tensor = aten::convolution(%cat_62.1, %primals_94.1, %primals_93.1, %3250, %3251, %3252, %123, %3254, %121) # <eval_with_key>.3:1172:22
%relu__83.1 : Tensor = aten::relu_(%convolution_101.1) # <eval_with_key>.3:1173:15
%3261 : Tensor[] = prim::ListConstruct(%relu.1, %relu__83.1, %sub_59.1)
%cat_64.1 : Tensor = aten::cat(%3261, %121) # <eval_with_key>.3:1175:13
%3265 : Tensor[] = prim::ListConstruct(%add_42.1, %cat_64.1)
%cat_65.1 : Tensor = aten::cat(%3265, %121) # <eval_with_key>.3:1176:13
%3270 : int[] = prim::ListConstruct(%121, %121)
%3271 : int[] = prim::ListConstruct(%121, %121)
%3272 : int[] = prim::ListConstruct(%121, %121)
%3274 : int[] = prim::ListConstruct(%124, %124)
%convolution_102.1 : Tensor = aten::convolution(%cat_65.1, %primals_106.1, %primals_105.1, %3270, %3271, %3272, %123, %3274, %121) # <eval_with_key>.3:1177:22
%sigmoid_12.1 : Tensor = aten::sigmoid(%convolution_102.1) # <eval_with_key>.3:1178:17
%3281 : int[] = prim::ListConstruct(%121, %121)
%3282 : int[] = prim::ListConstruct(%121, %121)
%3283 : int[] = prim::ListConstruct(%121, %121)
%3285 : int[] = prim::ListConstruct(%124, %124)
%convolution_103.1 : Tensor = aten::convolution(%cat_65.1, %primals_104.1, %primals_103.1, %3281, %3282, %3283, %123, %3285, %121) # <eval_with_key>.3:1179:22
%sigmoid_13.1 : Tensor = aten::sigmoid(%convolution_103.1) # <eval_with_key>.3:1180:17
%mul_72.1 : Tensor = aten::mul(%sigmoid_13.1, %add_42.1) # <eval_with_key>.3:1181:13
%3294 : Tensor[] = prim::ListConstruct(%mul_72.1, %cat_64.1)
%cat_66.1 : Tensor = aten::cat(%3294, %121) # <eval_with_key>.3:1182:13
%3299 : int[] = prim::ListConstruct(%121, %121)
%3300 : int[] = prim::ListConstruct(%121, %121)
%3301 : int[] = prim::ListConstruct(%121, %121)
%3303 : int[] = prim::ListConstruct(%124, %124)
%convolution_104.1 : Tensor = aten::convolution(%cat_66.1, %primals_102.1, %primals_101.1, %3299, %3300, %3301, %123, %3303, %121) # <eval_with_key>.3:1183:22
%tanh_7.1 : Tensor = aten::tanh(%convolution_104.1) # <eval_with_key>.3:1184:13
%rsub_6.1 : Tensor = aten::rsub(%sigmoid_12.1, %121, %121) # <eval_with_key>.3:1185:13
%mul_73.1 : Tensor = aten::mul(%rsub_6.1, %add_42.1) # <eval_with_key>.3:1186:13
%mul_74.1 : Tensor = aten::mul(%sigmoid_12.1, %tanh_7.1) # <eval_with_key>.3:1187:13
%add_48.1 : Tensor = aten::add(%mul_73.1, %mul_74.1, %121) # <eval_with_key>.3:1188:13
%3323 : int[] = prim::ListConstruct(%121, %121)
%3324 : int[] = prim::ListConstruct(%121, %121)
%3325 : int[] = prim::ListConstruct(%121, %121)
%3327 : int[] = prim::ListConstruct(%124, %124)
%convolution_105.1 : Tensor = aten::convolution(%add_48.1, %primals_90.1, %primals_89.1, %3323, %3324, %3325, %123, %3327, %121) # <eval_with_key>.3:1189:22
%relu__84.1 : Tensor = aten::relu_(%convolution_105.1) # <eval_with_key>.3:1190:15
%3334 : int[] = prim::ListConstruct(%121, %121)
%3335 : int[] = prim::ListConstruct(%121, %121)
%3336 : int[] = prim::ListConstruct(%121, %121)
%3338 : int[] = prim::ListConstruct(%124, %124)
%convolution_106.1 : Tensor = aten::convolution(%relu__84.1, %primals_92.1, %primals_91.1, %3334, %3335, %3336, %123, %3338, %121) # <eval_with_key>.3:1191:22
%add_49.1 : Tensor = aten::add(%add_43.1, %convolution_106.1, %121) # <eval_with_key>.3:1192:13
%sub_60.1 : Tensor = aten::sub(%add_49.1, %1263, %121) # <eval_with_key>.3:1194:13
%3349 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_6.1 : Tensor = aten::upsample_bilinear2d(%sub_60.1, %3349, %137, %133, %133) # <eval_with_key>.3:1195:28
%mul_75.1 : Tensor = aten::mul(%upsample_bilinear2d_6.1, %1279) # <eval_with_key>.3:1197:13
%3358 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_13.1 : Tensor = aten::permute(%add_49.1, %3358) # <eval_with_key>.3:1200:17
%3361 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_82.1 : Tensor = aten::view(%permute_13.1, %3361) # <eval_with_key>.3:1202:14
%add_50.1 : Tensor = aten::add(%view_82.1, %1288, %121) # <eval_with_key>.3:1206:13
%3368 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_25.1 : Tensor[] = aten::split_with_sizes(%add_50.1, %3368, %1097) # <eval_with_key>.3:1207:26
%getitem_115.1 : Tensor = aten::__getitem__(%split_with_sizes_25.1, %124) # <eval_with_key>.3:1208:18
%getitem_116.1 : Tensor = aten::__getitem__(%split_with_sizes_25.1, %121) # <eval_with_key>.3:1209:18
%mul_76.1 : Tensor = aten::mul(%getitem_115.1, %1299) # <eval_with_key>.3:1213:13
%div_73.1 : Tensor = aten::div(%mul_76.1, %1302) # <eval_with_key>.3:1215:13
%sub_61.1 : Tensor = aten::sub(%div_73.1, %1305, %121) # <eval_with_key>.3:1217:13
%mul_77.1 : Tensor = aten::mul(%getitem_116.1, %1299) # <eval_with_key>.3:1219:13
%div_74.1 : Tensor = aten::div(%mul_77.1, %1302) # <eval_with_key>.3:1221:13
%sub_62.1 : Tensor = aten::sub(%div_74.1, %1305, %121) # <eval_with_key>.3:1223:13
%3396 : Tensor[] = prim::ListConstruct(%sub_61.1, %sub_62.1)
%cat_67.1 : Tensor = aten::cat(%3396, %1097) # <eval_with_key>.3:1224:13
%grid_sampler_2d_28.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_67.1, %124, %124, %137) # <eval_with_key>.3:1225:25
%3403 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_83.1 : Tensor = aten::view(%grid_sampler_2d_28.1, %3403) # <eval_with_key>.3:1226:14
%div_75.1 : Tensor = aten::div(%view_82.1, %1299) # <eval_with_key>.3:1229:13
%add_51.1 : Tensor = aten::add(%div_75.1, %1288, %121) # <eval_with_key>.3:1231:13
%3413 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_26.1 : Tensor[] = aten::split_with_sizes(%add_51.1, %3413, %1097) # <eval_with_key>.3:1232:26
%getitem_117.1 : Tensor = aten::__getitem__(%split_with_sizes_26.1, %124) # <eval_with_key>.3:1233:18
%getitem_118.1 : Tensor = aten::__getitem__(%split_with_sizes_26.1, %121) # <eval_with_key>.3:1234:18
%mul_78.1 : Tensor = aten::mul(%getitem_117.1, %1299) # <eval_with_key>.3:1238:13
%div_76.1 : Tensor = aten::div(%mul_78.1, %1347) # <eval_with_key>.3:1240:13
%sub_63.1 : Tensor = aten::sub(%div_76.1, %1305, %121) # <eval_with_key>.3:1242:13
%mul_79.1 : Tensor = aten::mul(%getitem_118.1, %1299) # <eval_with_key>.3:1244:13
%div_77.1 : Tensor = aten::div(%mul_79.1, %1347) # <eval_with_key>.3:1246:13
%sub_64.1 : Tensor = aten::sub(%div_77.1, %1305, %121) # <eval_with_key>.3:1248:13
%3441 : Tensor[] = prim::ListConstruct(%sub_63.1, %sub_64.1)
%cat_68.1 : Tensor = aten::cat(%3441, %1097) # <eval_with_key>.3:1249:13
%grid_sampler_2d_29.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_68.1, %124, %124, %137) # <eval_with_key>.3:1250:25
%3448 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_84.1 : Tensor = aten::view(%grid_sampler_2d_29.1, %3448) # <eval_with_key>.3:1251:14
%div_78.1 : Tensor = aten::div(%div_75.1, %1299) # <eval_with_key>.3:1254:13
%add_52.1 : Tensor = aten::add(%div_78.1, %1288, %121) # <eval_with_key>.3:1256:13
%3458 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_27.1 : Tensor[] = aten::split_with_sizes(%add_52.1, %3458, %1097) # <eval_with_key>.3:1257:26
%getitem_119.1 : Tensor = aten::__getitem__(%split_with_sizes_27.1, %124) # <eval_with_key>.3:1258:18
%getitem_120.1 : Tensor = aten::__getitem__(%split_with_sizes_27.1, %121) # <eval_with_key>.3:1259:18
%mul_80.1 : Tensor = aten::mul(%getitem_119.1, %1299) # <eval_with_key>.3:1263:13
%div_79.1 : Tensor = aten::div(%mul_80.1, %1392) # <eval_with_key>.3:1265:13
%sub_65.1 : Tensor = aten::sub(%div_79.1, %1305, %121) # <eval_with_key>.3:1267:13
%mul_81.1 : Tensor = aten::mul(%getitem_120.1, %1299) # <eval_with_key>.3:1269:13
%div_80.1 : Tensor = aten::div(%mul_81.1, %1392) # <eval_with_key>.3:1271:13
%sub_66.1 : Tensor = aten::sub(%div_80.1, %1305, %121) # <eval_with_key>.3:1273:13
%3486 : Tensor[] = prim::ListConstruct(%sub_65.1, %sub_66.1)
%cat_69.1 : Tensor = aten::cat(%3486, %1097) # <eval_with_key>.3:1274:13
%grid_sampler_2d_30.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_69.1, %124, %124, %137) # <eval_with_key>.3:1275:25
%3493 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_85.1 : Tensor = aten::view(%grid_sampler_2d_30.1, %3493) # <eval_with_key>.3:1276:14
%div_81.1 : Tensor = aten::div(%div_78.1, %1299) # <eval_with_key>.3:1279:13
%add_53.1 : Tensor = aten::add(%div_81.1, %1288, %121) # <eval_with_key>.3:1281:13
%3503 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_28.1 : Tensor[] = aten::split_with_sizes(%add_53.1, %3503, %1097) # <eval_with_key>.3:1282:26
%getitem_121.1 : Tensor = aten::__getitem__(%split_with_sizes_28.1, %124) # <eval_with_key>.3:1283:18
%getitem_122.1 : Tensor = aten::__getitem__(%split_with_sizes_28.1, %121) # <eval_with_key>.3:1284:18
%mul_82.1 : Tensor = aten::mul(%getitem_121.1, %1299) # <eval_with_key>.3:1288:13
%div_82.1 : Tensor = aten::div(%mul_82.1, %1299) # <eval_with_key>.3:1290:13
%sub_67.1 : Tensor = aten::sub(%div_82.1, %1305, %121) # <eval_with_key>.3:1292:13
%mul_83.1 : Tensor = aten::mul(%getitem_122.1, %1299) # <eval_with_key>.3:1294:13
%div_83.1 : Tensor = aten::div(%mul_83.1, %1299) # <eval_with_key>.3:1296:13
%sub_68.1 : Tensor = aten::sub(%div_83.1, %1305, %121) # <eval_with_key>.3:1298:13
%3531 : Tensor[] = prim::ListConstruct(%sub_67.1, %sub_68.1)
%cat_70.1 : Tensor = aten::cat(%3531, %1097) # <eval_with_key>.3:1299:13
%grid_sampler_2d_31.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_70.1, %124, %124, %137) # <eval_with_key>.3:1300:25
%3538 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_86.1 : Tensor = aten::view(%grid_sampler_2d_31.1, %3538) # <eval_with_key>.3:1301:14
%3544 : Tensor[] = prim::ListConstruct(%view_83.1, %view_84.1, %view_85.1, %view_86.1)
%cat_71.1 : Tensor = aten::cat(%3544, %1097) # <eval_with_key>.3:1303:13
%3548 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_14.1 : Tensor = aten::permute(%cat_71.1, %3548) # <eval_with_key>.3:1304:17
%clone_7.1 : Tensor = aten::clone(%permute_14.1, %124) # <eval_with_key>.3:1306:14
%sub_69.1 : Tensor = aten::sub(%add_49.1, %1263, %121) # <eval_with_key>.3:1308:13
%3559 : int[] = prim::ListConstruct(%121, %121)
%3560 : int[] = prim::ListConstruct(%124, %124)
%3561 : int[] = prim::ListConstruct(%121, %121)
%3563 : int[] = prim::ListConstruct(%124, %124)
%convolution_107.1 : Tensor = aten::convolution(%clone_7.1, %primals_96.1, %primals_95.1, %3559, %3560, %3561, %123, %3563, %121) # <eval_with_key>.3:1309:22
%relu__85.1 : Tensor = aten::relu_(%convolution_107.1) # <eval_with_key>.3:1310:15
%3570 : int[] = prim::ListConstruct(%121, %121)
%3571 : int[] = prim::ListConstruct(%119, %119)
%3572 : int[] = prim::ListConstruct(%121, %121)
%3574 : int[] = prim::ListConstruct(%124, %124)
%convolution_108.1 : Tensor = aten::convolution(%sub_69.1, %primals_98.1, %primals_97.1, %3570, %3571, %3572, %123, %3574, %121) # <eval_with_key>.3:1311:22
%relu__86.1 : Tensor = aten::relu_(%convolution_108.1) # <eval_with_key>.3:1312:15
%3581 : int[] = prim::ListConstruct(%121, %121)
%3582 : int[] = prim::ListConstruct(%121, %121)
%3583 : int[] = prim::ListConstruct(%121, %121)
%3585 : int[] = prim::ListConstruct(%124, %124)
%convolution_109.1 : Tensor = aten::convolution(%relu__86.1, %primals_100.1, %primals_99.1, %3581, %3582, %3583, %123, %3585, %121) # <eval_with_key>.3:1313:22
%relu__87.1 : Tensor = aten::relu_(%convolution_109.1) # <eval_with_key>.3:1314:15
%3591 : Tensor[] = prim::ListConstruct(%relu__85.1, %relu__87.1)
%cat_72.1 : Tensor = aten::cat(%3591, %121) # <eval_with_key>.3:1315:13
%3596 : int[] = prim::ListConstruct(%121, %121)
%3597 : int[] = prim::ListConstruct(%121, %121)
%3598 : int[] = prim::ListConstruct(%121, %121)
%3600 : int[] = prim::ListConstruct(%124, %124)
%convolution_110.1 : Tensor = aten::convolution(%cat_72.1, %primals_94.1, %primals_93.1, %3596, %3597, %3598, %123, %3600, %121) # <eval_with_key>.3:1316:22
%relu__88.1 : Tensor = aten::relu_(%convolution_110.1) # <eval_with_key>.3:1317:15
%3607 : Tensor[] = prim::ListConstruct(%relu.1, %relu__88.1, %sub_69.1)
%cat_74.1 : Tensor = aten::cat(%3607, %121) # <eval_with_key>.3:1319:13
%3611 : Tensor[] = prim::ListConstruct(%add_48.1, %cat_74.1)
%cat_75.1 : Tensor = aten::cat(%3611, %121) # <eval_with_key>.3:1320:13
%3616 : int[] = prim::ListConstruct(%121, %121)
%3617 : int[] = prim::ListConstruct(%121, %121)
%3618 : int[] = prim::ListConstruct(%121, %121)
%3620 : int[] = prim::ListConstruct(%124, %124)
%convolution_111.1 : Tensor = aten::convolution(%cat_75.1, %primals_106.1, %primals_105.1, %3616, %3617, %3618, %123, %3620, %121) # <eval_with_key>.3:1321:22
%sigmoid_14.1 : Tensor = aten::sigmoid(%convolution_111.1) # <eval_with_key>.3:1322:17
%3627 : int[] = prim::ListConstruct(%121, %121)
%3628 : int[] = prim::ListConstruct(%121, %121)
%3629 : int[] = prim::ListConstruct(%121, %121)
%3631 : int[] = prim::ListConstruct(%124, %124)
%convolution_112.1 : Tensor = aten::convolution(%cat_75.1, %primals_104.1, %primals_103.1, %3627, %3628, %3629, %123, %3631, %121) # <eval_with_key>.3:1323:22
%sigmoid_15.1 : Tensor = aten::sigmoid(%convolution_112.1) # <eval_with_key>.3:1324:17
%mul_84.1 : Tensor = aten::mul(%sigmoid_15.1, %add_48.1) # <eval_with_key>.3:1325:13
%3640 : Tensor[] = prim::ListConstruct(%mul_84.1, %cat_74.1)
%cat_76.1 : Tensor = aten::cat(%3640, %121) # <eval_with_key>.3:1326:13
%3645 : int[] = prim::ListConstruct(%121, %121)
%3646 : int[] = prim::ListConstruct(%121, %121)
%3647 : int[] = prim::ListConstruct(%121, %121)
%3649 : int[] = prim::ListConstruct(%124, %124)
%convolution_113.1 : Tensor = aten::convolution(%cat_76.1, %primals_102.1, %primals_101.1, %3645, %3646, %3647, %123, %3649, %121) # <eval_with_key>.3:1327:22
%tanh_8.1 : Tensor = aten::tanh(%convolution_113.1) # <eval_with_key>.3:1328:13
%rsub_7.1 : Tensor = aten::rsub(%sigmoid_14.1, %121, %121) # <eval_with_key>.3:1329:13
%mul_85.1 : Tensor = aten::mul(%rsub_7.1, %add_48.1) # <eval_with_key>.3:1330:13
%mul_86.1 : Tensor = aten::mul(%sigmoid_14.1, %tanh_8.1) # <eval_with_key>.3:1331:13
%add_54.1 : Tensor = aten::add(%mul_85.1, %mul_86.1, %121) # <eval_with_key>.3:1332:13
%3669 : int[] = prim::ListConstruct(%121, %121)
%3670 : int[] = prim::ListConstruct(%121, %121)
%3671 : int[] = prim::ListConstruct(%121, %121)
%3673 : int[] = prim::ListConstruct(%124, %124)
%convolution_114.1 : Tensor = aten::convolution(%add_54.1, %primals_90.1, %primals_89.1, %3669, %3670, %3671, %123, %3673, %121) # <eval_with_key>.3:1333:22
%relu__89.1 : Tensor = aten::relu_(%convolution_114.1) # <eval_with_key>.3:1334:15
%3680 : int[] = prim::ListConstruct(%121, %121)
%3681 : int[] = prim::ListConstruct(%121, %121)
%3682 : int[] = prim::ListConstruct(%121, %121)
%3684 : int[] = prim::ListConstruct(%124, %124)
%convolution_115.1 : Tensor = aten::convolution(%relu__89.1, %primals_92.1, %primals_91.1, %3680, %3681, %3682, %123, %3684, %121) # <eval_with_key>.3:1335:22
%add_55.1 : Tensor = aten::add(%add_49.1, %convolution_115.1, %121) # <eval_with_key>.3:1336:13
%sub_70.1 : Tensor = aten::sub(%add_55.1, %1263, %121) # <eval_with_key>.3:1338:13
%3695 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_7.1 : Tensor = aten::upsample_bilinear2d(%sub_70.1, %3695, %137, %133, %133) # <eval_with_key>.3:1339:28
%mul_87.1 : Tensor = aten::mul(%upsample_bilinear2d_7.1, %1279) # <eval_with_key>.3:1341:13
%3704 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_15.1 : Tensor = aten::permute(%add_55.1, %3704) # <eval_with_key>.3:1344:17
%3707 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_87.1 : Tensor = aten::view(%permute_15.1, %3707) # <eval_with_key>.3:1346:14
%add_56.1 : Tensor = aten::add(%view_87.1, %1288, %121) # <eval_with_key>.3:1350:13
%3714 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_29.1 : Tensor[] = aten::split_with_sizes(%add_56.1, %3714, %1097) # <eval_with_key>.3:1351:26
%getitem_123.1 : Tensor = aten::__getitem__(%split_with_sizes_29.1, %124) # <eval_with_key>.3:1352:18
%getitem_124.1 : Tensor = aten::__getitem__(%split_with_sizes_29.1, %121) # <eval_with_key>.3:1353:18
%mul_88.1 : Tensor = aten::mul(%getitem_123.1, %1299) # <eval_with_key>.3:1357:13
%div_85.1 : Tensor = aten::div(%mul_88.1, %1302) # <eval_with_key>.3:1359:13
%sub_71.1 : Tensor = aten::sub(%div_85.1, %1305, %121) # <eval_with_key>.3:1361:13
%mul_89.1 : Tensor = aten::mul(%getitem_124.1, %1299) # <eval_with_key>.3:1363:13
%div_86.1 : Tensor = aten::div(%mul_89.1, %1302) # <eval_with_key>.3:1365:13
%sub_72.1 : Tensor = aten::sub(%div_86.1, %1305, %121) # <eval_with_key>.3:1367:13
%3742 : Tensor[] = prim::ListConstruct(%sub_71.1, %sub_72.1)
%cat_77.1 : Tensor = aten::cat(%3742, %1097) # <eval_with_key>.3:1368:13
%grid_sampler_2d_32.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_77.1, %124, %124, %137) # <eval_with_key>.3:1369:25
%3749 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_88.1 : Tensor = aten::view(%grid_sampler_2d_32.1, %3749) # <eval_with_key>.3:1370:14
%div_87.1 : Tensor = aten::div(%view_87.1, %1299) # <eval_with_key>.3:1373:13
%add_57.1 : Tensor = aten::add(%div_87.1, %1288, %121) # <eval_with_key>.3:1375:13
%3759 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_30.1 : Tensor[] = aten::split_with_sizes(%add_57.1, %3759, %1097) # <eval_with_key>.3:1376:26
%getitem_125.1 : Tensor = aten::__getitem__(%split_with_sizes_30.1, %124) # <eval_with_key>.3:1377:18
%getitem_126.1 : Tensor = aten::__getitem__(%split_with_sizes_30.1, %121) # <eval_with_key>.3:1378:18
%mul_90.1 : Tensor = aten::mul(%getitem_125.1, %1299) # <eval_with_key>.3:1382:13
%div_88.1 : Tensor = aten::div(%mul_90.1, %1347) # <eval_with_key>.3:1384:13
%sub_73.1 : Tensor = aten::sub(%div_88.1, %1305, %121) # <eval_with_key>.3:1386:13
%mul_91.1 : Tensor = aten::mul(%getitem_126.1, %1299) # <eval_with_key>.3:1388:13
%div_89.1 : Tensor = aten::div(%mul_91.1, %1347) # <eval_with_key>.3:1390:13
%sub_74.1 : Tensor = aten::sub(%div_89.1, %1305, %121) # <eval_with_key>.3:1392:13
%3787 : Tensor[] = prim::ListConstruct(%sub_73.1, %sub_74.1)
%cat_78.1 : Tensor = aten::cat(%3787, %1097) # <eval_with_key>.3:1393:13
%grid_sampler_2d_33.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_78.1, %124, %124, %137) # <eval_with_key>.3:1394:25
%3794 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_89.1 : Tensor = aten::view(%grid_sampler_2d_33.1, %3794) # <eval_with_key>.3:1395:14
%div_90.1 : Tensor = aten::div(%div_87.1, %1299) # <eval_with_key>.3:1398:13
%add_58.1 : Tensor = aten::add(%div_90.1, %1288, %121) # <eval_with_key>.3:1400:13
%3804 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_31.1 : Tensor[] = aten::split_with_sizes(%add_58.1, %3804, %1097) # <eval_with_key>.3:1401:26
%getitem_127.1 : Tensor = aten::__getitem__(%split_with_sizes_31.1, %124) # <eval_with_key>.3:1402:18
%getitem_128.1 : Tensor = aten::__getitem__(%split_with_sizes_31.1, %121) # <eval_with_key>.3:1403:18
%mul_92.1 : Tensor = aten::mul(%getitem_127.1, %1299) # <eval_with_key>.3:1407:13
%div_91.1 : Tensor = aten::div(%mul_92.1, %1392) # <eval_with_key>.3:1409:13
%sub_75.1 : Tensor = aten::sub(%div_91.1, %1305, %121) # <eval_with_key>.3:1411:13
%mul_93.1 : Tensor = aten::mul(%getitem_128.1, %1299) # <eval_with_key>.3:1413:13
%div_92.1 : Tensor = aten::div(%mul_93.1, %1392) # <eval_with_key>.3:1415:13
%sub_76.1 : Tensor = aten::sub(%div_92.1, %1305, %121) # <eval_with_key>.3:1417:13
%3832 : Tensor[] = prim::ListConstruct(%sub_75.1, %sub_76.1)
%cat_79.1 : Tensor = aten::cat(%3832, %1097) # <eval_with_key>.3:1418:13
%grid_sampler_2d_34.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_79.1, %124, %124, %137) # <eval_with_key>.3:1419:25
%3839 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_90.1 : Tensor = aten::view(%grid_sampler_2d_34.1, %3839) # <eval_with_key>.3:1420:14
%div_93.1 : Tensor = aten::div(%div_90.1, %1299) # <eval_with_key>.3:1423:13
%add_59.1 : Tensor = aten::add(%div_93.1, %1288, %121) # <eval_with_key>.3:1425:13
%3849 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_32.1 : Tensor[] = aten::split_with_sizes(%add_59.1, %3849, %1097) # <eval_with_key>.3:1426:26
%getitem_129.1 : Tensor = aten::__getitem__(%split_with_sizes_32.1, %124) # <eval_with_key>.3:1427:18
%getitem_130.1 : Tensor = aten::__getitem__(%split_with_sizes_32.1, %121) # <eval_with_key>.3:1428:18
%mul_94.1 : Tensor = aten::mul(%getitem_129.1, %1299) # <eval_with_key>.3:1432:13
%div_94.1 : Tensor = aten::div(%mul_94.1, %1299) # <eval_with_key>.3:1434:13
%sub_77.1 : Tensor = aten::sub(%div_94.1, %1305, %121) # <eval_with_key>.3:1436:13
%mul_95.1 : Tensor = aten::mul(%getitem_130.1, %1299) # <eval_with_key>.3:1438:13
%div_95.1 : Tensor = aten::div(%mul_95.1, %1299) # <eval_with_key>.3:1440:13
%sub_78.1 : Tensor = aten::sub(%div_95.1, %1305, %121) # <eval_with_key>.3:1442:13
%3877 : Tensor[] = prim::ListConstruct(%sub_77.1, %sub_78.1)
%cat_80.1 : Tensor = aten::cat(%3877, %1097) # <eval_with_key>.3:1443:13
%grid_sampler_2d_35.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_80.1, %124, %124, %137) # <eval_with_key>.3:1444:25
%3884 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_91.1 : Tensor = aten::view(%grid_sampler_2d_35.1, %3884) # <eval_with_key>.3:1445:14
%3890 : Tensor[] = prim::ListConstruct(%view_88.1, %view_89.1, %view_90.1, %view_91.1)
%cat_81.1 : Tensor = aten::cat(%3890, %1097) # <eval_with_key>.3:1447:13
%3894 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_16.1 : Tensor = aten::permute(%cat_81.1, %3894) # <eval_with_key>.3:1448:17
%clone_8.1 : Tensor = aten::clone(%permute_16.1, %124) # <eval_with_key>.3:1450:14
%sub_79.1 : Tensor = aten::sub(%add_55.1, %1263, %121) # <eval_with_key>.3:1452:13
%3905 : int[] = prim::ListConstruct(%121, %121)
%3906 : int[] = prim::ListConstruct(%124, %124)
%3907 : int[] = prim::ListConstruct(%121, %121)
%3909 : int[] = prim::ListConstruct(%124, %124)
%convolution_116.1 : Tensor = aten::convolution(%clone_8.1, %primals_96.1, %primals_95.1, %3905, %3906, %3907, %123, %3909, %121) # <eval_with_key>.3:1453:22
%relu__90.1 : Tensor = aten::relu_(%convolution_116.1) # <eval_with_key>.3:1454:15
%3916 : int[] = prim::ListConstruct(%121, %121)
%3917 : int[] = prim::ListConstruct(%119, %119)
%3918 : int[] = prim::ListConstruct(%121, %121)
%3920 : int[] = prim::ListConstruct(%124, %124)
%convolution_117.1 : Tensor = aten::convolution(%sub_79.1, %primals_98.1, %primals_97.1, %3916, %3917, %3918, %123, %3920, %121) # <eval_with_key>.3:1455:22
%relu__91.1 : Tensor = aten::relu_(%convolution_117.1) # <eval_with_key>.3:1456:15
%3927 : int[] = prim::ListConstruct(%121, %121)
%3928 : int[] = prim::ListConstruct(%121, %121)
%3929 : int[] = prim::ListConstruct(%121, %121)
%3931 : int[] = prim::ListConstruct(%124, %124)
%convolution_118.1 : Tensor = aten::convolution(%relu__91.1, %primals_100.1, %primals_99.1, %3927, %3928, %3929, %123, %3931, %121) # <eval_with_key>.3:1457:22
%relu__92.1 : Tensor = aten::relu_(%convolution_118.1) # <eval_with_key>.3:1458:15
%3937 : Tensor[] = prim::ListConstruct(%relu__90.1, %relu__92.1)
%cat_82.1 : Tensor = aten::cat(%3937, %121) # <eval_with_key>.3:1459:13
%3942 : int[] = prim::ListConstruct(%121, %121)
%3943 : int[] = prim::ListConstruct(%121, %121)
%3944 : int[] = prim::ListConstruct(%121, %121)
%3946 : int[] = prim::ListConstruct(%124, %124)
%convolution_119.1 : Tensor = aten::convolution(%cat_82.1, %primals_94.1, %primals_93.1, %3942, %3943, %3944, %123, %3946, %121) # <eval_with_key>.3:1460:22
%relu__93.1 : Tensor = aten::relu_(%convolution_119.1) # <eval_with_key>.3:1461:15
%3953 : Tensor[] = prim::ListConstruct(%relu.1, %relu__93.1, %sub_79.1)
%cat_84.1 : Tensor = aten::cat(%3953, %121) # <eval_with_key>.3:1463:13
%3957 : Tensor[] = prim::ListConstruct(%add_54.1, %cat_84.1)
%cat_85.1 : Tensor = aten::cat(%3957, %121) # <eval_with_key>.3:1464:13
%3962 : int[] = prim::ListConstruct(%121, %121)
%3963 : int[] = prim::ListConstruct(%121, %121)
%3964 : int[] = prim::ListConstruct(%121, %121)
%3966 : int[] = prim::ListConstruct(%124, %124)
%convolution_120.1 : Tensor = aten::convolution(%cat_85.1, %primals_106.1, %primals_105.1, %3962, %3963, %3964, %123, %3966, %121) # <eval_with_key>.3:1465:22
%sigmoid_16.1 : Tensor = aten::sigmoid(%convolution_120.1) # <eval_with_key>.3:1466:17
%3973 : int[] = prim::ListConstruct(%121, %121)
%3974 : int[] = prim::ListConstruct(%121, %121)
%3975 : int[] = prim::ListConstruct(%121, %121)
%3977 : int[] = prim::ListConstruct(%124, %124)
%convolution_121.1 : Tensor = aten::convolution(%cat_85.1, %primals_104.1, %primals_103.1, %3973, %3974, %3975, %123, %3977, %121) # <eval_with_key>.3:1467:22
%sigmoid_17.1 : Tensor = aten::sigmoid(%convolution_121.1) # <eval_with_key>.3:1468:17
%mul_96.1 : Tensor = aten::mul(%sigmoid_17.1, %add_54.1) # <eval_with_key>.3:1469:13
%3986 : Tensor[] = prim::ListConstruct(%mul_96.1, %cat_84.1)
%cat_86.1 : Tensor = aten::cat(%3986, %121) # <eval_with_key>.3:1470:13
%3991 : int[] = prim::ListConstruct(%121, %121)
%3992 : int[] = prim::ListConstruct(%121, %121)
%3993 : int[] = prim::ListConstruct(%121, %121)
%3995 : int[] = prim::ListConstruct(%124, %124)
%convolution_122.1 : Tensor = aten::convolution(%cat_86.1, %primals_102.1, %primals_101.1, %3991, %3992, %3993, %123, %3995, %121) # <eval_with_key>.3:1471:22
%tanh_9.1 : Tensor = aten::tanh(%convolution_122.1) # <eval_with_key>.3:1472:13
%rsub_8.1 : Tensor = aten::rsub(%sigmoid_16.1, %121, %121) # <eval_with_key>.3:1473:13
%mul_97.1 : Tensor = aten::mul(%rsub_8.1, %add_54.1) # <eval_with_key>.3:1474:13
%mul_98.1 : Tensor = aten::mul(%sigmoid_16.1, %tanh_9.1) # <eval_with_key>.3:1475:13
%add_60.1 : Tensor = aten::add(%mul_97.1, %mul_98.1, %121) # <eval_with_key>.3:1476:13
%4015 : int[] = prim::ListConstruct(%121, %121)
%4016 : int[] = prim::ListConstruct(%121, %121)
%4017 : int[] = prim::ListConstruct(%121, %121)
%4019 : int[] = prim::ListConstruct(%124, %124)
%convolution_123.1 : Tensor = aten::convolution(%add_60.1, %primals_90.1, %primals_89.1, %4015, %4016, %4017, %123, %4019, %121) # <eval_with_key>.3:1477:22
%relu__94.1 : Tensor = aten::relu_(%convolution_123.1) # <eval_with_key>.3:1478:15
%4026 : int[] = prim::ListConstruct(%121, %121)
%4027 : int[] = prim::ListConstruct(%121, %121)
%4028 : int[] = prim::ListConstruct(%121, %121)
%4030 : int[] = prim::ListConstruct(%124, %124)
%convolution_124.1 : Tensor = aten::convolution(%relu__94.1, %primals_92.1, %primals_91.1, %4026, %4027, %4028, %123, %4030, %121) # <eval_with_key>.3:1479:22
%add_61.1 : Tensor = aten::add(%add_55.1, %convolution_124.1, %121) # <eval_with_key>.3:1480:13
%sub_80.1 : Tensor = aten::sub(%add_61.1, %1263, %121) # <eval_with_key>.3:1482:13
%4041 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_8.1 : Tensor = aten::upsample_bilinear2d(%sub_80.1, %4041, %137, %133, %133) # <eval_with_key>.3:1483:28
%mul_99.1 : Tensor = aten::mul(%upsample_bilinear2d_8.1, %1279) # <eval_with_key>.3:1485:13
%4050 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_17.1 : Tensor = aten::permute(%add_61.1, %4050) # <eval_with_key>.3:1488:17
%4053 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_92.1 : Tensor = aten::view(%permute_17.1, %4053) # <eval_with_key>.3:1490:14
%add_62.1 : Tensor = aten::add(%view_92.1, %1288, %121) # <eval_with_key>.3:1494:13
%4060 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_33.1 : Tensor[] = aten::split_with_sizes(%add_62.1, %4060, %1097) # <eval_with_key>.3:1495:26
%getitem_131.1 : Tensor = aten::__getitem__(%split_with_sizes_33.1, %124) # <eval_with_key>.3:1496:18
%getitem_132.1 : Tensor = aten::__getitem__(%split_with_sizes_33.1, %121) # <eval_with_key>.3:1497:18
%mul_100.1 : Tensor = aten::mul(%getitem_131.1, %1299) # <eval_with_key>.3:1501:14
%div_97.1 : Tensor = aten::div(%mul_100.1, %1302) # <eval_with_key>.3:1503:13
%sub_81.1 : Tensor = aten::sub(%div_97.1, %1305, %121) # <eval_with_key>.3:1505:13
%mul_101.1 : Tensor = aten::mul(%getitem_132.1, %1299) # <eval_with_key>.3:1507:14
%div_98.1 : Tensor = aten::div(%mul_101.1, %1302) # <eval_with_key>.3:1509:13
%sub_82.1 : Tensor = aten::sub(%div_98.1, %1305, %121) # <eval_with_key>.3:1511:13
%4088 : Tensor[] = prim::ListConstruct(%sub_81.1, %sub_82.1)
%cat_87.1 : Tensor = aten::cat(%4088, %1097) # <eval_with_key>.3:1512:13
%grid_sampler_2d_36.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_87.1, %124, %124, %137) # <eval_with_key>.3:1513:25
%4095 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_93.1 : Tensor = aten::view(%grid_sampler_2d_36.1, %4095) # <eval_with_key>.3:1514:14
%div_99.1 : Tensor = aten::div(%view_92.1, %1299) # <eval_with_key>.3:1517:13
%add_63.1 : Tensor = aten::add(%div_99.1, %1288, %121) # <eval_with_key>.3:1519:13
%4105 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_34.1 : Tensor[] = aten::split_with_sizes(%add_63.1, %4105, %1097) # <eval_with_key>.3:1520:26
%getitem_133.1 : Tensor = aten::__getitem__(%split_with_sizes_34.1, %124) # <eval_with_key>.3:1521:18
%getitem_134.1 : Tensor = aten::__getitem__(%split_with_sizes_34.1, %121) # <eval_with_key>.3:1522:18
%mul_102.1 : Tensor = aten::mul(%getitem_133.1, %1299) # <eval_with_key>.3:1526:14
%div_100.1 : Tensor = aten::div(%mul_102.1, %1347) # <eval_with_key>.3:1528:14
%sub_83.1 : Tensor = aten::sub(%div_100.1, %1305, %121) # <eval_with_key>.3:1530:13
%mul_103.1 : Tensor = aten::mul(%getitem_134.1, %1299) # <eval_with_key>.3:1532:14
%div_101.1 : Tensor = aten::div(%mul_103.1, %1347) # <eval_with_key>.3:1534:14
%sub_84.1 : Tensor = aten::sub(%div_101.1, %1305, %121) # <eval_with_key>.3:1536:13
%4133 : Tensor[] = prim::ListConstruct(%sub_83.1, %sub_84.1)
%cat_88.1 : Tensor = aten::cat(%4133, %1097) # <eval_with_key>.3:1537:13
%grid_sampler_2d_37.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_88.1, %124, %124, %137) # <eval_with_key>.3:1538:25
%4140 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_94.1 : Tensor = aten::view(%grid_sampler_2d_37.1, %4140) # <eval_with_key>.3:1539:14
%div_102.1 : Tensor = aten::div(%div_99.1, %1299) # <eval_with_key>.3:1542:14
%add_64.1 : Tensor = aten::add(%div_102.1, %1288, %121) # <eval_with_key>.3:1544:13
%4150 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_35.1 : Tensor[] = aten::split_with_sizes(%add_64.1, %4150, %1097) # <eval_with_key>.3:1545:26
%getitem_135.1 : Tensor = aten::__getitem__(%split_with_sizes_35.1, %124) # <eval_with_key>.3:1546:18
%getitem_136.1 : Tensor = aten::__getitem__(%split_with_sizes_35.1, %121) # <eval_with_key>.3:1547:18
%mul_104.1 : Tensor = aten::mul(%getitem_135.1, %1299) # <eval_with_key>.3:1551:14
%div_103.1 : Tensor = aten::div(%mul_104.1, %1392) # <eval_with_key>.3:1553:14
%sub_85.1 : Tensor = aten::sub(%div_103.1, %1305, %121) # <eval_with_key>.3:1555:13
%mul_105.1 : Tensor = aten::mul(%getitem_136.1, %1299) # <eval_with_key>.3:1557:14
%div_104.1 : Tensor = aten::div(%mul_105.1, %1392) # <eval_with_key>.3:1559:14
%sub_86.1 : Tensor = aten::sub(%div_104.1, %1305, %121) # <eval_with_key>.3:1561:13
%4178 : Tensor[] = prim::ListConstruct(%sub_85.1, %sub_86.1)
%cat_89.1 : Tensor = aten::cat(%4178, %1097) # <eval_with_key>.3:1562:13
%grid_sampler_2d_38.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_89.1, %124, %124, %137) # <eval_with_key>.3:1563:25
%4185 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_95.1 : Tensor = aten::view(%grid_sampler_2d_38.1, %4185) # <eval_with_key>.3:1564:14
%div_105.1 : Tensor = aten::div(%div_102.1, %1299) # <eval_with_key>.3:1567:14
%add_65.1 : Tensor = aten::add(%div_105.1, %1288, %121) # <eval_with_key>.3:1569:13
%4195 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_36.1 : Tensor[] = aten::split_with_sizes(%add_65.1, %4195, %1097) # <eval_with_key>.3:1570:26
%getitem_137.1 : Tensor = aten::__getitem__(%split_with_sizes_36.1, %124) # <eval_with_key>.3:1571:18
%getitem_138.1 : Tensor = aten::__getitem__(%split_with_sizes_36.1, %121) # <eval_with_key>.3:1572:18
%mul_106.1 : Tensor = aten::mul(%getitem_137.1, %1299) # <eval_with_key>.3:1576:14
%div_106.1 : Tensor = aten::div(%mul_106.1, %1299) # <eval_with_key>.3:1578:14
%sub_87.1 : Tensor = aten::sub(%div_106.1, %1305, %121) # <eval_with_key>.3:1580:13
%mul_107.1 : Tensor = aten::mul(%getitem_138.1, %1299) # <eval_with_key>.3:1582:14
%div_107.1 : Tensor = aten::div(%mul_107.1, %1299) # <eval_with_key>.3:1584:14
%sub_88.1 : Tensor = aten::sub(%div_107.1, %1305, %121) # <eval_with_key>.3:1586:13
%4223 : Tensor[] = prim::ListConstruct(%sub_87.1, %sub_88.1)
%cat_90.1 : Tensor = aten::cat(%4223, %1097) # <eval_with_key>.3:1587:13
%grid_sampler_2d_39.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_90.1, %124, %124, %137) # <eval_with_key>.3:1588:25
%4230 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_96.1 : Tensor = aten::view(%grid_sampler_2d_39.1, %4230) # <eval_with_key>.3:1589:14
%4236 : Tensor[] = prim::ListConstruct(%view_93.1, %view_94.1, %view_95.1, %view_96.1)
%cat_91.1 : Tensor = aten::cat(%4236, %1097) # <eval_with_key>.3:1591:13
%4240 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_18.1 : Tensor = aten::permute(%cat_91.1, %4240) # <eval_with_key>.3:1592:17
%clone_9.1 : Tensor = aten::clone(%permute_18.1, %124) # <eval_with_key>.3:1594:14
%sub_89.1 : Tensor = aten::sub(%add_61.1, %1263, %121) # <eval_with_key>.3:1596:13
%4251 : int[] = prim::ListConstruct(%121, %121)
%4252 : int[] = prim::ListConstruct(%124, %124)
%4253 : int[] = prim::ListConstruct(%121, %121)
%4255 : int[] = prim::ListConstruct(%124, %124)
%convolution_125.1 : Tensor = aten::convolution(%clone_9.1, %primals_96.1, %primals_95.1, %4251, %4252, %4253, %123, %4255, %121) # <eval_with_key>.3:1597:22
%relu__95.1 : Tensor = aten::relu_(%convolution_125.1) # <eval_with_key>.3:1598:15
%4262 : int[] = prim::ListConstruct(%121, %121)
%4263 : int[] = prim::ListConstruct(%119, %119)
%4264 : int[] = prim::ListConstruct(%121, %121)
%4266 : int[] = prim::ListConstruct(%124, %124)
%convolution_126.1 : Tensor = aten::convolution(%sub_89.1, %primals_98.1, %primals_97.1, %4262, %4263, %4264, %123, %4266, %121) # <eval_with_key>.3:1599:22
%relu__96.1 : Tensor = aten::relu_(%convolution_126.1) # <eval_with_key>.3:1600:15
%4273 : int[] = prim::ListConstruct(%121, %121)
%4274 : int[] = prim::ListConstruct(%121, %121)
%4275 : int[] = prim::ListConstruct(%121, %121)
%4277 : int[] = prim::ListConstruct(%124, %124)
%convolution_127.1 : Tensor = aten::convolution(%relu__96.1, %primals_100.1, %primals_99.1, %4273, %4274, %4275, %123, %4277, %121) # <eval_with_key>.3:1601:22
%relu__97.1 : Tensor = aten::relu_(%convolution_127.1) # <eval_with_key>.3:1602:15
%4283 : Tensor[] = prim::ListConstruct(%relu__95.1, %relu__97.1)
%cat_92.1 : Tensor = aten::cat(%4283, %121) # <eval_with_key>.3:1603:13
%4288 : int[] = prim::ListConstruct(%121, %121)
%4289 : int[] = prim::ListConstruct(%121, %121)
%4290 : int[] = prim::ListConstruct(%121, %121)
%4292 : int[] = prim::ListConstruct(%124, %124)
%convolution_128.1 : Tensor = aten::convolution(%cat_92.1, %primals_94.1, %primals_93.1, %4288, %4289, %4290, %123, %4292, %121) # <eval_with_key>.3:1604:22
%relu__98.1 : Tensor = aten::relu_(%convolution_128.1) # <eval_with_key>.3:1605:15
%4299 : Tensor[] = prim::ListConstruct(%relu.1, %relu__98.1, %sub_89.1)
%cat_94.1 : Tensor = aten::cat(%4299, %121) # <eval_with_key>.3:1607:13
%4303 : Tensor[] = prim::ListConstruct(%add_60.1, %cat_94.1)
%cat_95.1 : Tensor = aten::cat(%4303, %121) # <eval_with_key>.3:1608:13
%4308 : int[] = prim::ListConstruct(%121, %121)
%4309 : int[] = prim::ListConstruct(%121, %121)
%4310 : int[] = prim::ListConstruct(%121, %121)
%4312 : int[] = prim::ListConstruct(%124, %124)
%convolution_129.1 : Tensor = aten::convolution(%cat_95.1, %primals_106.1, %primals_105.1, %4308, %4309, %4310, %123, %4312, %121) # <eval_with_key>.3:1609:22
%sigmoid_18.1 : Tensor = aten::sigmoid(%convolution_129.1) # <eval_with_key>.3:1610:17
%4319 : int[] = prim::ListConstruct(%121, %121)
%4320 : int[] = prim::ListConstruct(%121, %121)
%4321 : int[] = prim::ListConstruct(%121, %121)
%4323 : int[] = prim::ListConstruct(%124, %124)
%convolution_130.1 : Tensor = aten::convolution(%cat_95.1, %primals_104.1, %primals_103.1, %4319, %4320, %4321, %123, %4323, %121) # <eval_with_key>.3:1611:22
%sigmoid_19.1 : Tensor = aten::sigmoid(%convolution_130.1) # <eval_with_key>.3:1612:17
%mul_108.1 : Tensor = aten::mul(%sigmoid_19.1, %add_60.1) # <eval_with_key>.3:1613:14
%4332 : Tensor[] = prim::ListConstruct(%mul_108.1, %cat_94.1)
%cat_96.1 : Tensor = aten::cat(%4332, %121) # <eval_with_key>.3:1614:13
%4337 : int[] = prim::ListConstruct(%121, %121)
%4338 : int[] = prim::ListConstruct(%121, %121)
%4339 : int[] = prim::ListConstruct(%121, %121)
%4341 : int[] = prim::ListConstruct(%124, %124)
%convolution_131.1 : Tensor = aten::convolution(%cat_96.1, %primals_102.1, %primals_101.1, %4337, %4338, %4339, %123, %4341, %121) # <eval_with_key>.3:1615:22
%tanh_10.1 : Tensor = aten::tanh(%convolution_131.1) # <eval_with_key>.3:1616:14
%rsub_9.1 : Tensor = aten::rsub(%sigmoid_18.1, %121, %121) # <eval_with_key>.3:1617:13
%mul_109.1 : Tensor = aten::mul(%rsub_9.1, %add_60.1) # <eval_with_key>.3:1618:14
%mul_110.1 : Tensor = aten::mul(%sigmoid_18.1, %tanh_10.1) # <eval_with_key>.3:1619:14
%add_66.1 : Tensor = aten::add(%mul_109.1, %mul_110.1, %121) # <eval_with_key>.3:1620:13
%4361 : int[] = prim::ListConstruct(%121, %121)
%4362 : int[] = prim::ListConstruct(%121, %121)
%4363 : int[] = prim::ListConstruct(%121, %121)
%4365 : int[] = prim::ListConstruct(%124, %124)
%convolution_132.1 : Tensor = aten::convolution(%add_66.1, %primals_90.1, %primals_89.1, %4361, %4362, %4363, %123, %4365, %121) # <eval_with_key>.3:1621:22
%relu__99.1 : Tensor = aten::relu_(%convolution_132.1) # <eval_with_key>.3:1622:15
%4372 : int[] = prim::ListConstruct(%121, %121)
%4373 : int[] = prim::ListConstruct(%121, %121)
%4374 : int[] = prim::ListConstruct(%121, %121)
%4376 : int[] = prim::ListConstruct(%124, %124)
%convolution_133.1 : Tensor = aten::convolution(%relu__99.1, %primals_92.1, %primals_91.1, %4372, %4373, %4374, %123, %4376, %121) # <eval_with_key>.3:1623:22
%add_67.1 : Tensor = aten::add(%add_61.1, %convolution_133.1, %121) # <eval_with_key>.3:1624:13
%sub_90.1 : Tensor = aten::sub(%add_67.1, %1263, %121) # <eval_with_key>.3:1626:13
%4387 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_9.1 : Tensor = aten::upsample_bilinear2d(%sub_90.1, %4387, %137, %133, %133) # <eval_with_key>.3:1627:28
%mul_111.1 : Tensor = aten::mul(%upsample_bilinear2d_9.1, %1279) # <eval_with_key>.3:1629:14
%4396 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_19.1 : Tensor = aten::permute(%add_67.1, %4396) # <eval_with_key>.3:1632:17
%4399 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_97.1 : Tensor = aten::view(%permute_19.1, %4399) # <eval_with_key>.3:1634:14
%add_68.1 : Tensor = aten::add(%view_97.1, %1288, %121) # <eval_with_key>.3:1638:13
%4406 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_37.1 : Tensor[] = aten::split_with_sizes(%add_68.1, %4406, %1097) # <eval_with_key>.3:1639:26
%getitem_139.1 : Tensor = aten::__getitem__(%split_with_sizes_37.1, %124) # <eval_with_key>.3:1640:18
%getitem_140.1 : Tensor = aten::__getitem__(%split_with_sizes_37.1, %121) # <eval_with_key>.3:1641:18
%mul_112.1 : Tensor = aten::mul(%getitem_139.1, %1299) # <eval_with_key>.3:1645:14
%div_109.1 : Tensor = aten::div(%mul_112.1, %1302) # <eval_with_key>.3:1647:14
%sub_91.1 : Tensor = aten::sub(%div_109.1, %1305, %121) # <eval_with_key>.3:1649:13
%mul_113.1 : Tensor = aten::mul(%getitem_140.1, %1299) # <eval_with_key>.3:1651:14
%div_110.1 : Tensor = aten::div(%mul_113.1, %1302) # <eval_with_key>.3:1653:14
%sub_92.1 : Tensor = aten::sub(%div_110.1, %1305, %121) # <eval_with_key>.3:1655:13
%4434 : Tensor[] = prim::ListConstruct(%sub_91.1, %sub_92.1)
%cat_97.1 : Tensor = aten::cat(%4434, %1097) # <eval_with_key>.3:1656:13
%grid_sampler_2d_40.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_97.1, %124, %124, %137) # <eval_with_key>.3:1657:25
%4441 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_98.1 : Tensor = aten::view(%grid_sampler_2d_40.1, %4441) # <eval_with_key>.3:1658:14
%div_111.1 : Tensor = aten::div(%view_97.1, %1299) # <eval_with_key>.3:1661:14
%add_69.1 : Tensor = aten::add(%div_111.1, %1288, %121) # <eval_with_key>.3:1663:13
%4451 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_38.1 : Tensor[] = aten::split_with_sizes(%add_69.1, %4451, %1097) # <eval_with_key>.3:1664:26
%getitem_141.1 : Tensor = aten::__getitem__(%split_with_sizes_38.1, %124) # <eval_with_key>.3:1665:18
%getitem_142.1 : Tensor = aten::__getitem__(%split_with_sizes_38.1, %121) # <eval_with_key>.3:1666:18
%mul_114.1 : Tensor = aten::mul(%getitem_141.1, %1299) # <eval_with_key>.3:1670:14
%div_112.1 : Tensor = aten::div(%mul_114.1, %1347) # <eval_with_key>.3:1672:14
%sub_93.1 : Tensor = aten::sub(%div_112.1, %1305, %121) # <eval_with_key>.3:1674:13
%mul_115.1 : Tensor = aten::mul(%getitem_142.1, %1299) # <eval_with_key>.3:1676:14
%div_113.1 : Tensor = aten::div(%mul_115.1, %1347) # <eval_with_key>.3:1678:14
%sub_94.1 : Tensor = aten::sub(%div_113.1, %1305, %121) # <eval_with_key>.3:1680:13
%4479 : Tensor[] = prim::ListConstruct(%sub_93.1, %sub_94.1)
%cat_98.1 : Tensor = aten::cat(%4479, %1097) # <eval_with_key>.3:1681:13
%grid_sampler_2d_41.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_98.1, %124, %124, %137) # <eval_with_key>.3:1682:25
%4486 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_99.1 : Tensor = aten::view(%grid_sampler_2d_41.1, %4486) # <eval_with_key>.3:1683:14
%div_114.1 : Tensor = aten::div(%div_111.1, %1299) # <eval_with_key>.3:1686:14
%add_70.1 : Tensor = aten::add(%div_114.1, %1288, %121) # <eval_with_key>.3:1688:13
%4496 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_39.1 : Tensor[] = aten::split_with_sizes(%add_70.1, %4496, %1097) # <eval_with_key>.3:1689:26
%getitem_143.1 : Tensor = aten::__getitem__(%split_with_sizes_39.1, %124) # <eval_with_key>.3:1690:18
%getitem_144.1 : Tensor = aten::__getitem__(%split_with_sizes_39.1, %121) # <eval_with_key>.3:1691:18
%mul_116.1 : Tensor = aten::mul(%getitem_143.1, %1299) # <eval_with_key>.3:1695:14
%div_115.1 : Tensor = aten::div(%mul_116.1, %1392) # <eval_with_key>.3:1697:14
%sub_95.1 : Tensor = aten::sub(%div_115.1, %1305, %121) # <eval_with_key>.3:1699:13
%mul_117.1 : Tensor = aten::mul(%getitem_144.1, %1299) # <eval_with_key>.3:1701:14
%div_116.1 : Tensor = aten::div(%mul_117.1, %1392) # <eval_with_key>.3:1703:14
%sub_96.1 : Tensor = aten::sub(%div_116.1, %1305, %121) # <eval_with_key>.3:1705:13
%4524 : Tensor[] = prim::ListConstruct(%sub_95.1, %sub_96.1)
%cat_99.1 : Tensor = aten::cat(%4524, %1097) # <eval_with_key>.3:1706:13
%grid_sampler_2d_42.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_99.1, %124, %124, %137) # <eval_with_key>.3:1707:25
%4531 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_100.1 : Tensor = aten::view(%grid_sampler_2d_42.1, %4531) # <eval_with_key>.3:1708:15
%div_117.1 : Tensor = aten::div(%div_114.1, %1299) # <eval_with_key>.3:1711:14
%add_71.1 : Tensor = aten::add(%div_117.1, %1288, %121) # <eval_with_key>.3:1713:13
%4541 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_40.1 : Tensor[] = aten::split_with_sizes(%add_71.1, %4541, %1097) # <eval_with_key>.3:1714:26
%getitem_145.1 : Tensor = aten::__getitem__(%split_with_sizes_40.1, %124) # <eval_with_key>.3:1715:18
%getitem_146.1 : Tensor = aten::__getitem__(%split_with_sizes_40.1, %121) # <eval_with_key>.3:1716:18
%mul_118.1 : Tensor = aten::mul(%getitem_145.1, %1299) # <eval_with_key>.3:1720:14
%div_118.1 : Tensor = aten::div(%mul_118.1, %1299) # <eval_with_key>.3:1722:14
%sub_97.1 : Tensor = aten::sub(%div_118.1, %1305, %121) # <eval_with_key>.3:1724:13
%mul_119.1 : Tensor = aten::mul(%getitem_146.1, %1299) # <eval_with_key>.3:1726:14
%div_119.1 : Tensor = aten::div(%mul_119.1, %1299) # <eval_with_key>.3:1728:14
%sub_98.1 : Tensor = aten::sub(%div_119.1, %1305, %121) # <eval_with_key>.3:1730:13
%4569 : Tensor[] = prim::ListConstruct(%sub_97.1, %sub_98.1)
%cat_100.1 : Tensor = aten::cat(%4569, %1097) # <eval_with_key>.3:1731:14
%grid_sampler_2d_43.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_100.1, %124, %124, %137) # <eval_with_key>.3:1732:25
%4576 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_101.1 : Tensor = aten::view(%grid_sampler_2d_43.1, %4576) # <eval_with_key>.3:1733:15
%4582 : Tensor[] = prim::ListConstruct(%view_98.1, %view_99.1, %view_100.1, %view_101.1)
%cat_101.1 : Tensor = aten::cat(%4582, %1097) # <eval_with_key>.3:1735:14
%4586 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_20.1 : Tensor = aten::permute(%cat_101.1, %4586) # <eval_with_key>.3:1736:17
%clone_10.1 : Tensor = aten::clone(%permute_20.1, %124) # <eval_with_key>.3:1738:15
%sub_99.1 : Tensor = aten::sub(%add_67.1, %1263, %121) # <eval_with_key>.3:1740:13
%4597 : int[] = prim::ListConstruct(%121, %121)
%4598 : int[] = prim::ListConstruct(%124, %124)
%4599 : int[] = prim::ListConstruct(%121, %121)
%4601 : int[] = prim::ListConstruct(%124, %124)
%convolution_134.1 : Tensor = aten::convolution(%clone_10.1, %primals_96.1, %primals_95.1, %4597, %4598, %4599, %123, %4601, %121) # <eval_with_key>.3:1741:22
%relu__100.1 : Tensor = aten::relu_(%convolution_134.1) # <eval_with_key>.3:1742:16
%4608 : int[] = prim::ListConstruct(%121, %121)
%4609 : int[] = prim::ListConstruct(%119, %119)
%4610 : int[] = prim::ListConstruct(%121, %121)
%4612 : int[] = prim::ListConstruct(%124, %124)
%convolution_135.1 : Tensor = aten::convolution(%sub_99.1, %primals_98.1, %primals_97.1, %4608, %4609, %4610, %123, %4612, %121) # <eval_with_key>.3:1743:22
%relu__101.1 : Tensor = aten::relu_(%convolution_135.1) # <eval_with_key>.3:1744:16
%4619 : int[] = prim::ListConstruct(%121, %121)
%4620 : int[] = prim::ListConstruct(%121, %121)
%4621 : int[] = prim::ListConstruct(%121, %121)
%4623 : int[] = prim::ListConstruct(%124, %124)
%convolution_136.1 : Tensor = aten::convolution(%relu__101.1, %primals_100.1, %primals_99.1, %4619, %4620, %4621, %123, %4623, %121) # <eval_with_key>.3:1745:22
%relu__102.1 : Tensor = aten::relu_(%convolution_136.1) # <eval_with_key>.3:1746:16
%4629 : Tensor[] = prim::ListConstruct(%relu__100.1, %relu__102.1)
%cat_102.1 : Tensor = aten::cat(%4629, %121) # <eval_with_key>.3:1747:14
%4634 : int[] = prim::ListConstruct(%121, %121)
%4635 : int[] = prim::ListConstruct(%121, %121)
%4636 : int[] = prim::ListConstruct(%121, %121)
%4638 : int[] = prim::ListConstruct(%124, %124)
%convolution_137.1 : Tensor = aten::convolution(%cat_102.1, %primals_94.1, %primals_93.1, %4634, %4635, %4636, %123, %4638, %121) # <eval_with_key>.3:1748:22
%relu__103.1 : Tensor = aten::relu_(%convolution_137.1) # <eval_with_key>.3:1749:16
%4645 : Tensor[] = prim::ListConstruct(%relu.1, %relu__103.1, %sub_99.1)
%cat_104.1 : Tensor = aten::cat(%4645, %121) # <eval_with_key>.3:1751:14
%4649 : Tensor[] = prim::ListConstruct(%add_66.1, %cat_104.1)
%cat_105.1 : Tensor = aten::cat(%4649, %121) # <eval_with_key>.3:1752:14
%4654 : int[] = prim::ListConstruct(%121, %121)
%4655 : int[] = prim::ListConstruct(%121, %121)
%4656 : int[] = prim::ListConstruct(%121, %121)
%4658 : int[] = prim::ListConstruct(%124, %124)
%convolution_138.1 : Tensor = aten::convolution(%cat_105.1, %primals_106.1, %primals_105.1, %4654, %4655, %4656, %123, %4658, %121) # <eval_with_key>.3:1753:22
%sigmoid_20.1 : Tensor = aten::sigmoid(%convolution_138.1) # <eval_with_key>.3:1754:17
%4665 : int[] = prim::ListConstruct(%121, %121)
%4666 : int[] = prim::ListConstruct(%121, %121)
%4667 : int[] = prim::ListConstruct(%121, %121)
%4669 : int[] = prim::ListConstruct(%124, %124)
%convolution_139.1 : Tensor = aten::convolution(%cat_105.1, %primals_104.1, %primals_103.1, %4665, %4666, %4667, %123, %4669, %121) # <eval_with_key>.3:1755:22
%sigmoid_21.1 : Tensor = aten::sigmoid(%convolution_139.1) # <eval_with_key>.3:1756:17
%mul_120.1 : Tensor = aten::mul(%sigmoid_21.1, %add_66.1) # <eval_with_key>.3:1757:14
%4678 : Tensor[] = prim::ListConstruct(%mul_120.1, %cat_104.1)
%cat_106.1 : Tensor = aten::cat(%4678, %121) # <eval_with_key>.3:1758:14
%4683 : int[] = prim::ListConstruct(%121, %121)
%4684 : int[] = prim::ListConstruct(%121, %121)
%4685 : int[] = prim::ListConstruct(%121, %121)
%4687 : int[] = prim::ListConstruct(%124, %124)
%convolution_140.1 : Tensor = aten::convolution(%cat_106.1, %primals_102.1, %primals_101.1, %4683, %4684, %4685, %123, %4687, %121) # <eval_with_key>.3:1759:22
%tanh_11.1 : Tensor = aten::tanh(%convolution_140.1) # <eval_with_key>.3:1760:14
%rsub_10.1 : Tensor = aten::rsub(%sigmoid_20.1, %121, %121) # <eval_with_key>.3:1761:14
%mul_121.1 : Tensor = aten::mul(%rsub_10.1, %add_66.1) # <eval_with_key>.3:1762:14
%mul_122.1 : Tensor = aten::mul(%sigmoid_20.1, %tanh_11.1) # <eval_with_key>.3:1763:14
%add_72.1 : Tensor = aten::add(%mul_121.1, %mul_122.1, %121) # <eval_with_key>.3:1764:13
%4707 : int[] = prim::ListConstruct(%121, %121)
%4708 : int[] = prim::ListConstruct(%121, %121)
%4709 : int[] = prim::ListConstruct(%121, %121)
%4711 : int[] = prim::ListConstruct(%124, %124)
%convolution_141.1 : Tensor = aten::convolution(%add_72.1, %primals_90.1, %primals_89.1, %4707, %4708, %4709, %123, %4711, %121) # <eval_with_key>.3:1765:22
%relu__104.1 : Tensor = aten::relu_(%convolution_141.1) # <eval_with_key>.3:1766:16
%4718 : int[] = prim::ListConstruct(%121, %121)
%4719 : int[] = prim::ListConstruct(%121, %121)
%4720 : int[] = prim::ListConstruct(%121, %121)
%4722 : int[] = prim::ListConstruct(%124, %124)
%convolution_142.1 : Tensor = aten::convolution(%relu__104.1, %primals_92.1, %primals_91.1, %4718, %4719, %4720, %123, %4722, %121) # <eval_with_key>.3:1767:22
%add_73.1 : Tensor = aten::add(%add_67.1, %convolution_142.1, %121) # <eval_with_key>.3:1768:13
%sub_100.1 : Tensor = aten::sub(%add_73.1, %1263, %121) # <eval_with_key>.3:1770:14
%4733 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_10.1 : Tensor = aten::upsample_bilinear2d(%sub_100.1, %4733, %137, %133, %133) # <eval_with_key>.3:1771:29
%mul_123.1 : Tensor = aten::mul(%upsample_bilinear2d_10.1, %1279) # <eval_with_key>.3:1773:14
%4742 : int[] = prim::ListConstruct(%124, %117, %119, %121)
%permute_21.1 : Tensor = aten::permute(%add_73.1, %4742) # <eval_with_key>.3:1776:17
%4745 : int[] = prim::ListConstruct(%748, %121, %121, %117)
%view_102.1 : Tensor = aten::view(%permute_21.1, %4745) # <eval_with_key>.3:1778:15
%add_74.1 : Tensor = aten::add(%view_102.1, %1288, %121) # <eval_with_key>.3:1782:13
%4752 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_41.1 : Tensor[] = aten::split_with_sizes(%add_74.1, %4752, %1097) # <eval_with_key>.3:1783:26
%getitem_147.1 : Tensor = aten::__getitem__(%split_with_sizes_41.1, %124) # <eval_with_key>.3:1784:18
%getitem_148.1 : Tensor = aten::__getitem__(%split_with_sizes_41.1, %121) # <eval_with_key>.3:1785:18
%mul_124.1 : Tensor = aten::mul(%getitem_147.1, %1299) # <eval_with_key>.3:1789:14
%div_121.1 : Tensor = aten::div(%mul_124.1, %1302) # <eval_with_key>.3:1791:14
%sub_101.1 : Tensor = aten::sub(%div_121.1, %1305, %121) # <eval_with_key>.3:1793:14
%mul_125.1 : Tensor = aten::mul(%getitem_148.1, %1299) # <eval_with_key>.3:1795:14
%div_122.1 : Tensor = aten::div(%mul_125.1, %1302) # <eval_with_key>.3:1797:14
%sub_102.1 : Tensor = aten::sub(%div_122.1, %1305, %121) # <eval_with_key>.3:1799:14
%4780 : Tensor[] = prim::ListConstruct(%sub_101.1, %sub_102.1)
%cat_107.1 : Tensor = aten::cat(%4780, %1097) # <eval_with_key>.3:1800:14
%grid_sampler_2d_44.1 : Tensor = aten::grid_sampler_2d(%view_47.1, %cat_107.1, %124, %124, %137) # <eval_with_key>.3:1801:25
%4787 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_103.1 : Tensor = aten::view(%grid_sampler_2d_44.1, %4787) # <eval_with_key>.3:1802:15
%div_123.1 : Tensor = aten::div(%view_102.1, %1299) # <eval_with_key>.3:1805:14
%add_75.1 : Tensor = aten::add(%div_123.1, %1288, %121) # <eval_with_key>.3:1807:13
%4797 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_42.1 : Tensor[] = aten::split_with_sizes(%add_75.1, %4797, %1097) # <eval_with_key>.3:1808:26
%getitem_149.1 : Tensor = aten::__getitem__(%split_with_sizes_42.1, %124) # <eval_with_key>.3:1809:18
%getitem_150.1 : Tensor = aten::__getitem__(%split_with_sizes_42.1, %121) # <eval_with_key>.3:1810:18
%mul_126.1 : Tensor = aten::mul(%getitem_149.1, %1299) # <eval_with_key>.3:1814:14
%div_124.1 : Tensor = aten::div(%mul_126.1, %1347) # <eval_with_key>.3:1816:14
%sub_103.1 : Tensor = aten::sub(%div_124.1, %1305, %121) # <eval_with_key>.3:1818:14
%mul_127.1 : Tensor = aten::mul(%getitem_150.1, %1299) # <eval_with_key>.3:1820:14
%div_125.1 : Tensor = aten::div(%mul_127.1, %1347) # <eval_with_key>.3:1822:14
%sub_104.1 : Tensor = aten::sub(%div_125.1, %1305, %121) # <eval_with_key>.3:1824:14
%4825 : Tensor[] = prim::ListConstruct(%sub_103.1, %sub_104.1)
%cat_108.1 : Tensor = aten::cat(%4825, %1097) # <eval_with_key>.3:1825:14
%grid_sampler_2d_45.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d.1, %cat_108.1, %124, %124, %137) # <eval_with_key>.3:1826:25
%4832 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_104.1 : Tensor = aten::view(%grid_sampler_2d_45.1, %4832) # <eval_with_key>.3:1827:15
%div_126.1 : Tensor = aten::div(%div_123.1, %1299) # <eval_with_key>.3:1830:14
%add_76.1 : Tensor = aten::add(%div_126.1, %1288, %121) # <eval_with_key>.3:1832:13
%4842 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_43.1 : Tensor[] = aten::split_with_sizes(%add_76.1, %4842, %1097) # <eval_with_key>.3:1833:26
%getitem_151.1 : Tensor = aten::__getitem__(%split_with_sizes_43.1, %124) # <eval_with_key>.3:1834:18
%getitem_152.1 : Tensor = aten::__getitem__(%split_with_sizes_43.1, %121) # <eval_with_key>.3:1835:18
%mul_128.1 : Tensor = aten::mul(%getitem_151.1, %1299) # <eval_with_key>.3:1839:14
%div_127.1 : Tensor = aten::div(%mul_128.1, %1392) # <eval_with_key>.3:1841:14
%sub_105.1 : Tensor = aten::sub(%div_127.1, %1305, %121) # <eval_with_key>.3:1843:14
%mul_129.1 : Tensor = aten::mul(%getitem_152.1, %1299) # <eval_with_key>.3:1845:14
%div_128.1 : Tensor = aten::div(%mul_129.1, %1392) # <eval_with_key>.3:1847:14
%sub_106.1 : Tensor = aten::sub(%div_128.1, %1305, %121) # <eval_with_key>.3:1849:14
%4870 : Tensor[] = prim::ListConstruct(%sub_105.1, %sub_106.1)
%cat_109.1 : Tensor = aten::cat(%4870, %1097) # <eval_with_key>.3:1850:14
%grid_sampler_2d_46.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_1.1, %cat_109.1, %124, %124, %137) # <eval_with_key>.3:1851:25
%4877 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_105.1 : Tensor = aten::view(%grid_sampler_2d_46.1, %4877) # <eval_with_key>.3:1852:15
%div_129.1 : Tensor = aten::div(%div_126.1, %1299) # <eval_with_key>.3:1855:14
%add_77.1 : Tensor = aten::add(%div_129.1, %1288, %121) # <eval_with_key>.3:1857:13
%4887 : int[] = prim::ListConstruct(%121, %121)
%split_with_sizes_44.1 : Tensor[] = aten::split_with_sizes(%add_77.1, %4887, %1097) # <eval_with_key>.3:1858:26
%getitem_153.1 : Tensor = aten::__getitem__(%split_with_sizes_44.1, %124) # <eval_with_key>.3:1859:18
%getitem_154.1 : Tensor = aten::__getitem__(%split_with_sizes_44.1, %121) # <eval_with_key>.3:1860:18
%mul_130.1 : Tensor = aten::mul(%getitem_153.1, %1299) # <eval_with_key>.3:1864:14
%div_130.1 : Tensor = aten::div(%mul_130.1, %1299) # <eval_with_key>.3:1866:14
%sub_107.1 : Tensor = aten::sub(%div_130.1, %1305, %121) # <eval_with_key>.3:1868:14
%mul_131.1 : Tensor = aten::mul(%getitem_154.1, %1299) # <eval_with_key>.3:1870:14
%div_131.1 : Tensor = aten::div(%mul_131.1, %1299) # <eval_with_key>.3:1872:14
%sub_108.1 : Tensor = aten::sub(%div_131.1, %1305, %121) # <eval_with_key>.3:1874:14
%4915 : Tensor[] = prim::ListConstruct(%sub_107.1, %sub_108.1)
%cat_110.1 : Tensor = aten::cat(%4915, %1097) # <eval_with_key>.3:1875:14
%grid_sampler_2d_47.1 : Tensor = aten::grid_sampler_2d(%avg_pool2d_2.1, %cat_110.1, %124, %124, %137) # <eval_with_key>.3:1876:25
%4922 : int[] = prim::ListConstruct(%121, %566, %566, %1097)
%view_106.1 : Tensor = aten::view(%grid_sampler_2d_47.1, %4922) # <eval_with_key>.3:1877:15
%4928 : Tensor[] = prim::ListConstruct(%view_103.1, %view_104.1, %view_105.1, %view_106.1)
%cat_111.1 : Tensor = aten::cat(%4928, %1097) # <eval_with_key>.3:1879:14
%4932 : int[] = prim::ListConstruct(%124, %119, %121, %117)
%permute_22.1 : Tensor = aten::permute(%cat_111.1, %4932) # <eval_with_key>.3:1880:17
%clone_11.1 : Tensor = aten::clone(%permute_22.1, %124) # <eval_with_key>.3:1882:15
%sub_109.1 : Tensor = aten::sub(%add_73.1, %1263, %121) # <eval_with_key>.3:1884:14
%4943 : int[] = prim::ListConstruct(%121, %121)
%4944 : int[] = prim::ListConstruct(%124, %124)
%4945 : int[] = prim::ListConstruct(%121, %121)
%4947 : int[] = prim::ListConstruct(%124, %124)
%convolution_143.1 : Tensor = aten::convolution(%clone_11.1, %primals_96.1, %primals_95.1, %4943, %4944, %4945, %123, %4947, %121) # <eval_with_key>.3:1885:22
%relu__105.1 : Tensor = aten::relu_(%convolution_143.1) # <eval_with_key>.3:1886:16
%4954 : int[] = prim::ListConstruct(%121, %121)
%4955 : int[] = prim::ListConstruct(%119, %119)
%4956 : int[] = prim::ListConstruct(%121, %121)
%4958 : int[] = prim::ListConstruct(%124, %124)
%convolution_144.1 : Tensor = aten::convolution(%sub_109.1, %primals_98.1, %primals_97.1, %4954, %4955, %4956, %123, %4958, %121) # <eval_with_key>.3:1887:22
%relu__106.1 : Tensor = aten::relu_(%convolution_144.1) # <eval_with_key>.3:1888:16
%4965 : int[] = prim::ListConstruct(%121, %121)
%4966 : int[] = prim::ListConstruct(%121, %121)
%4967 : int[] = prim::ListConstruct(%121, %121)
%4969 : int[] = prim::ListConstruct(%124, %124)
%convolution_145.1 : Tensor = aten::convolution(%relu__106.1, %primals_100.1, %primals_99.1, %4965, %4966, %4967, %123, %4969, %121) # <eval_with_key>.3:1889:22
%relu__107.1 : Tensor = aten::relu_(%convolution_145.1) # <eval_with_key>.3:1890:16
%4975 : Tensor[] = prim::ListConstruct(%relu__105.1, %relu__107.1)
%cat_112.1 : Tensor = aten::cat(%4975, %121) # <eval_with_key>.3:1891:14
%4980 : int[] = prim::ListConstruct(%121, %121)
%4981 : int[] = prim::ListConstruct(%121, %121)
%4982 : int[] = prim::ListConstruct(%121, %121)
%4984 : int[] = prim::ListConstruct(%124, %124)
%convolution_146.1 : Tensor = aten::convolution(%cat_112.1, %primals_94.1, %primals_93.1, %4980, %4981, %4982, %123, %4984, %121) # <eval_with_key>.3:1892:22
%relu__108.1 : Tensor = aten::relu_(%convolution_146.1) # <eval_with_key>.3:1893:16
%4991 : Tensor[] = prim::ListConstruct(%relu.1, %relu__108.1, %sub_109.1)
%cat_114.1 : Tensor = aten::cat(%4991, %121) # <eval_with_key>.3:1895:14
%4995 : Tensor[] = prim::ListConstruct(%add_72.1, %cat_114.1)
%cat_115.1 : Tensor = aten::cat(%4995, %121) # <eval_with_key>.3:1896:14
%5000 : int[] = prim::ListConstruct(%121, %121)
%5001 : int[] = prim::ListConstruct(%121, %121)
%5002 : int[] = prim::ListConstruct(%121, %121)
%5004 : int[] = prim::ListConstruct(%124, %124)
%convolution_147.1 : Tensor = aten::convolution(%cat_115.1, %primals_106.1, %primals_105.1, %5000, %5001, %5002, %123, %5004, %121) # <eval_with_key>.3:1897:22
%sigmoid_22.1 : Tensor = aten::sigmoid(%convolution_147.1) # <eval_with_key>.3:1898:17
%5011 : int[] = prim::ListConstruct(%121, %121)
%5012 : int[] = prim::ListConstruct(%121, %121)
%5013 : int[] = prim::ListConstruct(%121, %121)
%5015 : int[] = prim::ListConstruct(%124, %124)
%convolution_148.1 : Tensor = aten::convolution(%cat_115.1, %primals_104.1, %primals_103.1, %5011, %5012, %5013, %123, %5015, %121) # <eval_with_key>.3:1899:22
%sigmoid_23.1 : Tensor = aten::sigmoid(%convolution_148.1) # <eval_with_key>.3:1900:17
%mul_132.1 : Tensor = aten::mul(%sigmoid_23.1, %add_72.1) # <eval_with_key>.3:1901:14
%5024 : Tensor[] = prim::ListConstruct(%mul_132.1, %cat_114.1)
%cat_116.1 : Tensor = aten::cat(%5024, %121) # <eval_with_key>.3:1902:14
%5029 : int[] = prim::ListConstruct(%121, %121)
%5030 : int[] = prim::ListConstruct(%121, %121)
%5031 : int[] = prim::ListConstruct(%121, %121)
%5033 : int[] = prim::ListConstruct(%124, %124)
%convolution_149.1 : Tensor = aten::convolution(%cat_116.1, %primals_102.1, %primals_101.1, %5029, %5030, %5031, %123, %5033, %121) # <eval_with_key>.3:1903:22
%tanh_12.1 : Tensor = aten::tanh(%convolution_149.1) # <eval_with_key>.3:1904:14
%rsub_11.1 : Tensor = aten::rsub(%sigmoid_22.1, %121, %121) # <eval_with_key>.3:1905:14
%mul_133.1 : Tensor = aten::mul(%rsub_11.1, %add_72.1) # <eval_with_key>.3:1906:14
%mul_134.1 : Tensor = aten::mul(%sigmoid_22.1, %tanh_12.1) # <eval_with_key>.3:1907:14
%add_78.1 : Tensor = aten::add(%mul_133.1, %mul_134.1, %121) # <eval_with_key>.3:1908:13
%5053 : int[] = prim::ListConstruct(%121, %121)
%5054 : int[] = prim::ListConstruct(%121, %121)
%5055 : int[] = prim::ListConstruct(%121, %121)
%5057 : int[] = prim::ListConstruct(%124, %124)
%convolution_150.1 : Tensor = aten::convolution(%add_78.1, %primals_90.1, %primals_89.1, %5053, %5054, %5055, %123, %5057, %121) # <eval_with_key>.3:1909:22
%relu__109.1 : Tensor = aten::relu_(%convolution_150.1) # <eval_with_key>.3:1910:16
%5064 : int[] = prim::ListConstruct(%121, %121)
%5065 : int[] = prim::ListConstruct(%121, %121)
%5066 : int[] = prim::ListConstruct(%121, %121)
%5068 : int[] = prim::ListConstruct(%124, %124)
%convolution_151.1 : Tensor = aten::convolution(%relu__109.1, %primals_92.1, %primals_91.1, %5064, %5065, %5066, %123, %5068, %121) # <eval_with_key>.3:1911:22
%add_79.1 : Tensor = aten::add(%add_73.1, %convolution_151.1, %121) # <eval_with_key>.3:1912:13
%sub_110.1 : Tensor = aten::sub(%add_79.1, %1263, %121) # <eval_with_key>.3:1914:14
%5079 : int[] = prim::ListConstruct(%1272, %1272)
%upsample_bilinear2d_11.1 : Tensor = aten::upsample_bilinear2d(%sub_110.1, %5079, %137, %133, %133) # <eval_with_key>.3:1915:29
%mul_135.1 : Tensor = aten::mul(%upsample_bilinear2d_11.1, %1279) # <eval_with_key>.3:1917:14
%5099 : (Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor) = prim::TupleConstruct(%mul_3.1, %mul_15.1, %mul_27.1, %mul_39.1, %mul_51.1, %mul_63.1, %mul_75.1, %mul_87.1, %mul_99.1, %mul_111.1, %mul_123.1, %mul_135.1)
return (%5099)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment