Created
October 10, 2022 18:04
-
-
Save pashu123/f34c95d83514293637e1658aba849b36 to your computer and use it in GitHub Desktop.
This file has been truncated, but you can view the full file.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #loc0 = loc(unknown) | |
| module attributes {torch.debug_module_name = "_lambda"} { | |
| func.func private @__torch__.torch.fx.graph_module._lambda.forward(%arg0: !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> loc(unknown), %arg1: !torch.tensor {torch.type_bound = !torch.vtensor<[2,4,64,64],f32>} loc(unknown), %arg2: !torch.tensor {torch.type_bound = !torch.vtensor<[],si64>} loc(unknown), %arg3: !torch.tensor {torch.type_bound = !torch.vtensor<[2,77,768],f32>} loc(unknown)) -> !torch.tensor { | |
| %3919 = torch.tensor_static_info_cast %arg1 : !torch.tensor to !torch.tensor<[2,4,64,64],f32> loc(#loc0) | |
| %3920 = torch.tensor_static_info_cast %arg2 : !torch.tensor to !torch.tensor<[],si64> loc(#loc0) | |
| %3921 = torch.tensor_static_info_cast %arg3 : !torch.tensor to !torch.tensor<[2,77,768],f32> loc(#loc0) | |
| %3922 = torch.prim.GetAttr %arg0["_param_constant365"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3923 = torch.prim.GetAttr %arg0["_param_constant364"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3924 = torch.prim.GetAttr %arg0["_param_constant363"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3925 = torch.prim.GetAttr %arg0["_param_constant362"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3926 = torch.prim.GetAttr %arg0["_param_constant361"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3927 = torch.prim.GetAttr %arg0["_tensor_constant3546"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3928 = torch.prim.GetAttr %arg0["_tensor_constant3545"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3929 = torch.prim.GetAttr %arg0["_tensor_constant3542"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3930 = torch.prim.GetAttr %arg0["_tensor_constant3541"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3931 = torch.prim.GetAttr %arg0["_tensor_constant3540"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3932 = torch.prim.GetAttr %arg0["_tensor_constant3539"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3933 = torch.prim.GetAttr %arg0["_tensor_constant3538"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3934 = torch.prim.GetAttr %arg0["_param_constant360"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3935 = torch.prim.GetAttr %arg0["_tensor_constant3532"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3936 = torch.prim.GetAttr %arg0["_tensor_constant3531"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3937 = torch.prim.GetAttr %arg0["_param_constant359"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3938 = torch.prim.GetAttr %arg0["_tensor_constant3525"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3939 = torch.prim.GetAttr %arg0["_tensor_constant3524"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3940 = torch.prim.GetAttr %arg0["_param_constant358"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3941 = torch.prim.GetAttr %arg0["_tensor_constant3517"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3942 = torch.prim.GetAttr %arg0["_tensor_constant3516"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3943 = torch.prim.GetAttr %arg0["_tensor_constant3513"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3944 = torch.prim.GetAttr %arg0["_tensor_constant3512"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3945 = torch.prim.GetAttr %arg0["_tensor_constant3511"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3946 = torch.prim.GetAttr %arg0["_tensor_constant3510"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3947 = torch.prim.GetAttr %arg0["_tensor_constant3509"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3948 = torch.prim.GetAttr %arg0["_tensor_constant3503"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3949 = torch.prim.GetAttr %arg0["_tensor_constant3502"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3950 = torch.prim.GetAttr %arg0["_tensor_constant3499"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3951 = torch.prim.GetAttr %arg0["_tensor_constant3498"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3952 = torch.prim.GetAttr %arg0["_tensor_constant3497"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3953 = torch.prim.GetAttr %arg0["_tensor_constant3496"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3954 = torch.prim.GetAttr %arg0["_tensor_constant3495"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3955 = torch.prim.GetAttr %arg0["_tensor_constant3489"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3956 = torch.prim.GetAttr %arg0["_tensor_constant3488"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3957 = torch.prim.GetAttr %arg0["_tensor_constant3485"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3958 = torch.prim.GetAttr %arg0["_tensor_constant3484"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3959 = torch.prim.GetAttr %arg0["_tensor_constant3483"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3960 = torch.prim.GetAttr %arg0["_tensor_constant3482"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3961 = torch.prim.GetAttr %arg0["_tensor_constant3481"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3962 = torch.prim.GetAttr %arg0["_param_constant357"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3963 = torch.prim.GetAttr %arg0["_tensor_constant3474"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3964 = torch.prim.GetAttr %arg0["_tensor_constant3473"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3965 = torch.prim.GetAttr %arg0["_tensor_constant3470"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3966 = torch.prim.GetAttr %arg0["_tensor_constant3469"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3967 = torch.prim.GetAttr %arg0["_tensor_constant3468"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3968 = torch.prim.GetAttr %arg0["_tensor_constant3467"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3969 = torch.prim.GetAttr %arg0["_tensor_constant3466"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3970 = torch.prim.GetAttr %arg0["_param_constant356"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3971 = torch.prim.GetAttr %arg0["_tensor_constant3459"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3972 = torch.prim.GetAttr %arg0["_tensor_constant3458"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3973 = torch.prim.GetAttr %arg0["_tensor_constant3455"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3974 = torch.prim.GetAttr %arg0["_tensor_constant3454"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3975 = torch.prim.GetAttr %arg0["_tensor_constant3453"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3976 = torch.prim.GetAttr %arg0["_tensor_constant3452"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3977 = torch.prim.GetAttr %arg0["_tensor_constant3451"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3978 = torch.prim.GetAttr %arg0["_param_constant355"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3979 = torch.prim.GetAttr %arg0["_tensor_constant3444"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3980 = torch.prim.GetAttr %arg0["_tensor_constant3443"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3981 = torch.prim.GetAttr %arg0["_tensor_constant3440"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3982 = torch.prim.GetAttr %arg0["_tensor_constant3439"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3983 = torch.prim.GetAttr %arg0["_tensor_constant3438"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3984 = torch.prim.GetAttr %arg0["_tensor_constant3437"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3985 = torch.prim.GetAttr %arg0["_tensor_constant3436"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3986 = torch.prim.GetAttr %arg0["_param_constant354"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3987 = torch.prim.GetAttr %arg0["_tensor_constant3429"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3988 = torch.prim.GetAttr %arg0["_tensor_constant3428"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3989 = torch.prim.GetAttr %arg0["_tensor_constant3425"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3990 = torch.prim.GetAttr %arg0["_tensor_constant3424"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3991 = torch.prim.GetAttr %arg0["_tensor_constant3423"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3992 = torch.prim.GetAttr %arg0["_tensor_constant3422"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3993 = torch.prim.GetAttr %arg0["_tensor_constant3421"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3994 = torch.prim.GetAttr %arg0["_param_constant353"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3995 = torch.prim.GetAttr %arg0["_tensor_constant3414"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3996 = torch.prim.GetAttr %arg0["_tensor_constant3413"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3997 = torch.prim.GetAttr %arg0["_tensor_constant3410"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3998 = torch.prim.GetAttr %arg0["_tensor_constant3409"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %3999 = torch.prim.GetAttr %arg0["_tensor_constant3408"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4000 = torch.prim.GetAttr %arg0["_tensor_constant3407"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4001 = torch.prim.GetAttr %arg0["_tensor_constant3406"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4002 = torch.prim.GetAttr %arg0["_param_constant352"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4003 = torch.prim.GetAttr %arg0["_tensor_constant3399"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4004 = torch.prim.GetAttr %arg0["_tensor_constant3398"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4005 = torch.prim.GetAttr %arg0["_tensor_constant3395"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4006 = torch.prim.GetAttr %arg0["_tensor_constant3394"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4007 = torch.prim.GetAttr %arg0["_tensor_constant3393"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4008 = torch.prim.GetAttr %arg0["_tensor_constant3392"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4009 = torch.prim.GetAttr %arg0["_tensor_constant3391"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4010 = torch.prim.GetAttr %arg0["_param_constant351"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4011 = torch.prim.GetAttr %arg0["_tensor_constant3384"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4012 = torch.prim.GetAttr %arg0["_tensor_constant3383"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4013 = torch.prim.GetAttr %arg0["_tensor_constant3380"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4014 = torch.prim.GetAttr %arg0["_tensor_constant3379"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4015 = torch.prim.GetAttr %arg0["_tensor_constant3378"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4016 = torch.prim.GetAttr %arg0["_tensor_constant3377"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4017 = torch.prim.GetAttr %arg0["_tensor_constant3376"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4018 = torch.prim.GetAttr %arg0["_param_constant350"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4019 = torch.prim.GetAttr %arg0["_tensor_constant3369"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4020 = torch.prim.GetAttr %arg0["_tensor_constant3368"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4021 = torch.prim.GetAttr %arg0["_tensor_constant3365"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4022 = torch.prim.GetAttr %arg0["_tensor_constant3364"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4023 = torch.prim.GetAttr %arg0["_tensor_constant3363"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4024 = torch.prim.GetAttr %arg0["_tensor_constant3362"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4025 = torch.prim.GetAttr %arg0["_tensor_constant3361"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4026 = torch.prim.GetAttr %arg0["_param_constant349"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4027 = torch.prim.GetAttr %arg0["_param_constant348"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4028 = torch.prim.GetAttr %arg0["_param_constant347"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4029 = torch.prim.GetAttr %arg0["_param_constant346"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4030 = torch.prim.GetAttr %arg0["_param_constant345"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4031 = torch.prim.GetAttr %arg0["_tensor_constant3354"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4032 = torch.prim.GetAttr %arg0["_tensor_constant3353"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4033 = torch.prim.GetAttr %arg0["_tensor_constant3350"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4034 = torch.prim.GetAttr %arg0["_tensor_constant3349"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4035 = torch.prim.GetAttr %arg0["_tensor_constant3348"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4036 = torch.prim.GetAttr %arg0["_tensor_constant3347"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4037 = torch.prim.GetAttr %arg0["_tensor_constant3346"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4038 = torch.prim.GetAttr %arg0["_param_constant344"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4039 = torch.prim.GetAttr %arg0["_param_constant343"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4040 = torch.prim.GetAttr %arg0["_param_constant342"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4041 = torch.prim.GetAttr %arg0["_tensor_constant3339"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4042 = torch.prim.GetAttr %arg0["_tensor_constant3338"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4043 = torch.prim.GetAttr %arg0["_tensor_constant3335"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4044 = torch.prim.GetAttr %arg0["_tensor_constant3334"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4045 = torch.prim.GetAttr %arg0["_tensor_constant3333"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4046 = torch.prim.GetAttr %arg0["_tensor_constant3332"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4047 = torch.prim.GetAttr %arg0["_tensor_constant3331"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4048 = torch.prim.GetAttr %arg0["_param_constant341"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4049 = torch.prim.GetAttr %arg0["_tensor_constant3325"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4050 = torch.prim.GetAttr %arg0["_tensor_constant3324"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4051 = torch.prim.GetAttr %arg0["_param_constant340"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4052 = torch.prim.GetAttr %arg0["_tensor_constant3318"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4053 = torch.prim.GetAttr %arg0["_tensor_constant3317"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4054 = torch.prim.GetAttr %arg0["_param_constant339"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4055 = torch.prim.GetAttr %arg0["_tensor_constant3310"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4056 = torch.prim.GetAttr %arg0["_tensor_constant3309"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4057 = torch.prim.GetAttr %arg0["_tensor_constant3306"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4058 = torch.prim.GetAttr %arg0["_tensor_constant3305"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4059 = torch.prim.GetAttr %arg0["_tensor_constant3304"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4060 = torch.prim.GetAttr %arg0["_tensor_constant3303"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4061 = torch.prim.GetAttr %arg0["_tensor_constant3302"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4062 = torch.prim.GetAttr %arg0["_tensor_constant3296"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4063 = torch.prim.GetAttr %arg0["_tensor_constant3295"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4064 = torch.prim.GetAttr %arg0["_tensor_constant3292"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4065 = torch.prim.GetAttr %arg0["_tensor_constant3291"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4066 = torch.prim.GetAttr %arg0["_tensor_constant3290"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4067 = torch.prim.GetAttr %arg0["_tensor_constant3289"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4068 = torch.prim.GetAttr %arg0["_tensor_constant3288"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4069 = torch.prim.GetAttr %arg0["_tensor_constant3282"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4070 = torch.prim.GetAttr %arg0["_tensor_constant3281"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4071 = torch.prim.GetAttr %arg0["_tensor_constant3278"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4072 = torch.prim.GetAttr %arg0["_tensor_constant3277"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4073 = torch.prim.GetAttr %arg0["_tensor_constant3276"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4074 = torch.prim.GetAttr %arg0["_tensor_constant3275"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4075 = torch.prim.GetAttr %arg0["_tensor_constant3274"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4076 = torch.prim.GetAttr %arg0["_param_constant338"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4077 = torch.prim.GetAttr %arg0["_tensor_constant3267"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4078 = torch.prim.GetAttr %arg0["_tensor_constant3266"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4079 = torch.prim.GetAttr %arg0["_tensor_constant3263"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4080 = torch.prim.GetAttr %arg0["_tensor_constant3262"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4081 = torch.prim.GetAttr %arg0["_tensor_constant3261"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4082 = torch.prim.GetAttr %arg0["_tensor_constant3260"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4083 = torch.prim.GetAttr %arg0["_tensor_constant3259"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4084 = torch.prim.GetAttr %arg0["_param_constant337"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4085 = torch.prim.GetAttr %arg0["_tensor_constant3252"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4086 = torch.prim.GetAttr %arg0["_tensor_constant3251"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4087 = torch.prim.GetAttr %arg0["_tensor_constant3248"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4088 = torch.prim.GetAttr %arg0["_tensor_constant3247"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4089 = torch.prim.GetAttr %arg0["_tensor_constant3246"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4090 = torch.prim.GetAttr %arg0["_tensor_constant3245"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4091 = torch.prim.GetAttr %arg0["_tensor_constant3244"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4092 = torch.prim.GetAttr %arg0["_param_constant336"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4093 = torch.prim.GetAttr %arg0["_tensor_constant3237"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4094 = torch.prim.GetAttr %arg0["_tensor_constant3236"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4095 = torch.prim.GetAttr %arg0["_tensor_constant3233"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4096 = torch.prim.GetAttr %arg0["_tensor_constant3232"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4097 = torch.prim.GetAttr %arg0["_tensor_constant3231"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4098 = torch.prim.GetAttr %arg0["_tensor_constant3230"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4099 = torch.prim.GetAttr %arg0["_tensor_constant3229"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4100 = torch.prim.GetAttr %arg0["_param_constant335"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4101 = torch.prim.GetAttr %arg0["_tensor_constant3222"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4102 = torch.prim.GetAttr %arg0["_tensor_constant3221"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4103 = torch.prim.GetAttr %arg0["_tensor_constant3218"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4104 = torch.prim.GetAttr %arg0["_tensor_constant3217"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4105 = torch.prim.GetAttr %arg0["_tensor_constant3216"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4106 = torch.prim.GetAttr %arg0["_tensor_constant3215"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4107 = torch.prim.GetAttr %arg0["_tensor_constant3214"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4108 = torch.prim.GetAttr %arg0["_param_constant334"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4109 = torch.prim.GetAttr %arg0["_tensor_constant3207"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4110 = torch.prim.GetAttr %arg0["_tensor_constant3206"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4111 = torch.prim.GetAttr %arg0["_tensor_constant3203"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4112 = torch.prim.GetAttr %arg0["_tensor_constant3202"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4113 = torch.prim.GetAttr %arg0["_tensor_constant3201"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4114 = torch.prim.GetAttr %arg0["_tensor_constant3200"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4115 = torch.prim.GetAttr %arg0["_tensor_constant3199"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4116 = torch.prim.GetAttr %arg0["_param_constant333"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4117 = torch.prim.GetAttr %arg0["_tensor_constant3192"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4118 = torch.prim.GetAttr %arg0["_tensor_constant3191"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4119 = torch.prim.GetAttr %arg0["_tensor_constant3188"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4120 = torch.prim.GetAttr %arg0["_tensor_constant3187"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4121 = torch.prim.GetAttr %arg0["_tensor_constant3186"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4122 = torch.prim.GetAttr %arg0["_tensor_constant3185"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4123 = torch.prim.GetAttr %arg0["_tensor_constant3184"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4124 = torch.prim.GetAttr %arg0["_param_constant332"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4125 = torch.prim.GetAttr %arg0["_tensor_constant3177"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4126 = torch.prim.GetAttr %arg0["_tensor_constant3176"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4127 = torch.prim.GetAttr %arg0["_tensor_constant3173"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4128 = torch.prim.GetAttr %arg0["_tensor_constant3172"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4129 = torch.prim.GetAttr %arg0["_tensor_constant3171"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4130 = torch.prim.GetAttr %arg0["_tensor_constant3170"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4131 = torch.prim.GetAttr %arg0["_tensor_constant3169"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4132 = torch.prim.GetAttr %arg0["_param_constant331"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4133 = torch.prim.GetAttr %arg0["_tensor_constant3162"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4134 = torch.prim.GetAttr %arg0["_tensor_constant3161"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4135 = torch.prim.GetAttr %arg0["_tensor_constant3158"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4136 = torch.prim.GetAttr %arg0["_tensor_constant3157"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4137 = torch.prim.GetAttr %arg0["_tensor_constant3156"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4138 = torch.prim.GetAttr %arg0["_tensor_constant3155"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4139 = torch.prim.GetAttr %arg0["_tensor_constant3154"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4140 = torch.prim.GetAttr %arg0["_param_constant330"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4141 = torch.prim.GetAttr %arg0["_param_constant329"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4142 = torch.prim.GetAttr %arg0["_param_constant328"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4143 = torch.prim.GetAttr %arg0["_param_constant327"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4144 = torch.prim.GetAttr %arg0["_param_constant326"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4145 = torch.prim.GetAttr %arg0["_tensor_constant3147"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4146 = torch.prim.GetAttr %arg0["_tensor_constant3146"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4147 = torch.prim.GetAttr %arg0["_tensor_constant3143"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4148 = torch.prim.GetAttr %arg0["_tensor_constant3142"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4149 = torch.prim.GetAttr %arg0["_tensor_constant3141"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4150 = torch.prim.GetAttr %arg0["_tensor_constant3140"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4151 = torch.prim.GetAttr %arg0["_tensor_constant3139"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4152 = torch.prim.GetAttr %arg0["_param_constant325"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4153 = torch.prim.GetAttr %arg0["_param_constant324"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4154 = torch.prim.GetAttr %arg0["_param_constant323"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4155 = torch.prim.GetAttr %arg0["_tensor_constant3132"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4156 = torch.prim.GetAttr %arg0["_tensor_constant3131"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4157 = torch.prim.GetAttr %arg0["_tensor_constant3128"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4158 = torch.prim.GetAttr %arg0["_tensor_constant3127"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4159 = torch.prim.GetAttr %arg0["_tensor_constant3126"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4160 = torch.prim.GetAttr %arg0["_tensor_constant3125"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4161 = torch.prim.GetAttr %arg0["_tensor_constant3124"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4162 = torch.prim.GetAttr %arg0["_param_constant322"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4163 = torch.prim.GetAttr %arg0["_tensor_constant3118"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4164 = torch.prim.GetAttr %arg0["_tensor_constant3117"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4165 = torch.prim.GetAttr %arg0["_param_constant321"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4166 = torch.prim.GetAttr %arg0["_tensor_constant3111"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4167 = torch.prim.GetAttr %arg0["_tensor_constant3110"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4168 = torch.prim.GetAttr %arg0["_param_constant320"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4169 = torch.prim.GetAttr %arg0["_tensor_constant3103"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4170 = torch.prim.GetAttr %arg0["_tensor_constant3102"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4171 = torch.prim.GetAttr %arg0["_tensor_constant3099"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4172 = torch.prim.GetAttr %arg0["_tensor_constant3098"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4173 = torch.prim.GetAttr %arg0["_tensor_constant3097"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4174 = torch.prim.GetAttr %arg0["_tensor_constant3096"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4175 = torch.prim.GetAttr %arg0["_tensor_constant3095"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4176 = torch.prim.GetAttr %arg0["_tensor_constant3089"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4177 = torch.prim.GetAttr %arg0["_tensor_constant3088"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4178 = torch.prim.GetAttr %arg0["_tensor_constant3085"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4179 = torch.prim.GetAttr %arg0["_tensor_constant3084"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4180 = torch.prim.GetAttr %arg0["_tensor_constant3083"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4181 = torch.prim.GetAttr %arg0["_tensor_constant3082"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4182 = torch.prim.GetAttr %arg0["_tensor_constant3081"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4183 = torch.prim.GetAttr %arg0["_tensor_constant3075"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4184 = torch.prim.GetAttr %arg0["_tensor_constant3074"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4185 = torch.prim.GetAttr %arg0["_tensor_constant3071"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4186 = torch.prim.GetAttr %arg0["_tensor_constant3070"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4187 = torch.prim.GetAttr %arg0["_tensor_constant3069"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4188 = torch.prim.GetAttr %arg0["_tensor_constant3068"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4189 = torch.prim.GetAttr %arg0["_tensor_constant3067"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4190 = torch.prim.GetAttr %arg0["_param_constant319"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4191 = torch.prim.GetAttr %arg0["_tensor_constant3060"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4192 = torch.prim.GetAttr %arg0["_tensor_constant3059"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4193 = torch.prim.GetAttr %arg0["_tensor_constant3056"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4194 = torch.prim.GetAttr %arg0["_tensor_constant3055"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4195 = torch.prim.GetAttr %arg0["_tensor_constant3054"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4196 = torch.prim.GetAttr %arg0["_tensor_constant3053"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4197 = torch.prim.GetAttr %arg0["_tensor_constant3052"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4198 = torch.prim.GetAttr %arg0["_param_constant318"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4199 = torch.prim.GetAttr %arg0["_tensor_constant3045"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4200 = torch.prim.GetAttr %arg0["_tensor_constant3044"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4201 = torch.prim.GetAttr %arg0["_tensor_constant3041"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4202 = torch.prim.GetAttr %arg0["_tensor_constant3040"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4203 = torch.prim.GetAttr %arg0["_tensor_constant3039"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4204 = torch.prim.GetAttr %arg0["_tensor_constant3038"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4205 = torch.prim.GetAttr %arg0["_tensor_constant3037"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4206 = torch.prim.GetAttr %arg0["_param_constant317"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4207 = torch.prim.GetAttr %arg0["_tensor_constant3030"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4208 = torch.prim.GetAttr %arg0["_tensor_constant3029"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4209 = torch.prim.GetAttr %arg0["_tensor_constant3026"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4210 = torch.prim.GetAttr %arg0["_tensor_constant3025"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4211 = torch.prim.GetAttr %arg0["_tensor_constant3024"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4212 = torch.prim.GetAttr %arg0["_tensor_constant3023"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4213 = torch.prim.GetAttr %arg0["_tensor_constant3022"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4214 = torch.prim.GetAttr %arg0["_param_constant316"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4215 = torch.prim.GetAttr %arg0["_tensor_constant3015"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4216 = torch.prim.GetAttr %arg0["_tensor_constant3014"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4217 = torch.prim.GetAttr %arg0["_tensor_constant3011"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4218 = torch.prim.GetAttr %arg0["_tensor_constant3010"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4219 = torch.prim.GetAttr %arg0["_tensor_constant3009"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4220 = torch.prim.GetAttr %arg0["_tensor_constant3008"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4221 = torch.prim.GetAttr %arg0["_tensor_constant3007"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4222 = torch.prim.GetAttr %arg0["_param_constant315"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4223 = torch.prim.GetAttr %arg0["_tensor_constant3000"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4224 = torch.prim.GetAttr %arg0["_tensor_constant2999"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4225 = torch.prim.GetAttr %arg0["_tensor_constant2996"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4226 = torch.prim.GetAttr %arg0["_tensor_constant2995"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4227 = torch.prim.GetAttr %arg0["_tensor_constant2994"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4228 = torch.prim.GetAttr %arg0["_tensor_constant2993"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4229 = torch.prim.GetAttr %arg0["_tensor_constant2992"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4230 = torch.prim.GetAttr %arg0["_param_constant314"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4231 = torch.prim.GetAttr %arg0["_tensor_constant2985"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4232 = torch.prim.GetAttr %arg0["_tensor_constant2984"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4233 = torch.prim.GetAttr %arg0["_tensor_constant2981"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4234 = torch.prim.GetAttr %arg0["_tensor_constant2980"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4235 = torch.prim.GetAttr %arg0["_tensor_constant2979"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4236 = torch.prim.GetAttr %arg0["_tensor_constant2978"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4237 = torch.prim.GetAttr %arg0["_tensor_constant2977"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4238 = torch.prim.GetAttr %arg0["_param_constant313"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4239 = torch.prim.GetAttr %arg0["_tensor_constant2970"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4240 = torch.prim.GetAttr %arg0["_tensor_constant2969"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4241 = torch.prim.GetAttr %arg0["_tensor_constant2966"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4242 = torch.prim.GetAttr %arg0["_tensor_constant2965"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4243 = torch.prim.GetAttr %arg0["_tensor_constant2964"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4244 = torch.prim.GetAttr %arg0["_tensor_constant2963"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4245 = torch.prim.GetAttr %arg0["_tensor_constant2962"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4246 = torch.prim.GetAttr %arg0["_param_constant312"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4247 = torch.prim.GetAttr %arg0["_tensor_constant2955"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4248 = torch.prim.GetAttr %arg0["_tensor_constant2954"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4249 = torch.prim.GetAttr %arg0["_tensor_constant2951"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4250 = torch.prim.GetAttr %arg0["_tensor_constant2950"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4251 = torch.prim.GetAttr %arg0["_tensor_constant2949"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4252 = torch.prim.GetAttr %arg0["_tensor_constant2948"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4253 = torch.prim.GetAttr %arg0["_tensor_constant2947"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4254 = torch.prim.GetAttr %arg0["_param_constant311"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4255 = torch.prim.GetAttr %arg0["_param_constant310"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4256 = torch.prim.GetAttr %arg0["_param_constant309"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4257 = torch.prim.GetAttr %arg0["_param_constant308"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4258 = torch.prim.GetAttr %arg0["_param_constant307"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4259 = torch.prim.GetAttr %arg0["_tensor_constant2940"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4260 = torch.prim.GetAttr %arg0["_tensor_constant2939"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4261 = torch.prim.GetAttr %arg0["_tensor_constant2936"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4262 = torch.prim.GetAttr %arg0["_tensor_constant2935"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4263 = torch.prim.GetAttr %arg0["_tensor_constant2934"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4264 = torch.prim.GetAttr %arg0["_tensor_constant2933"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4265 = torch.prim.GetAttr %arg0["_tensor_constant2932"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4266 = torch.prim.GetAttr %arg0["_param_constant306"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4267 = torch.prim.GetAttr %arg0["_param_constant305"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4268 = torch.prim.GetAttr %arg0["_param_constant304"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4269 = torch.prim.GetAttr %arg0["_tensor_constant2925"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4270 = torch.prim.GetAttr %arg0["_tensor_constant2924"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4271 = torch.prim.GetAttr %arg0["_tensor_constant2921"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4272 = torch.prim.GetAttr %arg0["_tensor_constant2920"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4273 = torch.prim.GetAttr %arg0["_tensor_constant2919"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4274 = torch.prim.GetAttr %arg0["_tensor_constant2918"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4275 = torch.prim.GetAttr %arg0["_tensor_constant2917"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4276 = torch.prim.GetAttr %arg0["_param_constant303"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4277 = torch.prim.GetAttr %arg0["_tensor_constant2910"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4278 = torch.prim.GetAttr %arg0["_tensor_constant2909"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4279 = torch.prim.GetAttr %arg0["_tensor_constant2906"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4280 = torch.prim.GetAttr %arg0["_tensor_constant2905"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4281 = torch.prim.GetAttr %arg0["_tensor_constant2904"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4282 = torch.prim.GetAttr %arg0["_tensor_constant2903"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4283 = torch.prim.GetAttr %arg0["_tensor_constant2902"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4284 = torch.prim.GetAttr %arg0["_param_constant302"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4285 = torch.prim.GetAttr %arg0["_tensor_constant2896"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4286 = torch.prim.GetAttr %arg0["_tensor_constant2895"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4287 = torch.prim.GetAttr %arg0["_param_constant301"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4288 = torch.prim.GetAttr %arg0["_tensor_constant2889"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4289 = torch.prim.GetAttr %arg0["_tensor_constant2888"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4290 = torch.prim.GetAttr %arg0["_param_constant300"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4291 = torch.prim.GetAttr %arg0["_tensor_constant2881"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4292 = torch.prim.GetAttr %arg0["_tensor_constant2880"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4293 = torch.prim.GetAttr %arg0["_tensor_constant2877"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4294 = torch.prim.GetAttr %arg0["_tensor_constant2876"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4295 = torch.prim.GetAttr %arg0["_tensor_constant2875"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4296 = torch.prim.GetAttr %arg0["_tensor_constant2874"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4297 = torch.prim.GetAttr %arg0["_tensor_constant2873"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4298 = torch.prim.GetAttr %arg0["_tensor_constant2867"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4299 = torch.prim.GetAttr %arg0["_tensor_constant2866"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4300 = torch.prim.GetAttr %arg0["_tensor_constant2863"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4301 = torch.prim.GetAttr %arg0["_tensor_constant2862"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4302 = torch.prim.GetAttr %arg0["_tensor_constant2861"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4303 = torch.prim.GetAttr %arg0["_tensor_constant2860"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4304 = torch.prim.GetAttr %arg0["_tensor_constant2859"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4305 = torch.prim.GetAttr %arg0["_tensor_constant2853"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4306 = torch.prim.GetAttr %arg0["_tensor_constant2852"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4307 = torch.prim.GetAttr %arg0["_tensor_constant2849"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4308 = torch.prim.GetAttr %arg0["_tensor_constant2848"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4309 = torch.prim.GetAttr %arg0["_tensor_constant2847"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4310 = torch.prim.GetAttr %arg0["_tensor_constant2846"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4311 = torch.prim.GetAttr %arg0["_tensor_constant2845"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4312 = torch.prim.GetAttr %arg0["_param_constant299"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4313 = torch.prim.GetAttr %arg0["_tensor_constant2838"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4314 = torch.prim.GetAttr %arg0["_tensor_constant2837"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4315 = torch.prim.GetAttr %arg0["_tensor_constant2834"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4316 = torch.prim.GetAttr %arg0["_tensor_constant2833"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4317 = torch.prim.GetAttr %arg0["_tensor_constant2832"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4318 = torch.prim.GetAttr %arg0["_tensor_constant2831"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4319 = torch.prim.GetAttr %arg0["_tensor_constant2830"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4320 = torch.prim.GetAttr %arg0["_param_constant298"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4321 = torch.prim.GetAttr %arg0["_tensor_constant2823"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4322 = torch.prim.GetAttr %arg0["_tensor_constant2822"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4323 = torch.prim.GetAttr %arg0["_tensor_constant2819"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4324 = torch.prim.GetAttr %arg0["_tensor_constant2818"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4325 = torch.prim.GetAttr %arg0["_tensor_constant2817"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4326 = torch.prim.GetAttr %arg0["_tensor_constant2816"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4327 = torch.prim.GetAttr %arg0["_tensor_constant2815"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4328 = torch.prim.GetAttr %arg0["_param_constant297"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4329 = torch.prim.GetAttr %arg0["_tensor_constant2808"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4330 = torch.prim.GetAttr %arg0["_tensor_constant2807"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4331 = torch.prim.GetAttr %arg0["_tensor_constant2804"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4332 = torch.prim.GetAttr %arg0["_tensor_constant2803"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4333 = torch.prim.GetAttr %arg0["_tensor_constant2802"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4334 = torch.prim.GetAttr %arg0["_tensor_constant2801"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4335 = torch.prim.GetAttr %arg0["_tensor_constant2800"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4336 = torch.prim.GetAttr %arg0["_param_constant296"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4337 = torch.prim.GetAttr %arg0["_tensor_constant2793"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4338 = torch.prim.GetAttr %arg0["_tensor_constant2792"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4339 = torch.prim.GetAttr %arg0["_tensor_constant2789"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4340 = torch.prim.GetAttr %arg0["_tensor_constant2788"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4341 = torch.prim.GetAttr %arg0["_tensor_constant2787"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4342 = torch.prim.GetAttr %arg0["_tensor_constant2786"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4343 = torch.prim.GetAttr %arg0["_tensor_constant2785"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4344 = torch.prim.GetAttr %arg0["_param_constant295"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4345 = torch.prim.GetAttr %arg0["_tensor_constant2778"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4346 = torch.prim.GetAttr %arg0["_tensor_constant2777"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4347 = torch.prim.GetAttr %arg0["_tensor_constant2774"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4348 = torch.prim.GetAttr %arg0["_tensor_constant2773"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4349 = torch.prim.GetAttr %arg0["_tensor_constant2772"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4350 = torch.prim.GetAttr %arg0["_tensor_constant2771"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4351 = torch.prim.GetAttr %arg0["_tensor_constant2770"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4352 = torch.prim.GetAttr %arg0["_param_constant294"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4353 = torch.prim.GetAttr %arg0["_tensor_constant2763"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4354 = torch.prim.GetAttr %arg0["_tensor_constant2762"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4355 = torch.prim.GetAttr %arg0["_tensor_constant2759"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4356 = torch.prim.GetAttr %arg0["_tensor_constant2758"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4357 = torch.prim.GetAttr %arg0["_tensor_constant2757"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4358 = torch.prim.GetAttr %arg0["_tensor_constant2756"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4359 = torch.prim.GetAttr %arg0["_tensor_constant2755"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4360 = torch.prim.GetAttr %arg0["_param_constant293"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4361 = torch.prim.GetAttr %arg0["_tensor_constant2748"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4362 = torch.prim.GetAttr %arg0["_tensor_constant2747"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4363 = torch.prim.GetAttr %arg0["_tensor_constant2744"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4364 = torch.prim.GetAttr %arg0["_tensor_constant2743"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4365 = torch.prim.GetAttr %arg0["_tensor_constant2742"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4366 = torch.prim.GetAttr %arg0["_tensor_constant2741"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4367 = torch.prim.GetAttr %arg0["_tensor_constant2740"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4368 = torch.prim.GetAttr %arg0["_param_constant292"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4369 = torch.prim.GetAttr %arg0["_tensor_constant2733"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4370 = torch.prim.GetAttr %arg0["_tensor_constant2732"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4371 = torch.prim.GetAttr %arg0["_tensor_constant2729"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4372 = torch.prim.GetAttr %arg0["_tensor_constant2728"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4373 = torch.prim.GetAttr %arg0["_tensor_constant2727"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4374 = torch.prim.GetAttr %arg0["_tensor_constant2726"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4375 = torch.prim.GetAttr %arg0["_tensor_constant2725"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4376 = torch.prim.GetAttr %arg0["_param_constant291"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4377 = torch.prim.GetAttr %arg0["_param_constant290"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4378 = torch.prim.GetAttr %arg0["_param_constant289"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4379 = torch.prim.GetAttr %arg0["_param_constant288"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4380 = torch.prim.GetAttr %arg0["_param_constant287"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4381 = torch.prim.GetAttr %arg0["_tensor_constant2718"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4382 = torch.prim.GetAttr %arg0["_tensor_constant2717"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4383 = torch.prim.GetAttr %arg0["_tensor_constant2714"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4384 = torch.prim.GetAttr %arg0["_tensor_constant2713"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4385 = torch.prim.GetAttr %arg0["_tensor_constant2712"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4386 = torch.prim.GetAttr %arg0["_tensor_constant2711"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4387 = torch.prim.GetAttr %arg0["_tensor_constant2710"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4388 = torch.prim.GetAttr %arg0["_param_constant286"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4389 = torch.prim.GetAttr %arg0["_param_constant285"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4390 = torch.prim.GetAttr %arg0["_param_constant284"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4391 = torch.prim.GetAttr %arg0["_tensor_constant2703"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4392 = torch.prim.GetAttr %arg0["_tensor_constant2702"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4393 = torch.prim.GetAttr %arg0["_tensor_constant2699"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4394 = torch.prim.GetAttr %arg0["_tensor_constant2698"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4395 = torch.prim.GetAttr %arg0["_tensor_constant2697"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4396 = torch.prim.GetAttr %arg0["_tensor_constant2696"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4397 = torch.prim.GetAttr %arg0["_tensor_constant2695"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4398 = torch.prim.GetAttr %arg0["_param_constant283"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4399 = torch.prim.GetAttr %arg0["_tensor_constant2689"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4400 = torch.prim.GetAttr %arg0["_tensor_constant2688"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4401 = torch.prim.GetAttr %arg0["_param_constant282"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4402 = torch.prim.GetAttr %arg0["_tensor_constant2682"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4403 = torch.prim.GetAttr %arg0["_tensor_constant2681"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4404 = torch.prim.GetAttr %arg0["_param_constant281"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4405 = torch.prim.GetAttr %arg0["_tensor_constant2674"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4406 = torch.prim.GetAttr %arg0["_tensor_constant2673"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4407 = torch.prim.GetAttr %arg0["_tensor_constant2670"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4408 = torch.prim.GetAttr %arg0["_tensor_constant2669"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4409 = torch.prim.GetAttr %arg0["_tensor_constant2668"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4410 = torch.prim.GetAttr %arg0["_tensor_constant2667"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4411 = torch.prim.GetAttr %arg0["_tensor_constant2666"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4412 = torch.prim.GetAttr %arg0["_tensor_constant2660"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4413 = torch.prim.GetAttr %arg0["_tensor_constant2659"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4414 = torch.prim.GetAttr %arg0["_tensor_constant2656"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4415 = torch.prim.GetAttr %arg0["_tensor_constant2655"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4416 = torch.prim.GetAttr %arg0["_tensor_constant2654"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4417 = torch.prim.GetAttr %arg0["_tensor_constant2653"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4418 = torch.prim.GetAttr %arg0["_tensor_constant2652"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4419 = torch.prim.GetAttr %arg0["_tensor_constant2646"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4420 = torch.prim.GetAttr %arg0["_tensor_constant2645"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4421 = torch.prim.GetAttr %arg0["_tensor_constant2642"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4422 = torch.prim.GetAttr %arg0["_tensor_constant2641"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4423 = torch.prim.GetAttr %arg0["_tensor_constant2640"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4424 = torch.prim.GetAttr %arg0["_tensor_constant2639"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4425 = torch.prim.GetAttr %arg0["_tensor_constant2638"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4426 = torch.prim.GetAttr %arg0["_param_constant280"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4427 = torch.prim.GetAttr %arg0["_tensor_constant2631"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4428 = torch.prim.GetAttr %arg0["_tensor_constant2630"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4429 = torch.prim.GetAttr %arg0["_tensor_constant2627"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4430 = torch.prim.GetAttr %arg0["_tensor_constant2626"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4431 = torch.prim.GetAttr %arg0["_tensor_constant2625"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4432 = torch.prim.GetAttr %arg0["_tensor_constant2624"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4433 = torch.prim.GetAttr %arg0["_tensor_constant2623"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4434 = torch.prim.GetAttr %arg0["_param_constant279"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4435 = torch.prim.GetAttr %arg0["_tensor_constant2616"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4436 = torch.prim.GetAttr %arg0["_tensor_constant2615"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4437 = torch.prim.GetAttr %arg0["_tensor_constant2612"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4438 = torch.prim.GetAttr %arg0["_tensor_constant2611"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4439 = torch.prim.GetAttr %arg0["_tensor_constant2610"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4440 = torch.prim.GetAttr %arg0["_tensor_constant2609"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4441 = torch.prim.GetAttr %arg0["_tensor_constant2608"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4442 = torch.prim.GetAttr %arg0["_param_constant278"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4443 = torch.prim.GetAttr %arg0["_tensor_constant2601"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4444 = torch.prim.GetAttr %arg0["_tensor_constant2600"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4445 = torch.prim.GetAttr %arg0["_tensor_constant2597"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4446 = torch.prim.GetAttr %arg0["_tensor_constant2596"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4447 = torch.prim.GetAttr %arg0["_tensor_constant2595"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4448 = torch.prim.GetAttr %arg0["_tensor_constant2594"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4449 = torch.prim.GetAttr %arg0["_tensor_constant2593"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4450 = torch.prim.GetAttr %arg0["_param_constant277"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4451 = torch.prim.GetAttr %arg0["_tensor_constant2586"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4452 = torch.prim.GetAttr %arg0["_tensor_constant2585"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4453 = torch.prim.GetAttr %arg0["_tensor_constant2582"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4454 = torch.prim.GetAttr %arg0["_tensor_constant2581"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4455 = torch.prim.GetAttr %arg0["_tensor_constant2580"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4456 = torch.prim.GetAttr %arg0["_tensor_constant2579"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4457 = torch.prim.GetAttr %arg0["_tensor_constant2578"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4458 = torch.prim.GetAttr %arg0["_param_constant276"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4459 = torch.prim.GetAttr %arg0["_tensor_constant2571"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4460 = torch.prim.GetAttr %arg0["_tensor_constant2570"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4461 = torch.prim.GetAttr %arg0["_tensor_constant2567"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4462 = torch.prim.GetAttr %arg0["_tensor_constant2566"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4463 = torch.prim.GetAttr %arg0["_tensor_constant2565"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4464 = torch.prim.GetAttr %arg0["_tensor_constant2564"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4465 = torch.prim.GetAttr %arg0["_tensor_constant2563"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4466 = torch.prim.GetAttr %arg0["_param_constant275"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4467 = torch.prim.GetAttr %arg0["_tensor_constant2556"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4468 = torch.prim.GetAttr %arg0["_tensor_constant2555"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4469 = torch.prim.GetAttr %arg0["_tensor_constant2552"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4470 = torch.prim.GetAttr %arg0["_tensor_constant2551"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4471 = torch.prim.GetAttr %arg0["_tensor_constant2550"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4472 = torch.prim.GetAttr %arg0["_tensor_constant2549"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4473 = torch.prim.GetAttr %arg0["_tensor_constant2548"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4474 = torch.prim.GetAttr %arg0["_param_constant274"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4475 = torch.prim.GetAttr %arg0["_tensor_constant2541"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4476 = torch.prim.GetAttr %arg0["_tensor_constant2540"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4477 = torch.prim.GetAttr %arg0["_tensor_constant2537"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4478 = torch.prim.GetAttr %arg0["_tensor_constant2536"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4479 = torch.prim.GetAttr %arg0["_tensor_constant2535"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4480 = torch.prim.GetAttr %arg0["_tensor_constant2534"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4481 = torch.prim.GetAttr %arg0["_tensor_constant2533"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4482 = torch.prim.GetAttr %arg0["_param_constant273"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4483 = torch.prim.GetAttr %arg0["_tensor_constant2526"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4484 = torch.prim.GetAttr %arg0["_tensor_constant2525"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4485 = torch.prim.GetAttr %arg0["_tensor_constant2522"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4486 = torch.prim.GetAttr %arg0["_tensor_constant2521"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4487 = torch.prim.GetAttr %arg0["_tensor_constant2520"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4488 = torch.prim.GetAttr %arg0["_tensor_constant2519"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4489 = torch.prim.GetAttr %arg0["_tensor_constant2518"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4490 = torch.prim.GetAttr %arg0["_param_constant272"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4491 = torch.prim.GetAttr %arg0["_param_constant271"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4492 = torch.prim.GetAttr %arg0["_param_constant270"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4493 = torch.prim.GetAttr %arg0["_param_constant269"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4494 = torch.prim.GetAttr %arg0["_param_constant268"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4495 = torch.prim.GetAttr %arg0["_tensor_constant2511"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4496 = torch.prim.GetAttr %arg0["_tensor_constant2510"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4497 = torch.prim.GetAttr %arg0["_tensor_constant2507"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4498 = torch.prim.GetAttr %arg0["_tensor_constant2506"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4499 = torch.prim.GetAttr %arg0["_tensor_constant2505"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4500 = torch.prim.GetAttr %arg0["_tensor_constant2504"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4501 = torch.prim.GetAttr %arg0["_tensor_constant2503"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4502 = torch.prim.GetAttr %arg0["_param_constant267"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4503 = torch.prim.GetAttr %arg0["_param_constant266"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4504 = torch.prim.GetAttr %arg0["_param_constant265"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4505 = torch.prim.GetAttr %arg0["_tensor_constant2496"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4506 = torch.prim.GetAttr %arg0["_tensor_constant2495"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4507 = torch.prim.GetAttr %arg0["_tensor_constant2492"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4508 = torch.prim.GetAttr %arg0["_tensor_constant2491"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4509 = torch.prim.GetAttr %arg0["_tensor_constant2490"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4510 = torch.prim.GetAttr %arg0["_tensor_constant2489"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4511 = torch.prim.GetAttr %arg0["_tensor_constant2488"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4512 = torch.prim.GetAttr %arg0["_param_constant264"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4513 = torch.prim.GetAttr %arg0["_tensor_constant2482"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4514 = torch.prim.GetAttr %arg0["_tensor_constant2481"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4515 = torch.prim.GetAttr %arg0["_param_constant263"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4516 = torch.prim.GetAttr %arg0["_tensor_constant2475"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4517 = torch.prim.GetAttr %arg0["_tensor_constant2474"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4518 = torch.prim.GetAttr %arg0["_param_constant262"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4519 = torch.prim.GetAttr %arg0["_tensor_constant2467"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4520 = torch.prim.GetAttr %arg0["_tensor_constant2466"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4521 = torch.prim.GetAttr %arg0["_tensor_constant2463"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4522 = torch.prim.GetAttr %arg0["_tensor_constant2462"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4523 = torch.prim.GetAttr %arg0["_tensor_constant2461"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4524 = torch.prim.GetAttr %arg0["_tensor_constant2460"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4525 = torch.prim.GetAttr %arg0["_tensor_constant2459"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4526 = torch.prim.GetAttr %arg0["_tensor_constant2453"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4527 = torch.prim.GetAttr %arg0["_tensor_constant2452"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4528 = torch.prim.GetAttr %arg0["_tensor_constant2449"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4529 = torch.prim.GetAttr %arg0["_tensor_constant2448"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4530 = torch.prim.GetAttr %arg0["_tensor_constant2447"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4531 = torch.prim.GetAttr %arg0["_tensor_constant2446"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4532 = torch.prim.GetAttr %arg0["_tensor_constant2445"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4533 = torch.prim.GetAttr %arg0["_tensor_constant2439"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4534 = torch.prim.GetAttr %arg0["_tensor_constant2438"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4535 = torch.prim.GetAttr %arg0["_tensor_constant2435"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4536 = torch.prim.GetAttr %arg0["_tensor_constant2434"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4537 = torch.prim.GetAttr %arg0["_tensor_constant2433"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4538 = torch.prim.GetAttr %arg0["_tensor_constant2432"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4539 = torch.prim.GetAttr %arg0["_tensor_constant2431"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4540 = torch.prim.GetAttr %arg0["_param_constant261"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4541 = torch.prim.GetAttr %arg0["_tensor_constant2424"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4542 = torch.prim.GetAttr %arg0["_tensor_constant2423"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4543 = torch.prim.GetAttr %arg0["_tensor_constant2420"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4544 = torch.prim.GetAttr %arg0["_tensor_constant2419"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4545 = torch.prim.GetAttr %arg0["_tensor_constant2418"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4546 = torch.prim.GetAttr %arg0["_tensor_constant2417"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4547 = torch.prim.GetAttr %arg0["_tensor_constant2416"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4548 = torch.prim.GetAttr %arg0["_param_constant260"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4549 = torch.prim.GetAttr %arg0["_tensor_constant2409"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4550 = torch.prim.GetAttr %arg0["_tensor_constant2408"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4551 = torch.prim.GetAttr %arg0["_tensor_constant2405"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4552 = torch.prim.GetAttr %arg0["_tensor_constant2404"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4553 = torch.prim.GetAttr %arg0["_tensor_constant2403"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4554 = torch.prim.GetAttr %arg0["_tensor_constant2402"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4555 = torch.prim.GetAttr %arg0["_tensor_constant2401"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4556 = torch.prim.GetAttr %arg0["_param_constant259"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4557 = torch.prim.GetAttr %arg0["_tensor_constant2394"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4558 = torch.prim.GetAttr %arg0["_tensor_constant2393"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4559 = torch.prim.GetAttr %arg0["_tensor_constant2390"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4560 = torch.prim.GetAttr %arg0["_tensor_constant2389"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4561 = torch.prim.GetAttr %arg0["_tensor_constant2388"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4562 = torch.prim.GetAttr %arg0["_tensor_constant2387"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4563 = torch.prim.GetAttr %arg0["_tensor_constant2386"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4564 = torch.prim.GetAttr %arg0["_param_constant258"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4565 = torch.prim.GetAttr %arg0["_tensor_constant2379"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4566 = torch.prim.GetAttr %arg0["_tensor_constant2378"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4567 = torch.prim.GetAttr %arg0["_tensor_constant2375"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4568 = torch.prim.GetAttr %arg0["_tensor_constant2374"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4569 = torch.prim.GetAttr %arg0["_tensor_constant2373"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4570 = torch.prim.GetAttr %arg0["_tensor_constant2372"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4571 = torch.prim.GetAttr %arg0["_tensor_constant2371"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4572 = torch.prim.GetAttr %arg0["_param_constant257"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4573 = torch.prim.GetAttr %arg0["_tensor_constant2364"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4574 = torch.prim.GetAttr %arg0["_tensor_constant2363"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4575 = torch.prim.GetAttr %arg0["_tensor_constant2360"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4576 = torch.prim.GetAttr %arg0["_tensor_constant2359"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4577 = torch.prim.GetAttr %arg0["_tensor_constant2358"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4578 = torch.prim.GetAttr %arg0["_tensor_constant2357"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4579 = torch.prim.GetAttr %arg0["_tensor_constant2356"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4580 = torch.prim.GetAttr %arg0["_param_constant256"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4581 = torch.prim.GetAttr %arg0["_tensor_constant2349"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4582 = torch.prim.GetAttr %arg0["_tensor_constant2348"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4583 = torch.prim.GetAttr %arg0["_tensor_constant2345"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4584 = torch.prim.GetAttr %arg0["_tensor_constant2344"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4585 = torch.prim.GetAttr %arg0["_tensor_constant2343"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4586 = torch.prim.GetAttr %arg0["_tensor_constant2342"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4587 = torch.prim.GetAttr %arg0["_tensor_constant2341"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4588 = torch.prim.GetAttr %arg0["_param_constant255"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4589 = torch.prim.GetAttr %arg0["_tensor_constant2334"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4590 = torch.prim.GetAttr %arg0["_tensor_constant2333"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4591 = torch.prim.GetAttr %arg0["_tensor_constant2330"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4592 = torch.prim.GetAttr %arg0["_tensor_constant2329"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4593 = torch.prim.GetAttr %arg0["_tensor_constant2328"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4594 = torch.prim.GetAttr %arg0["_tensor_constant2327"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4595 = torch.prim.GetAttr %arg0["_tensor_constant2326"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4596 = torch.prim.GetAttr %arg0["_param_constant254"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4597 = torch.prim.GetAttr %arg0["_tensor_constant2319"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4598 = torch.prim.GetAttr %arg0["_tensor_constant2318"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4599 = torch.prim.GetAttr %arg0["_tensor_constant2315"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4600 = torch.prim.GetAttr %arg0["_tensor_constant2314"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4601 = torch.prim.GetAttr %arg0["_tensor_constant2313"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4602 = torch.prim.GetAttr %arg0["_tensor_constant2312"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4603 = torch.prim.GetAttr %arg0["_tensor_constant2311"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4604 = torch.prim.GetAttr %arg0["_param_constant253"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4605 = torch.prim.GetAttr %arg0["_param_constant252"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4606 = torch.prim.GetAttr %arg0["_param_constant251"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4607 = torch.prim.GetAttr %arg0["_param_constant250"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4608 = torch.prim.GetAttr %arg0["_param_constant249"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4609 = torch.prim.GetAttr %arg0["_tensor_constant2304"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4610 = torch.prim.GetAttr %arg0["_tensor_constant2303"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4611 = torch.prim.GetAttr %arg0["_tensor_constant2300"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4612 = torch.prim.GetAttr %arg0["_tensor_constant2299"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4613 = torch.prim.GetAttr %arg0["_tensor_constant2298"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4614 = torch.prim.GetAttr %arg0["_tensor_constant2297"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4615 = torch.prim.GetAttr %arg0["_tensor_constant2296"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4616 = torch.prim.GetAttr %arg0["_param_constant248"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4617 = torch.prim.GetAttr %arg0["_param_constant247"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4618 = torch.prim.GetAttr %arg0["_param_constant246"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4619 = torch.prim.GetAttr %arg0["_tensor_constant2289"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4620 = torch.prim.GetAttr %arg0["_tensor_constant2288"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4621 = torch.prim.GetAttr %arg0["_tensor_constant2285"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4622 = torch.prim.GetAttr %arg0["_tensor_constant2284"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4623 = torch.prim.GetAttr %arg0["_tensor_constant2283"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4624 = torch.prim.GetAttr %arg0["_tensor_constant2282"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4625 = torch.prim.GetAttr %arg0["_tensor_constant2281"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4626 = torch.prim.GetAttr %arg0["_param_constant245"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4627 = torch.prim.GetAttr %arg0["_tensor_constant2274"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4628 = torch.prim.GetAttr %arg0["_tensor_constant2273"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4629 = torch.prim.GetAttr %arg0["_tensor_constant2270"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4630 = torch.prim.GetAttr %arg0["_tensor_constant2269"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4631 = torch.prim.GetAttr %arg0["_tensor_constant2268"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4632 = torch.prim.GetAttr %arg0["_tensor_constant2267"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4633 = torch.prim.GetAttr %arg0["_tensor_constant2266"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4634 = torch.prim.GetAttr %arg0["_param_constant244"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4635 = torch.prim.GetAttr %arg0["_tensor_constant2260"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4636 = torch.prim.GetAttr %arg0["_tensor_constant2259"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4637 = torch.prim.GetAttr %arg0["_param_constant243"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4638 = torch.prim.GetAttr %arg0["_tensor_constant2253"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4639 = torch.prim.GetAttr %arg0["_tensor_constant2252"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4640 = torch.prim.GetAttr %arg0["_param_constant242"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4641 = torch.prim.GetAttr %arg0["_tensor_constant2245"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4642 = torch.prim.GetAttr %arg0["_tensor_constant2244"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4643 = torch.prim.GetAttr %arg0["_tensor_constant2241"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4644 = torch.prim.GetAttr %arg0["_tensor_constant2240"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4645 = torch.prim.GetAttr %arg0["_tensor_constant2239"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4646 = torch.prim.GetAttr %arg0["_tensor_constant2238"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4647 = torch.prim.GetAttr %arg0["_tensor_constant2237"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4648 = torch.prim.GetAttr %arg0["_tensor_constant2231"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4649 = torch.prim.GetAttr %arg0["_tensor_constant2230"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4650 = torch.prim.GetAttr %arg0["_tensor_constant2227"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4651 = torch.prim.GetAttr %arg0["_tensor_constant2226"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4652 = torch.prim.GetAttr %arg0["_tensor_constant2225"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4653 = torch.prim.GetAttr %arg0["_tensor_constant2224"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4654 = torch.prim.GetAttr %arg0["_tensor_constant2223"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4655 = torch.prim.GetAttr %arg0["_tensor_constant2217"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4656 = torch.prim.GetAttr %arg0["_tensor_constant2216"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4657 = torch.prim.GetAttr %arg0["_tensor_constant2213"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4658 = torch.prim.GetAttr %arg0["_tensor_constant2212"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4659 = torch.prim.GetAttr %arg0["_tensor_constant2211"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4660 = torch.prim.GetAttr %arg0["_tensor_constant2210"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4661 = torch.prim.GetAttr %arg0["_tensor_constant2209"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4662 = torch.prim.GetAttr %arg0["_param_constant241"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4663 = torch.prim.GetAttr %arg0["_tensor_constant2202"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4664 = torch.prim.GetAttr %arg0["_tensor_constant2201"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4665 = torch.prim.GetAttr %arg0["_tensor_constant2198"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4666 = torch.prim.GetAttr %arg0["_tensor_constant2197"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4667 = torch.prim.GetAttr %arg0["_tensor_constant2196"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4668 = torch.prim.GetAttr %arg0["_tensor_constant2195"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4669 = torch.prim.GetAttr %arg0["_tensor_constant2194"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4670 = torch.prim.GetAttr %arg0["_param_constant240"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4671 = torch.prim.GetAttr %arg0["_tensor_constant2187"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4672 = torch.prim.GetAttr %arg0["_tensor_constant2186"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4673 = torch.prim.GetAttr %arg0["_tensor_constant2183"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4674 = torch.prim.GetAttr %arg0["_tensor_constant2182"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4675 = torch.prim.GetAttr %arg0["_tensor_constant2181"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4676 = torch.prim.GetAttr %arg0["_tensor_constant2180"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4677 = torch.prim.GetAttr %arg0["_tensor_constant2179"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4678 = torch.prim.GetAttr %arg0["_param_constant239"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4679 = torch.prim.GetAttr %arg0["_tensor_constant2172"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4680 = torch.prim.GetAttr %arg0["_tensor_constant2171"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4681 = torch.prim.GetAttr %arg0["_tensor_constant2168"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4682 = torch.prim.GetAttr %arg0["_tensor_constant2167"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4683 = torch.prim.GetAttr %arg0["_tensor_constant2166"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4684 = torch.prim.GetAttr %arg0["_tensor_constant2165"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4685 = torch.prim.GetAttr %arg0["_tensor_constant2164"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4686 = torch.prim.GetAttr %arg0["_param_constant238"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4687 = torch.prim.GetAttr %arg0["_tensor_constant2157"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4688 = torch.prim.GetAttr %arg0["_tensor_constant2156"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4689 = torch.prim.GetAttr %arg0["_tensor_constant2153"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4690 = torch.prim.GetAttr %arg0["_tensor_constant2152"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4691 = torch.prim.GetAttr %arg0["_tensor_constant2151"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4692 = torch.prim.GetAttr %arg0["_tensor_constant2150"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4693 = torch.prim.GetAttr %arg0["_tensor_constant2149"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4694 = torch.prim.GetAttr %arg0["_param_constant237"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4695 = torch.prim.GetAttr %arg0["_tensor_constant2142"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4696 = torch.prim.GetAttr %arg0["_tensor_constant2141"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4697 = torch.prim.GetAttr %arg0["_tensor_constant2138"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4698 = torch.prim.GetAttr %arg0["_tensor_constant2137"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4699 = torch.prim.GetAttr %arg0["_tensor_constant2136"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4700 = torch.prim.GetAttr %arg0["_tensor_constant2135"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4701 = torch.prim.GetAttr %arg0["_tensor_constant2134"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4702 = torch.prim.GetAttr %arg0["_param_constant236"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4703 = torch.prim.GetAttr %arg0["_tensor_constant2127"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4704 = torch.prim.GetAttr %arg0["_tensor_constant2126"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4705 = torch.prim.GetAttr %arg0["_tensor_constant2123"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4706 = torch.prim.GetAttr %arg0["_tensor_constant2122"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4707 = torch.prim.GetAttr %arg0["_tensor_constant2121"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4708 = torch.prim.GetAttr %arg0["_tensor_constant2120"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4709 = torch.prim.GetAttr %arg0["_tensor_constant2119"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4710 = torch.prim.GetAttr %arg0["_param_constant235"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4711 = torch.prim.GetAttr %arg0["_tensor_constant2112"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4712 = torch.prim.GetAttr %arg0["_tensor_constant2111"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4713 = torch.prim.GetAttr %arg0["_tensor_constant2108"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4714 = torch.prim.GetAttr %arg0["_tensor_constant2107"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4715 = torch.prim.GetAttr %arg0["_tensor_constant2106"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4716 = torch.prim.GetAttr %arg0["_tensor_constant2105"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4717 = torch.prim.GetAttr %arg0["_tensor_constant2104"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4718 = torch.prim.GetAttr %arg0["_param_constant234"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4719 = torch.prim.GetAttr %arg0["_tensor_constant2097"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4720 = torch.prim.GetAttr %arg0["_tensor_constant2096"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4721 = torch.prim.GetAttr %arg0["_tensor_constant2093"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4722 = torch.prim.GetAttr %arg0["_tensor_constant2092"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4723 = torch.prim.GetAttr %arg0["_tensor_constant2091"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4724 = torch.prim.GetAttr %arg0["_tensor_constant2090"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4725 = torch.prim.GetAttr %arg0["_tensor_constant2089"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4726 = torch.prim.GetAttr %arg0["_param_constant233"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4727 = torch.prim.GetAttr %arg0["_param_constant232"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4728 = torch.prim.GetAttr %arg0["_param_constant231"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4729 = torch.prim.GetAttr %arg0["_param_constant230"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4730 = torch.prim.GetAttr %arg0["_param_constant229"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4731 = torch.prim.GetAttr %arg0["_tensor_constant2082"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4732 = torch.prim.GetAttr %arg0["_tensor_constant2081"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4733 = torch.prim.GetAttr %arg0["_tensor_constant2078"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4734 = torch.prim.GetAttr %arg0["_tensor_constant2077"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4735 = torch.prim.GetAttr %arg0["_tensor_constant2076"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4736 = torch.prim.GetAttr %arg0["_tensor_constant2075"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4737 = torch.prim.GetAttr %arg0["_tensor_constant2074"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4738 = torch.prim.GetAttr %arg0["_param_constant228"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4739 = torch.prim.GetAttr %arg0["_param_constant227"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4740 = torch.prim.GetAttr %arg0["_param_constant226"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4741 = torch.prim.GetAttr %arg0["_tensor_constant2067"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4742 = torch.prim.GetAttr %arg0["_tensor_constant2066"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4743 = torch.prim.GetAttr %arg0["_tensor_constant2063"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4744 = torch.prim.GetAttr %arg0["_tensor_constant2062"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4745 = torch.prim.GetAttr %arg0["_tensor_constant2061"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4746 = torch.prim.GetAttr %arg0["_tensor_constant2060"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4747 = torch.prim.GetAttr %arg0["_tensor_constant2059"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4748 = torch.prim.GetAttr %arg0["_param_constant225"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4749 = torch.prim.GetAttr %arg0["_tensor_constant2053"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4750 = torch.prim.GetAttr %arg0["_tensor_constant2052"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4751 = torch.prim.GetAttr %arg0["_param_constant224"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4752 = torch.prim.GetAttr %arg0["_tensor_constant2046"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4753 = torch.prim.GetAttr %arg0["_tensor_constant2045"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4754 = torch.prim.GetAttr %arg0["_param_constant223"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4755 = torch.prim.GetAttr %arg0["_tensor_constant2038"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4756 = torch.prim.GetAttr %arg0["_tensor_constant2037"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4757 = torch.prim.GetAttr %arg0["_tensor_constant2034"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4758 = torch.prim.GetAttr %arg0["_tensor_constant2033"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4759 = torch.prim.GetAttr %arg0["_tensor_constant2032"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4760 = torch.prim.GetAttr %arg0["_tensor_constant2031"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4761 = torch.prim.GetAttr %arg0["_tensor_constant2030"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4762 = torch.prim.GetAttr %arg0["_tensor_constant2024"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4763 = torch.prim.GetAttr %arg0["_tensor_constant2023"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4764 = torch.prim.GetAttr %arg0["_tensor_constant2020"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4765 = torch.prim.GetAttr %arg0["_tensor_constant2019"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4766 = torch.prim.GetAttr %arg0["_tensor_constant2018"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4767 = torch.prim.GetAttr %arg0["_tensor_constant2017"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4768 = torch.prim.GetAttr %arg0["_tensor_constant2016"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4769 = torch.prim.GetAttr %arg0["_tensor_constant2010"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4770 = torch.prim.GetAttr %arg0["_tensor_constant2009"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4771 = torch.prim.GetAttr %arg0["_tensor_constant2006"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4772 = torch.prim.GetAttr %arg0["_tensor_constant2005"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4773 = torch.prim.GetAttr %arg0["_tensor_constant2004"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4774 = torch.prim.GetAttr %arg0["_tensor_constant2003"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4775 = torch.prim.GetAttr %arg0["_tensor_constant2002"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4776 = torch.prim.GetAttr %arg0["_param_constant222"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4777 = torch.prim.GetAttr %arg0["_tensor_constant1995"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4778 = torch.prim.GetAttr %arg0["_tensor_constant1994"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4779 = torch.prim.GetAttr %arg0["_tensor_constant1991"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4780 = torch.prim.GetAttr %arg0["_tensor_constant1990"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4781 = torch.prim.GetAttr %arg0["_tensor_constant1989"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4782 = torch.prim.GetAttr %arg0["_tensor_constant1988"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4783 = torch.prim.GetAttr %arg0["_tensor_constant1987"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4784 = torch.prim.GetAttr %arg0["_param_constant221"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4785 = torch.prim.GetAttr %arg0["_tensor_constant1980"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4786 = torch.prim.GetAttr %arg0["_tensor_constant1979"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4787 = torch.prim.GetAttr %arg0["_tensor_constant1976"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4788 = torch.prim.GetAttr %arg0["_tensor_constant1975"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4789 = torch.prim.GetAttr %arg0["_tensor_constant1974"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4790 = torch.prim.GetAttr %arg0["_tensor_constant1973"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4791 = torch.prim.GetAttr %arg0["_tensor_constant1972"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4792 = torch.prim.GetAttr %arg0["_param_constant220"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4793 = torch.prim.GetAttr %arg0["_tensor_constant1965"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4794 = torch.prim.GetAttr %arg0["_tensor_constant1964"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4795 = torch.prim.GetAttr %arg0["_tensor_constant1961"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4796 = torch.prim.GetAttr %arg0["_tensor_constant1960"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4797 = torch.prim.GetAttr %arg0["_tensor_constant1959"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4798 = torch.prim.GetAttr %arg0["_tensor_constant1958"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4799 = torch.prim.GetAttr %arg0["_tensor_constant1957"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4800 = torch.prim.GetAttr %arg0["_param_constant219"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4801 = torch.prim.GetAttr %arg0["_tensor_constant1950"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4802 = torch.prim.GetAttr %arg0["_tensor_constant1949"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4803 = torch.prim.GetAttr %arg0["_tensor_constant1946"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4804 = torch.prim.GetAttr %arg0["_tensor_constant1945"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4805 = torch.prim.GetAttr %arg0["_tensor_constant1944"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4806 = torch.prim.GetAttr %arg0["_tensor_constant1943"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4807 = torch.prim.GetAttr %arg0["_tensor_constant1942"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4808 = torch.prim.GetAttr %arg0["_param_constant218"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4809 = torch.prim.GetAttr %arg0["_tensor_constant1935"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4810 = torch.prim.GetAttr %arg0["_tensor_constant1934"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4811 = torch.prim.GetAttr %arg0["_tensor_constant1931"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4812 = torch.prim.GetAttr %arg0["_tensor_constant1930"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4813 = torch.prim.GetAttr %arg0["_tensor_constant1929"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4814 = torch.prim.GetAttr %arg0["_tensor_constant1928"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4815 = torch.prim.GetAttr %arg0["_tensor_constant1927"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4816 = torch.prim.GetAttr %arg0["_param_constant217"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4817 = torch.prim.GetAttr %arg0["_tensor_constant1920"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4818 = torch.prim.GetAttr %arg0["_tensor_constant1919"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4819 = torch.prim.GetAttr %arg0["_tensor_constant1916"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4820 = torch.prim.GetAttr %arg0["_tensor_constant1915"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4821 = torch.prim.GetAttr %arg0["_tensor_constant1914"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4822 = torch.prim.GetAttr %arg0["_tensor_constant1913"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4823 = torch.prim.GetAttr %arg0["_tensor_constant1912"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4824 = torch.prim.GetAttr %arg0["_param_constant216"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4825 = torch.prim.GetAttr %arg0["_tensor_constant1905"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4826 = torch.prim.GetAttr %arg0["_tensor_constant1904"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4827 = torch.prim.GetAttr %arg0["_tensor_constant1901"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4828 = torch.prim.GetAttr %arg0["_tensor_constant1900"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4829 = torch.prim.GetAttr %arg0["_tensor_constant1899"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4830 = torch.prim.GetAttr %arg0["_tensor_constant1898"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4831 = torch.prim.GetAttr %arg0["_tensor_constant1897"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4832 = torch.prim.GetAttr %arg0["_param_constant215"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4833 = torch.prim.GetAttr %arg0["_tensor_constant1890"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4834 = torch.prim.GetAttr %arg0["_tensor_constant1889"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4835 = torch.prim.GetAttr %arg0["_tensor_constant1886"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4836 = torch.prim.GetAttr %arg0["_tensor_constant1885"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4837 = torch.prim.GetAttr %arg0["_tensor_constant1884"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4838 = torch.prim.GetAttr %arg0["_tensor_constant1883"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4839 = torch.prim.GetAttr %arg0["_tensor_constant1882"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4840 = torch.prim.GetAttr %arg0["_param_constant214"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4841 = torch.prim.GetAttr %arg0["_param_constant213"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4842 = torch.prim.GetAttr %arg0["_param_constant212"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4843 = torch.prim.GetAttr %arg0["_param_constant211"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4844 = torch.prim.GetAttr %arg0["_param_constant210"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4845 = torch.prim.GetAttr %arg0["_tensor_constant1875"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4846 = torch.prim.GetAttr %arg0["_tensor_constant1874"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4847 = torch.prim.GetAttr %arg0["_tensor_constant1871"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4848 = torch.prim.GetAttr %arg0["_tensor_constant1870"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4849 = torch.prim.GetAttr %arg0["_tensor_constant1869"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4850 = torch.prim.GetAttr %arg0["_tensor_constant1868"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4851 = torch.prim.GetAttr %arg0["_tensor_constant1867"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4852 = torch.prim.GetAttr %arg0["_param_constant209"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4853 = torch.prim.GetAttr %arg0["_param_constant208"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4854 = torch.prim.GetAttr %arg0["_param_constant207"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4855 = torch.prim.GetAttr %arg0["_tensor_constant1860"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4856 = torch.prim.GetAttr %arg0["_tensor_constant1859"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4857 = torch.prim.GetAttr %arg0["_tensor_constant1856"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4858 = torch.prim.GetAttr %arg0["_tensor_constant1855"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4859 = torch.prim.GetAttr %arg0["_tensor_constant1854"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4860 = torch.prim.GetAttr %arg0["_tensor_constant1853"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4861 = torch.prim.GetAttr %arg0["_tensor_constant1852"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4862 = torch.prim.GetAttr %arg0["_param_constant206"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4863 = torch.prim.GetAttr %arg0["_tensor_constant1846"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4864 = torch.prim.GetAttr %arg0["_tensor_constant1845"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4865 = torch.prim.GetAttr %arg0["_param_constant205"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4866 = torch.prim.GetAttr %arg0["_tensor_constant1839"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4867 = torch.prim.GetAttr %arg0["_tensor_constant1838"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4868 = torch.prim.GetAttr %arg0["_param_constant204"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4869 = torch.prim.GetAttr %arg0["_tensor_constant1831"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4870 = torch.prim.GetAttr %arg0["_tensor_constant1830"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4871 = torch.prim.GetAttr %arg0["_tensor_constant1827"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4872 = torch.prim.GetAttr %arg0["_tensor_constant1826"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4873 = torch.prim.GetAttr %arg0["_tensor_constant1825"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4874 = torch.prim.GetAttr %arg0["_tensor_constant1824"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4875 = torch.prim.GetAttr %arg0["_tensor_constant1823"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4876 = torch.prim.GetAttr %arg0["_tensor_constant1817"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4877 = torch.prim.GetAttr %arg0["_tensor_constant1816"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4878 = torch.prim.GetAttr %arg0["_tensor_constant1813"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4879 = torch.prim.GetAttr %arg0["_tensor_constant1812"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4880 = torch.prim.GetAttr %arg0["_tensor_constant1811"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4881 = torch.prim.GetAttr %arg0["_tensor_constant1810"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4882 = torch.prim.GetAttr %arg0["_tensor_constant1809"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4883 = torch.prim.GetAttr %arg0["_tensor_constant1803"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4884 = torch.prim.GetAttr %arg0["_tensor_constant1802"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4885 = torch.prim.GetAttr %arg0["_tensor_constant1799"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4886 = torch.prim.GetAttr %arg0["_tensor_constant1798"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4887 = torch.prim.GetAttr %arg0["_tensor_constant1797"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4888 = torch.prim.GetAttr %arg0["_tensor_constant1796"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4889 = torch.prim.GetAttr %arg0["_tensor_constant1795"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4890 = torch.prim.GetAttr %arg0["_param_constant203"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4891 = torch.prim.GetAttr %arg0["_tensor_constant1788"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4892 = torch.prim.GetAttr %arg0["_tensor_constant1787"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4893 = torch.prim.GetAttr %arg0["_tensor_constant1784"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4894 = torch.prim.GetAttr %arg0["_tensor_constant1783"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4895 = torch.prim.GetAttr %arg0["_tensor_constant1782"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4896 = torch.prim.GetAttr %arg0["_tensor_constant1781"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4897 = torch.prim.GetAttr %arg0["_tensor_constant1780"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4898 = torch.prim.GetAttr %arg0["_param_constant202"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4899 = torch.prim.GetAttr %arg0["_tensor_constant1773"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4900 = torch.prim.GetAttr %arg0["_tensor_constant1772"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4901 = torch.prim.GetAttr %arg0["_tensor_constant1769"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4902 = torch.prim.GetAttr %arg0["_tensor_constant1768"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4903 = torch.prim.GetAttr %arg0["_tensor_constant1767"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4904 = torch.prim.GetAttr %arg0["_tensor_constant1766"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4905 = torch.prim.GetAttr %arg0["_tensor_constant1765"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4906 = torch.prim.GetAttr %arg0["_param_constant201"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4907 = torch.prim.GetAttr %arg0["_tensor_constant1758"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4908 = torch.prim.GetAttr %arg0["_tensor_constant1757"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4909 = torch.prim.GetAttr %arg0["_tensor_constant1754"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4910 = torch.prim.GetAttr %arg0["_tensor_constant1753"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4911 = torch.prim.GetAttr %arg0["_tensor_constant1752"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4912 = torch.prim.GetAttr %arg0["_tensor_constant1751"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4913 = torch.prim.GetAttr %arg0["_tensor_constant1750"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4914 = torch.prim.GetAttr %arg0["_param_constant200"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4915 = torch.prim.GetAttr %arg0["_tensor_constant1743"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4916 = torch.prim.GetAttr %arg0["_tensor_constant1742"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4917 = torch.prim.GetAttr %arg0["_tensor_constant1739"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4918 = torch.prim.GetAttr %arg0["_tensor_constant1738"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4919 = torch.prim.GetAttr %arg0["_tensor_constant1737"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4920 = torch.prim.GetAttr %arg0["_tensor_constant1736"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4921 = torch.prim.GetAttr %arg0["_tensor_constant1735"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4922 = torch.prim.GetAttr %arg0["_param_constant199"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4923 = torch.prim.GetAttr %arg0["_tensor_constant1728"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4924 = torch.prim.GetAttr %arg0["_tensor_constant1727"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4925 = torch.prim.GetAttr %arg0["_tensor_constant1724"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4926 = torch.prim.GetAttr %arg0["_tensor_constant1723"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4927 = torch.prim.GetAttr %arg0["_tensor_constant1722"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4928 = torch.prim.GetAttr %arg0["_tensor_constant1721"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4929 = torch.prim.GetAttr %arg0["_tensor_constant1720"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4930 = torch.prim.GetAttr %arg0["_param_constant198"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4931 = torch.prim.GetAttr %arg0["_tensor_constant1713"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4932 = torch.prim.GetAttr %arg0["_tensor_constant1712"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4933 = torch.prim.GetAttr %arg0["_tensor_constant1709"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4934 = torch.prim.GetAttr %arg0["_tensor_constant1708"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4935 = torch.prim.GetAttr %arg0["_tensor_constant1707"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4936 = torch.prim.GetAttr %arg0["_tensor_constant1706"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4937 = torch.prim.GetAttr %arg0["_tensor_constant1705"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4938 = torch.prim.GetAttr %arg0["_param_constant197"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4939 = torch.prim.GetAttr %arg0["_tensor_constant1698"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4940 = torch.prim.GetAttr %arg0["_tensor_constant1697"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4941 = torch.prim.GetAttr %arg0["_tensor_constant1694"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4942 = torch.prim.GetAttr %arg0["_tensor_constant1693"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4943 = torch.prim.GetAttr %arg0["_tensor_constant1692"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4944 = torch.prim.GetAttr %arg0["_tensor_constant1691"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4945 = torch.prim.GetAttr %arg0["_tensor_constant1690"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4946 = torch.prim.GetAttr %arg0["_param_constant196"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4947 = torch.prim.GetAttr %arg0["_tensor_constant1683"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4948 = torch.prim.GetAttr %arg0["_tensor_constant1682"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4949 = torch.prim.GetAttr %arg0["_tensor_constant1679"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4950 = torch.prim.GetAttr %arg0["_tensor_constant1678"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4951 = torch.prim.GetAttr %arg0["_tensor_constant1677"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4952 = torch.prim.GetAttr %arg0["_tensor_constant1676"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4953 = torch.prim.GetAttr %arg0["_tensor_constant1675"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4954 = torch.prim.GetAttr %arg0["_param_constant195"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4955 = torch.prim.GetAttr %arg0["_param_constant194"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4956 = torch.prim.GetAttr %arg0["_param_constant193"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4957 = torch.prim.GetAttr %arg0["_param_constant192"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4958 = torch.prim.GetAttr %arg0["_param_constant191"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4959 = torch.prim.GetAttr %arg0["_tensor_constant1668"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4960 = torch.prim.GetAttr %arg0["_tensor_constant1667"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4961 = torch.prim.GetAttr %arg0["_tensor_constant1664"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4962 = torch.prim.GetAttr %arg0["_tensor_constant1663"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4963 = torch.prim.GetAttr %arg0["_tensor_constant1662"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4964 = torch.prim.GetAttr %arg0["_tensor_constant1661"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4965 = torch.prim.GetAttr %arg0["_tensor_constant1660"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4966 = torch.prim.GetAttr %arg0["_param_constant190"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4967 = torch.prim.GetAttr %arg0["_param_constant189"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4968 = torch.prim.GetAttr %arg0["_param_constant188"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4969 = torch.prim.GetAttr %arg0["_tensor_constant1653"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4970 = torch.prim.GetAttr %arg0["_tensor_constant1652"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4971 = torch.prim.GetAttr %arg0["_tensor_constant1649"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4972 = torch.prim.GetAttr %arg0["_tensor_constant1648"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4973 = torch.prim.GetAttr %arg0["_tensor_constant1647"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4974 = torch.prim.GetAttr %arg0["_tensor_constant1646"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4975 = torch.prim.GetAttr %arg0["_tensor_constant1645"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4976 = torch.prim.GetAttr %arg0["_param_constant187"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4977 = torch.prim.GetAttr %arg0["_tensor_constant1638"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4978 = torch.prim.GetAttr %arg0["_tensor_constant1637"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4979 = torch.prim.GetAttr %arg0["_tensor_constant1634"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4980 = torch.prim.GetAttr %arg0["_tensor_constant1633"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4981 = torch.prim.GetAttr %arg0["_tensor_constant1632"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4982 = torch.prim.GetAttr %arg0["_tensor_constant1631"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4983 = torch.prim.GetAttr %arg0["_tensor_constant1630"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4984 = torch.prim.GetAttr %arg0["_param_constant186"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4985 = torch.prim.GetAttr %arg0["_tensor_constant1623"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4986 = torch.prim.GetAttr %arg0["_tensor_constant1622"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4987 = torch.prim.GetAttr %arg0["_tensor_constant1619"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4988 = torch.prim.GetAttr %arg0["_tensor_constant1618"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4989 = torch.prim.GetAttr %arg0["_tensor_constant1617"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4990 = torch.prim.GetAttr %arg0["_tensor_constant1616"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4991 = torch.prim.GetAttr %arg0["_tensor_constant1615"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4992 = torch.prim.GetAttr %arg0["_param_constant185"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4993 = torch.prim.GetAttr %arg0["_param_constant184"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4994 = torch.prim.GetAttr %arg0["_param_constant183"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4995 = torch.prim.GetAttr %arg0["_param_constant182"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4996 = torch.prim.GetAttr %arg0["_param_constant181"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4997 = torch.prim.GetAttr %arg0["_tensor_constant1608"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4998 = torch.prim.GetAttr %arg0["_tensor_constant1607"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %4999 = torch.prim.GetAttr %arg0["_tensor_constant1604"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5000 = torch.prim.GetAttr %arg0["_tensor_constant1603"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5001 = torch.prim.GetAttr %arg0["_tensor_constant1602"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5002 = torch.prim.GetAttr %arg0["_tensor_constant1601"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5003 = torch.prim.GetAttr %arg0["_tensor_constant1600"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5004 = torch.prim.GetAttr %arg0["_param_constant180"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5005 = torch.prim.GetAttr %arg0["_param_constant179"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5006 = torch.prim.GetAttr %arg0["_param_constant178"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5007 = torch.prim.GetAttr %arg0["_tensor_constant1593"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5008 = torch.prim.GetAttr %arg0["_tensor_constant1592"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5009 = torch.prim.GetAttr %arg0["_tensor_constant1589"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5010 = torch.prim.GetAttr %arg0["_tensor_constant1588"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5011 = torch.prim.GetAttr %arg0["_tensor_constant1587"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5012 = torch.prim.GetAttr %arg0["_tensor_constant1586"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5013 = torch.prim.GetAttr %arg0["_tensor_constant1585"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5014 = torch.prim.GetAttr %arg0["_param_constant177"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5015 = torch.prim.GetAttr %arg0["_tensor_constant1578"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5016 = torch.prim.GetAttr %arg0["_tensor_constant1577"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5017 = torch.prim.GetAttr %arg0["_tensor_constant1574"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5018 = torch.prim.GetAttr %arg0["_tensor_constant1573"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5019 = torch.prim.GetAttr %arg0["_tensor_constant1572"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5020 = torch.prim.GetAttr %arg0["_tensor_constant1571"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5021 = torch.prim.GetAttr %arg0["_tensor_constant1570"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5022 = torch.prim.GetAttr %arg0["_param_constant176"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5023 = torch.prim.GetAttr %arg0["_param_constant175"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5024 = torch.prim.GetAttr %arg0["_param_constant174"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5025 = torch.prim.GetAttr %arg0["_param_constant173"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5026 = torch.prim.GetAttr %arg0["_param_constant172"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5027 = torch.prim.GetAttr %arg0["_tensor_constant1563"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5028 = torch.prim.GetAttr %arg0["_tensor_constant1562"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5029 = torch.prim.GetAttr %arg0["_tensor_constant1559"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5030 = torch.prim.GetAttr %arg0["_tensor_constant1558"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5031 = torch.prim.GetAttr %arg0["_tensor_constant1557"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5032 = torch.prim.GetAttr %arg0["_tensor_constant1556"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5033 = torch.prim.GetAttr %arg0["_tensor_constant1555"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5034 = torch.prim.GetAttr %arg0["_param_constant171"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5035 = torch.prim.GetAttr %arg0["_param_constant170"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5036 = torch.prim.GetAttr %arg0["_param_constant169"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5037 = torch.prim.GetAttr %arg0["_tensor_constant1548"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5038 = torch.prim.GetAttr %arg0["_tensor_constant1547"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5039 = torch.prim.GetAttr %arg0["_tensor_constant1544"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5040 = torch.prim.GetAttr %arg0["_tensor_constant1543"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5041 = torch.prim.GetAttr %arg0["_tensor_constant1542"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5042 = torch.prim.GetAttr %arg0["_tensor_constant1541"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5043 = torch.prim.GetAttr %arg0["_tensor_constant1540"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5044 = torch.prim.GetAttr %arg0["_param_constant168"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5045 = torch.prim.GetAttr %arg0["_tensor_constant1533"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5046 = torch.prim.GetAttr %arg0["_tensor_constant1532"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5047 = torch.prim.GetAttr %arg0["_tensor_constant1529"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5048 = torch.prim.GetAttr %arg0["_tensor_constant1528"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5049 = torch.prim.GetAttr %arg0["_tensor_constant1527"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5050 = torch.prim.GetAttr %arg0["_tensor_constant1526"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5051 = torch.prim.GetAttr %arg0["_tensor_constant1525"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5052 = torch.prim.GetAttr %arg0["_param_constant167"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5053 = torch.prim.GetAttr %arg0["_param_constant166"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5054 = torch.prim.GetAttr %arg0["_param_constant165"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5055 = torch.prim.GetAttr %arg0["_param_constant164"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5056 = torch.prim.GetAttr %arg0["_param_constant163"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5057 = torch.prim.GetAttr %arg0["_tensor_constant1518"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5058 = torch.prim.GetAttr %arg0["_tensor_constant1517"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5059 = torch.prim.GetAttr %arg0["_tensor_constant1514"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5060 = torch.prim.GetAttr %arg0["_tensor_constant1513"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5061 = torch.prim.GetAttr %arg0["_tensor_constant1512"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5062 = torch.prim.GetAttr %arg0["_tensor_constant1511"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5063 = torch.prim.GetAttr %arg0["_tensor_constant1510"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5064 = torch.prim.GetAttr %arg0["_param_constant162"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5065 = torch.prim.GetAttr %arg0["_param_constant161"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5066 = torch.prim.GetAttr %arg0["_param_constant160"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5067 = torch.prim.GetAttr %arg0["_tensor_constant1503"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5068 = torch.prim.GetAttr %arg0["_tensor_constant1502"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5069 = torch.prim.GetAttr %arg0["_tensor_constant1499"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5070 = torch.prim.GetAttr %arg0["_tensor_constant1498"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5071 = torch.prim.GetAttr %arg0["_tensor_constant1497"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5072 = torch.prim.GetAttr %arg0["_tensor_constant1496"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5073 = torch.prim.GetAttr %arg0["_tensor_constant1495"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5074 = torch.prim.GetAttr %arg0["_param_constant159"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5075 = torch.prim.GetAttr %arg0["_param_constant158"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5076 = torch.prim.GetAttr %arg0["_param_constant157"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5077 = torch.prim.GetAttr %arg0["_param_constant156"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5078 = torch.prim.GetAttr %arg0["_param_constant155"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5079 = torch.prim.GetAttr %arg0["_tensor_constant1488"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5080 = torch.prim.GetAttr %arg0["_tensor_constant1487"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5081 = torch.prim.GetAttr %arg0["_tensor_constant1484"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5082 = torch.prim.GetAttr %arg0["_tensor_constant1483"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5083 = torch.prim.GetAttr %arg0["_tensor_constant1482"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5084 = torch.prim.GetAttr %arg0["_tensor_constant1481"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5085 = torch.prim.GetAttr %arg0["_tensor_constant1480"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5086 = torch.prim.GetAttr %arg0["_param_constant154"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5087 = torch.prim.GetAttr %arg0["_param_constant153"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5088 = torch.prim.GetAttr %arg0["_param_constant152"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5089 = torch.prim.GetAttr %arg0["_tensor_constant1473"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5090 = torch.prim.GetAttr %arg0["_tensor_constant1472"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5091 = torch.prim.GetAttr %arg0["_tensor_constant1469"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5092 = torch.prim.GetAttr %arg0["_tensor_constant1468"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5093 = torch.prim.GetAttr %arg0["_tensor_constant1467"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5094 = torch.prim.GetAttr %arg0["_tensor_constant1466"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5095 = torch.prim.GetAttr %arg0["_tensor_constant1465"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5096 = torch.prim.GetAttr %arg0["_param_constant151"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5097 = torch.prim.GetAttr %arg0["_tensor_constant1459"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5098 = torch.prim.GetAttr %arg0["_tensor_constant1458"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5099 = torch.prim.GetAttr %arg0["_param_constant150"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5100 = torch.prim.GetAttr %arg0["_tensor_constant1452"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5101 = torch.prim.GetAttr %arg0["_tensor_constant1451"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5102 = torch.prim.GetAttr %arg0["_param_constant149"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5103 = torch.prim.GetAttr %arg0["_tensor_constant1444"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5104 = torch.prim.GetAttr %arg0["_tensor_constant1443"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5105 = torch.prim.GetAttr %arg0["_tensor_constant1440"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5106 = torch.prim.GetAttr %arg0["_tensor_constant1439"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5107 = torch.prim.GetAttr %arg0["_tensor_constant1438"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5108 = torch.prim.GetAttr %arg0["_tensor_constant1437"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5109 = torch.prim.GetAttr %arg0["_tensor_constant1436"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5110 = torch.prim.GetAttr %arg0["_tensor_constant1430"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5111 = torch.prim.GetAttr %arg0["_tensor_constant1429"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5112 = torch.prim.GetAttr %arg0["_tensor_constant1426"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5113 = torch.prim.GetAttr %arg0["_tensor_constant1425"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5114 = torch.prim.GetAttr %arg0["_tensor_constant1424"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5115 = torch.prim.GetAttr %arg0["_tensor_constant1423"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5116 = torch.prim.GetAttr %arg0["_tensor_constant1422"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5117 = torch.prim.GetAttr %arg0["_tensor_constant1416"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5118 = torch.prim.GetAttr %arg0["_tensor_constant1415"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5119 = torch.prim.GetAttr %arg0["_tensor_constant1412"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5120 = torch.prim.GetAttr %arg0["_tensor_constant1411"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5121 = torch.prim.GetAttr %arg0["_tensor_constant1410"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5122 = torch.prim.GetAttr %arg0["_tensor_constant1409"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5123 = torch.prim.GetAttr %arg0["_tensor_constant1408"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5124 = torch.prim.GetAttr %arg0["_param_constant148"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5125 = torch.prim.GetAttr %arg0["_tensor_constant1401"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5126 = torch.prim.GetAttr %arg0["_tensor_constant1400"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5127 = torch.prim.GetAttr %arg0["_tensor_constant1397"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5128 = torch.prim.GetAttr %arg0["_tensor_constant1396"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5129 = torch.prim.GetAttr %arg0["_tensor_constant1395"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5130 = torch.prim.GetAttr %arg0["_tensor_constant1394"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5131 = torch.prim.GetAttr %arg0["_tensor_constant1393"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5132 = torch.prim.GetAttr %arg0["_param_constant147"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5133 = torch.prim.GetAttr %arg0["_tensor_constant1386"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5134 = torch.prim.GetAttr %arg0["_tensor_constant1385"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5135 = torch.prim.GetAttr %arg0["_tensor_constant1382"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5136 = torch.prim.GetAttr %arg0["_tensor_constant1381"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5137 = torch.prim.GetAttr %arg0["_tensor_constant1380"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5138 = torch.prim.GetAttr %arg0["_tensor_constant1379"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5139 = torch.prim.GetAttr %arg0["_tensor_constant1378"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5140 = torch.prim.GetAttr %arg0["_param_constant146"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5141 = torch.prim.GetAttr %arg0["_tensor_constant1371"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5142 = torch.prim.GetAttr %arg0["_tensor_constant1370"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5143 = torch.prim.GetAttr %arg0["_tensor_constant1367"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5144 = torch.prim.GetAttr %arg0["_tensor_constant1366"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5145 = torch.prim.GetAttr %arg0["_tensor_constant1365"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5146 = torch.prim.GetAttr %arg0["_tensor_constant1364"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5147 = torch.prim.GetAttr %arg0["_tensor_constant1363"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5148 = torch.prim.GetAttr %arg0["_param_constant145"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5149 = torch.prim.GetAttr %arg0["_tensor_constant1356"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5150 = torch.prim.GetAttr %arg0["_tensor_constant1355"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5151 = torch.prim.GetAttr %arg0["_tensor_constant1352"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5152 = torch.prim.GetAttr %arg0["_tensor_constant1351"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5153 = torch.prim.GetAttr %arg0["_tensor_constant1350"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5154 = torch.prim.GetAttr %arg0["_tensor_constant1349"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5155 = torch.prim.GetAttr %arg0["_tensor_constant1348"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5156 = torch.prim.GetAttr %arg0["_param_constant144"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5157 = torch.prim.GetAttr %arg0["_tensor_constant1341"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5158 = torch.prim.GetAttr %arg0["_tensor_constant1340"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5159 = torch.prim.GetAttr %arg0["_tensor_constant1337"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5160 = torch.prim.GetAttr %arg0["_tensor_constant1336"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5161 = torch.prim.GetAttr %arg0["_tensor_constant1335"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5162 = torch.prim.GetAttr %arg0["_tensor_constant1334"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5163 = torch.prim.GetAttr %arg0["_tensor_constant1333"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5164 = torch.prim.GetAttr %arg0["_param_constant143"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5165 = torch.prim.GetAttr %arg0["_tensor_constant1326"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5166 = torch.prim.GetAttr %arg0["_tensor_constant1325"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5167 = torch.prim.GetAttr %arg0["_tensor_constant1322"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5168 = torch.prim.GetAttr %arg0["_tensor_constant1321"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5169 = torch.prim.GetAttr %arg0["_tensor_constant1320"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5170 = torch.prim.GetAttr %arg0["_tensor_constant1319"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5171 = torch.prim.GetAttr %arg0["_tensor_constant1318"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5172 = torch.prim.GetAttr %arg0["_param_constant142"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5173 = torch.prim.GetAttr %arg0["_tensor_constant1311"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5174 = torch.prim.GetAttr %arg0["_tensor_constant1310"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5175 = torch.prim.GetAttr %arg0["_tensor_constant1307"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5176 = torch.prim.GetAttr %arg0["_tensor_constant1306"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5177 = torch.prim.GetAttr %arg0["_tensor_constant1305"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5178 = torch.prim.GetAttr %arg0["_tensor_constant1304"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5179 = torch.prim.GetAttr %arg0["_tensor_constant1303"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5180 = torch.prim.GetAttr %arg0["_param_constant141"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5181 = torch.prim.GetAttr %arg0["_param_constant140"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5182 = torch.prim.GetAttr %arg0["_param_constant139"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5183 = torch.prim.GetAttr %arg0["_param_constant138"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5184 = torch.prim.GetAttr %arg0["_param_constant137"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5185 = torch.prim.GetAttr %arg0["_tensor_constant1296"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5186 = torch.prim.GetAttr %arg0["_tensor_constant1295"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5187 = torch.prim.GetAttr %arg0["_tensor_constant1292"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5188 = torch.prim.GetAttr %arg0["_tensor_constant1291"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5189 = torch.prim.GetAttr %arg0["_tensor_constant1290"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5190 = torch.prim.GetAttr %arg0["_tensor_constant1289"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5191 = torch.prim.GetAttr %arg0["_tensor_constant1288"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5192 = torch.prim.GetAttr %arg0["_param_constant136"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5193 = torch.prim.GetAttr %arg0["_param_constant135"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5194 = torch.prim.GetAttr %arg0["_param_constant134"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5195 = torch.prim.GetAttr %arg0["_tensor_constant1281"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5196 = torch.prim.GetAttr %arg0["_tensor_constant1280"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5197 = torch.prim.GetAttr %arg0["_tensor_constant1277"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5198 = torch.prim.GetAttr %arg0["_tensor_constant1276"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5199 = torch.prim.GetAttr %arg0["_tensor_constant1275"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5200 = torch.prim.GetAttr %arg0["_tensor_constant1274"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5201 = torch.prim.GetAttr %arg0["_tensor_constant1273"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5202 = torch.prim.GetAttr %arg0["_param_constant133"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5203 = torch.prim.GetAttr %arg0["_param_constant132"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5204 = torch.prim.GetAttr %arg0["_param_constant131"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5205 = torch.prim.GetAttr %arg0["_param_constant130"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5206 = torch.prim.GetAttr %arg0["_param_constant129"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5207 = torch.prim.GetAttr %arg0["_tensor_constant1266"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5208 = torch.prim.GetAttr %arg0["_tensor_constant1265"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5209 = torch.prim.GetAttr %arg0["_tensor_constant1262"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5210 = torch.prim.GetAttr %arg0["_tensor_constant1261"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5211 = torch.prim.GetAttr %arg0["_tensor_constant1260"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5212 = torch.prim.GetAttr %arg0["_tensor_constant1259"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5213 = torch.prim.GetAttr %arg0["_tensor_constant1258"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5214 = torch.prim.GetAttr %arg0["_param_constant128"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5215 = torch.prim.GetAttr %arg0["_param_constant127"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5216 = torch.prim.GetAttr %arg0["_param_constant126"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5217 = torch.prim.GetAttr %arg0["_tensor_constant1251"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5218 = torch.prim.GetAttr %arg0["_tensor_constant1250"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5219 = torch.prim.GetAttr %arg0["_tensor_constant1247"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5220 = torch.prim.GetAttr %arg0["_tensor_constant1246"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5221 = torch.prim.GetAttr %arg0["_tensor_constant1245"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5222 = torch.prim.GetAttr %arg0["_tensor_constant1244"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5223 = torch.prim.GetAttr %arg0["_tensor_constant1243"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5224 = torch.prim.GetAttr %arg0["_param_constant125"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5225 = torch.prim.GetAttr %arg0["_param_constant124"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5226 = torch.prim.GetAttr %arg0["_param_constant123"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5227 = torch.prim.GetAttr %arg0["_param_constant122"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5228 = torch.prim.GetAttr %arg0["_param_constant121"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5229 = torch.prim.GetAttr %arg0["_tensor_constant1236"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5230 = torch.prim.GetAttr %arg0["_tensor_constant1235"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5231 = torch.prim.GetAttr %arg0["_tensor_constant1232"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5232 = torch.prim.GetAttr %arg0["_tensor_constant1231"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5233 = torch.prim.GetAttr %arg0["_tensor_constant1230"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5234 = torch.prim.GetAttr %arg0["_tensor_constant1229"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5235 = torch.prim.GetAttr %arg0["_tensor_constant1228"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5236 = torch.prim.GetAttr %arg0["_param_constant120"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5237 = torch.prim.GetAttr %arg0["_param_constant119"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5238 = torch.prim.GetAttr %arg0["_param_constant118"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5239 = torch.prim.GetAttr %arg0["_tensor_constant1221"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5240 = torch.prim.GetAttr %arg0["_tensor_constant1220"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5241 = torch.prim.GetAttr %arg0["_tensor_constant1217"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5242 = torch.prim.GetAttr %arg0["_tensor_constant1216"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5243 = torch.prim.GetAttr %arg0["_tensor_constant1215"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5244 = torch.prim.GetAttr %arg0["_tensor_constant1214"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5245 = torch.prim.GetAttr %arg0["_tensor_constant1213"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5246 = torch.prim.GetAttr %arg0["_param_constant117"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5247 = torch.prim.GetAttr %arg0["_tensor_constant1206"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5248 = torch.prim.GetAttr %arg0["_tensor_constant1205"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5249 = torch.prim.GetAttr %arg0["_tensor_constant1202"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5250 = torch.prim.GetAttr %arg0["_tensor_constant1201"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5251 = torch.prim.GetAttr %arg0["_tensor_constant1200"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5252 = torch.prim.GetAttr %arg0["_tensor_constant1199"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5253 = torch.prim.GetAttr %arg0["_tensor_constant1198"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5254 = torch.prim.GetAttr %arg0["_param_constant116"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5255 = torch.prim.GetAttr %arg0["_tensor_constant1192"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5256 = torch.prim.GetAttr %arg0["_tensor_constant1191"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5257 = torch.prim.GetAttr %arg0["_param_constant115"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5258 = torch.prim.GetAttr %arg0["_tensor_constant1185"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5259 = torch.prim.GetAttr %arg0["_tensor_constant1184"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5260 = torch.prim.GetAttr %arg0["_param_constant114"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5261 = torch.prim.GetAttr %arg0["_tensor_constant1177"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5262 = torch.prim.GetAttr %arg0["_tensor_constant1176"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5263 = torch.prim.GetAttr %arg0["_tensor_constant1173"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5264 = torch.prim.GetAttr %arg0["_tensor_constant1172"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5265 = torch.prim.GetAttr %arg0["_tensor_constant1171"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5266 = torch.prim.GetAttr %arg0["_tensor_constant1170"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5267 = torch.prim.GetAttr %arg0["_tensor_constant1169"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5268 = torch.prim.GetAttr %arg0["_tensor_constant1163"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5269 = torch.prim.GetAttr %arg0["_tensor_constant1162"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5270 = torch.prim.GetAttr %arg0["_tensor_constant1159"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5271 = torch.prim.GetAttr %arg0["_tensor_constant1158"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5272 = torch.prim.GetAttr %arg0["_tensor_constant1157"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5273 = torch.prim.GetAttr %arg0["_tensor_constant1156"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5274 = torch.prim.GetAttr %arg0["_tensor_constant1155"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5275 = torch.prim.GetAttr %arg0["_tensor_constant1149"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5276 = torch.prim.GetAttr %arg0["_tensor_constant1148"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5277 = torch.prim.GetAttr %arg0["_tensor_constant1145"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5278 = torch.prim.GetAttr %arg0["_tensor_constant1144"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5279 = torch.prim.GetAttr %arg0["_tensor_constant1143"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5280 = torch.prim.GetAttr %arg0["_tensor_constant1142"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5281 = torch.prim.GetAttr %arg0["_tensor_constant1141"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5282 = torch.prim.GetAttr %arg0["_param_constant113"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5283 = torch.prim.GetAttr %arg0["_tensor_constant1134"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5284 = torch.prim.GetAttr %arg0["_tensor_constant1133"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5285 = torch.prim.GetAttr %arg0["_tensor_constant1130"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5286 = torch.prim.GetAttr %arg0["_tensor_constant1129"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5287 = torch.prim.GetAttr %arg0["_tensor_constant1128"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5288 = torch.prim.GetAttr %arg0["_tensor_constant1127"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5289 = torch.prim.GetAttr %arg0["_tensor_constant1126"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5290 = torch.prim.GetAttr %arg0["_param_constant112"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5291 = torch.prim.GetAttr %arg0["_tensor_constant1119"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5292 = torch.prim.GetAttr %arg0["_tensor_constant1118"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5293 = torch.prim.GetAttr %arg0["_tensor_constant1115"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5294 = torch.prim.GetAttr %arg0["_tensor_constant1114"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5295 = torch.prim.GetAttr %arg0["_tensor_constant1113"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5296 = torch.prim.GetAttr %arg0["_tensor_constant1112"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5297 = torch.prim.GetAttr %arg0["_tensor_constant1111"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5298 = torch.prim.GetAttr %arg0["_param_constant111"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5299 = torch.prim.GetAttr %arg0["_tensor_constant1104"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5300 = torch.prim.GetAttr %arg0["_tensor_constant1103"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5301 = torch.prim.GetAttr %arg0["_tensor_constant1100"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5302 = torch.prim.GetAttr %arg0["_tensor_constant1099"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5303 = torch.prim.GetAttr %arg0["_tensor_constant1098"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5304 = torch.prim.GetAttr %arg0["_tensor_constant1097"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5305 = torch.prim.GetAttr %arg0["_tensor_constant1096"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5306 = torch.prim.GetAttr %arg0["_param_constant110"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5307 = torch.prim.GetAttr %arg0["_tensor_constant1089"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5308 = torch.prim.GetAttr %arg0["_tensor_constant1088"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5309 = torch.prim.GetAttr %arg0["_tensor_constant1085"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5310 = torch.prim.GetAttr %arg0["_tensor_constant1084"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5311 = torch.prim.GetAttr %arg0["_tensor_constant1083"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5312 = torch.prim.GetAttr %arg0["_tensor_constant1082"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5313 = torch.prim.GetAttr %arg0["_tensor_constant1081"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5314 = torch.prim.GetAttr %arg0["_param_constant109"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5315 = torch.prim.GetAttr %arg0["_tensor_constant1074"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5316 = torch.prim.GetAttr %arg0["_tensor_constant1073"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5317 = torch.prim.GetAttr %arg0["_tensor_constant1070"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5318 = torch.prim.GetAttr %arg0["_tensor_constant1069"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5319 = torch.prim.GetAttr %arg0["_tensor_constant1068"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5320 = torch.prim.GetAttr %arg0["_tensor_constant1067"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5321 = torch.prim.GetAttr %arg0["_tensor_constant1066"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5322 = torch.prim.GetAttr %arg0["_param_constant108"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5323 = torch.prim.GetAttr %arg0["_tensor_constant1059"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5324 = torch.prim.GetAttr %arg0["_tensor_constant1058"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5325 = torch.prim.GetAttr %arg0["_tensor_constant1055"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5326 = torch.prim.GetAttr %arg0["_tensor_constant1054"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5327 = torch.prim.GetAttr %arg0["_tensor_constant1053"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5328 = torch.prim.GetAttr %arg0["_tensor_constant1052"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5329 = torch.prim.GetAttr %arg0["_tensor_constant1051"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5330 = torch.prim.GetAttr %arg0["_param_constant107"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5331 = torch.prim.GetAttr %arg0["_tensor_constant1044"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5332 = torch.prim.GetAttr %arg0["_tensor_constant1043"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5333 = torch.prim.GetAttr %arg0["_tensor_constant1040"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5334 = torch.prim.GetAttr %arg0["_tensor_constant1039"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5335 = torch.prim.GetAttr %arg0["_tensor_constant1038"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5336 = torch.prim.GetAttr %arg0["_tensor_constant1037"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5337 = torch.prim.GetAttr %arg0["_tensor_constant1036"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5338 = torch.prim.GetAttr %arg0["_param_constant106"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5339 = torch.prim.GetAttr %arg0["_param_constant105"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5340 = torch.prim.GetAttr %arg0["_param_constant104"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5341 = torch.prim.GetAttr %arg0["_param_constant103"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5342 = torch.prim.GetAttr %arg0["_param_constant102"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5343 = torch.prim.GetAttr %arg0["_tensor_constant1029"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5344 = torch.prim.GetAttr %arg0["_tensor_constant1028"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5345 = torch.prim.GetAttr %arg0["_tensor_constant1025"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5346 = torch.prim.GetAttr %arg0["_tensor_constant1024"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5347 = torch.prim.GetAttr %arg0["_tensor_constant1023"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5348 = torch.prim.GetAttr %arg0["_tensor_constant1022"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5349 = torch.prim.GetAttr %arg0["_tensor_constant1021"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5350 = torch.prim.GetAttr %arg0["_param_constant101"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5351 = torch.prim.GetAttr %arg0["_param_constant100"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5352 = torch.prim.GetAttr %arg0["_param_constant99"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5353 = torch.prim.GetAttr %arg0["_tensor_constant1014"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5354 = torch.prim.GetAttr %arg0["_tensor_constant1013"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5355 = torch.prim.GetAttr %arg0["_tensor_constant1010"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5356 = torch.prim.GetAttr %arg0["_tensor_constant1009"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5357 = torch.prim.GetAttr %arg0["_tensor_constant1008"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5358 = torch.prim.GetAttr %arg0["_tensor_constant1007"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5359 = torch.prim.GetAttr %arg0["_tensor_constant1006"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5360 = torch.prim.GetAttr %arg0["_param_constant98"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5361 = torch.prim.GetAttr %arg0["_tensor_constant1000"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5362 = torch.prim.GetAttr %arg0["_tensor_constant999"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5363 = torch.prim.GetAttr %arg0["_param_constant97"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5364 = torch.prim.GetAttr %arg0["_tensor_constant993"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5365 = torch.prim.GetAttr %arg0["_tensor_constant992"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5366 = torch.prim.GetAttr %arg0["_param_constant96"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5367 = torch.prim.GetAttr %arg0["_tensor_constant985"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5368 = torch.prim.GetAttr %arg0["_tensor_constant984"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5369 = torch.prim.GetAttr %arg0["_tensor_constant981"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5370 = torch.prim.GetAttr %arg0["_tensor_constant980"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5371 = torch.prim.GetAttr %arg0["_tensor_constant979"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5372 = torch.prim.GetAttr %arg0["_tensor_constant978"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5373 = torch.prim.GetAttr %arg0["_tensor_constant977"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5374 = torch.prim.GetAttr %arg0["_tensor_constant971"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5375 = torch.prim.GetAttr %arg0["_tensor_constant970"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5376 = torch.prim.GetAttr %arg0["_tensor_constant967"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5377 = torch.prim.GetAttr %arg0["_tensor_constant966"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5378 = torch.prim.GetAttr %arg0["_tensor_constant965"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5379 = torch.prim.GetAttr %arg0["_tensor_constant964"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5380 = torch.prim.GetAttr %arg0["_tensor_constant963"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5381 = torch.prim.GetAttr %arg0["_tensor_constant957"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5382 = torch.prim.GetAttr %arg0["_tensor_constant956"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5383 = torch.prim.GetAttr %arg0["_tensor_constant953"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5384 = torch.prim.GetAttr %arg0["_tensor_constant952"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5385 = torch.prim.GetAttr %arg0["_tensor_constant951"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5386 = torch.prim.GetAttr %arg0["_tensor_constant950"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5387 = torch.prim.GetAttr %arg0["_tensor_constant949"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5388 = torch.prim.GetAttr %arg0["_param_constant95"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5389 = torch.prim.GetAttr %arg0["_tensor_constant942"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5390 = torch.prim.GetAttr %arg0["_tensor_constant941"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5391 = torch.prim.GetAttr %arg0["_tensor_constant938"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5392 = torch.prim.GetAttr %arg0["_tensor_constant937"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5393 = torch.prim.GetAttr %arg0["_tensor_constant936"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5394 = torch.prim.GetAttr %arg0["_tensor_constant935"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5395 = torch.prim.GetAttr %arg0["_tensor_constant934"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5396 = torch.prim.GetAttr %arg0["_param_constant94"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5397 = torch.prim.GetAttr %arg0["_tensor_constant927"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5398 = torch.prim.GetAttr %arg0["_tensor_constant926"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5399 = torch.prim.GetAttr %arg0["_tensor_constant923"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5400 = torch.prim.GetAttr %arg0["_tensor_constant922"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5401 = torch.prim.GetAttr %arg0["_tensor_constant921"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5402 = torch.prim.GetAttr %arg0["_tensor_constant920"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5403 = torch.prim.GetAttr %arg0["_tensor_constant919"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5404 = torch.prim.GetAttr %arg0["_param_constant93"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5405 = torch.prim.GetAttr %arg0["_tensor_constant912"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5406 = torch.prim.GetAttr %arg0["_tensor_constant911"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5407 = torch.prim.GetAttr %arg0["_tensor_constant908"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5408 = torch.prim.GetAttr %arg0["_tensor_constant907"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5409 = torch.prim.GetAttr %arg0["_tensor_constant906"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5410 = torch.prim.GetAttr %arg0["_tensor_constant905"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5411 = torch.prim.GetAttr %arg0["_tensor_constant904"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5412 = torch.prim.GetAttr %arg0["_param_constant92"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5413 = torch.prim.GetAttr %arg0["_tensor_constant897"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5414 = torch.prim.GetAttr %arg0["_tensor_constant896"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5415 = torch.prim.GetAttr %arg0["_tensor_constant893"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5416 = torch.prim.GetAttr %arg0["_tensor_constant892"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5417 = torch.prim.GetAttr %arg0["_tensor_constant891"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5418 = torch.prim.GetAttr %arg0["_tensor_constant890"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5419 = torch.prim.GetAttr %arg0["_tensor_constant889"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5420 = torch.prim.GetAttr %arg0["_param_constant91"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5421 = torch.prim.GetAttr %arg0["_tensor_constant882"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5422 = torch.prim.GetAttr %arg0["_tensor_constant881"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5423 = torch.prim.GetAttr %arg0["_tensor_constant878"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5424 = torch.prim.GetAttr %arg0["_tensor_constant877"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5425 = torch.prim.GetAttr %arg0["_tensor_constant876"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5426 = torch.prim.GetAttr %arg0["_tensor_constant875"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5427 = torch.prim.GetAttr %arg0["_tensor_constant874"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5428 = torch.prim.GetAttr %arg0["_param_constant90"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5429 = torch.prim.GetAttr %arg0["_tensor_constant867"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5430 = torch.prim.GetAttr %arg0["_tensor_constant866"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5431 = torch.prim.GetAttr %arg0["_tensor_constant863"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5432 = torch.prim.GetAttr %arg0["_tensor_constant862"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5433 = torch.prim.GetAttr %arg0["_tensor_constant861"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5434 = torch.prim.GetAttr %arg0["_tensor_constant860"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5435 = torch.prim.GetAttr %arg0["_tensor_constant859"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5436 = torch.prim.GetAttr %arg0["_param_constant89"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5437 = torch.prim.GetAttr %arg0["_tensor_constant852"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5438 = torch.prim.GetAttr %arg0["_tensor_constant851"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5439 = torch.prim.GetAttr %arg0["_tensor_constant848"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5440 = torch.prim.GetAttr %arg0["_tensor_constant847"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5441 = torch.prim.GetAttr %arg0["_tensor_constant846"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5442 = torch.prim.GetAttr %arg0["_tensor_constant845"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5443 = torch.prim.GetAttr %arg0["_tensor_constant844"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5444 = torch.prim.GetAttr %arg0["_param_constant88"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5445 = torch.prim.GetAttr %arg0["_tensor_constant837"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5446 = torch.prim.GetAttr %arg0["_tensor_constant836"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5447 = torch.prim.GetAttr %arg0["_tensor_constant833"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5448 = torch.prim.GetAttr %arg0["_tensor_constant832"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5449 = torch.prim.GetAttr %arg0["_tensor_constant831"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5450 = torch.prim.GetAttr %arg0["_tensor_constant830"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5451 = torch.prim.GetAttr %arg0["_tensor_constant829"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5452 = torch.prim.GetAttr %arg0["_param_constant87"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5453 = torch.prim.GetAttr %arg0["_param_constant86"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5454 = torch.prim.GetAttr %arg0["_param_constant85"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5455 = torch.prim.GetAttr %arg0["_param_constant84"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5456 = torch.prim.GetAttr %arg0["_param_constant83"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5457 = torch.prim.GetAttr %arg0["_tensor_constant822"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5458 = torch.prim.GetAttr %arg0["_tensor_constant821"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5459 = torch.prim.GetAttr %arg0["_tensor_constant818"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5460 = torch.prim.GetAttr %arg0["_tensor_constant817"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5461 = torch.prim.GetAttr %arg0["_tensor_constant816"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5462 = torch.prim.GetAttr %arg0["_tensor_constant815"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5463 = torch.prim.GetAttr %arg0["_tensor_constant814"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5464 = torch.prim.GetAttr %arg0["_param_constant82"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5465 = torch.prim.GetAttr %arg0["_param_constant81"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5466 = torch.prim.GetAttr %arg0["_param_constant80"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5467 = torch.prim.GetAttr %arg0["_tensor_constant807"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5468 = torch.prim.GetAttr %arg0["_tensor_constant806"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5469 = torch.prim.GetAttr %arg0["_tensor_constant803"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5470 = torch.prim.GetAttr %arg0["_tensor_constant802"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5471 = torch.prim.GetAttr %arg0["_tensor_constant801"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5472 = torch.prim.GetAttr %arg0["_tensor_constant800"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5473 = torch.prim.GetAttr %arg0["_tensor_constant799"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5474 = torch.prim.GetAttr %arg0["_param_constant79"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5475 = torch.prim.GetAttr %arg0["_tensor_constant792"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5476 = torch.prim.GetAttr %arg0["_tensor_constant791"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5477 = torch.prim.GetAttr %arg0["_tensor_constant788"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5478 = torch.prim.GetAttr %arg0["_tensor_constant787"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5479 = torch.prim.GetAttr %arg0["_tensor_constant786"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5480 = torch.prim.GetAttr %arg0["_tensor_constant785"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5481 = torch.prim.GetAttr %arg0["_tensor_constant784"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5482 = torch.prim.GetAttr %arg0["_param_constant78"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5483 = torch.prim.GetAttr %arg0["_tensor_constant778"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5484 = torch.prim.GetAttr %arg0["_tensor_constant777"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5485 = torch.prim.GetAttr %arg0["_param_constant77"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5486 = torch.prim.GetAttr %arg0["_tensor_constant771"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5487 = torch.prim.GetAttr %arg0["_tensor_constant770"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5488 = torch.prim.GetAttr %arg0["_param_constant76"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5489 = torch.prim.GetAttr %arg0["_tensor_constant763"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5490 = torch.prim.GetAttr %arg0["_tensor_constant762"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5491 = torch.prim.GetAttr %arg0["_tensor_constant759"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5492 = torch.prim.GetAttr %arg0["_tensor_constant758"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5493 = torch.prim.GetAttr %arg0["_tensor_constant757"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5494 = torch.prim.GetAttr %arg0["_tensor_constant756"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5495 = torch.prim.GetAttr %arg0["_tensor_constant755"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5496 = torch.prim.GetAttr %arg0["_tensor_constant749"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5497 = torch.prim.GetAttr %arg0["_tensor_constant748"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5498 = torch.prim.GetAttr %arg0["_tensor_constant745"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5499 = torch.prim.GetAttr %arg0["_tensor_constant744"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5500 = torch.prim.GetAttr %arg0["_tensor_constant743"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5501 = torch.prim.GetAttr %arg0["_tensor_constant742"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5502 = torch.prim.GetAttr %arg0["_tensor_constant741"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5503 = torch.prim.GetAttr %arg0["_tensor_constant735"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5504 = torch.prim.GetAttr %arg0["_tensor_constant734"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5505 = torch.prim.GetAttr %arg0["_tensor_constant731"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5506 = torch.prim.GetAttr %arg0["_tensor_constant730"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5507 = torch.prim.GetAttr %arg0["_tensor_constant729"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5508 = torch.prim.GetAttr %arg0["_tensor_constant728"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5509 = torch.prim.GetAttr %arg0["_tensor_constant727"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5510 = torch.prim.GetAttr %arg0["_param_constant75"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5511 = torch.prim.GetAttr %arg0["_tensor_constant720"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5512 = torch.prim.GetAttr %arg0["_tensor_constant719"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5513 = torch.prim.GetAttr %arg0["_tensor_constant716"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5514 = torch.prim.GetAttr %arg0["_tensor_constant715"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5515 = torch.prim.GetAttr %arg0["_tensor_constant714"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5516 = torch.prim.GetAttr %arg0["_tensor_constant713"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5517 = torch.prim.GetAttr %arg0["_tensor_constant712"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5518 = torch.prim.GetAttr %arg0["_param_constant74"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5519 = torch.prim.GetAttr %arg0["_tensor_constant705"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5520 = torch.prim.GetAttr %arg0["_tensor_constant704"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5521 = torch.prim.GetAttr %arg0["_tensor_constant701"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5522 = torch.prim.GetAttr %arg0["_tensor_constant700"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5523 = torch.prim.GetAttr %arg0["_tensor_constant699"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5524 = torch.prim.GetAttr %arg0["_tensor_constant698"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5525 = torch.prim.GetAttr %arg0["_tensor_constant697"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5526 = torch.prim.GetAttr %arg0["_param_constant73"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5527 = torch.prim.GetAttr %arg0["_tensor_constant690"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5528 = torch.prim.GetAttr %arg0["_tensor_constant689"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5529 = torch.prim.GetAttr %arg0["_tensor_constant686"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5530 = torch.prim.GetAttr %arg0["_tensor_constant685"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5531 = torch.prim.GetAttr %arg0["_tensor_constant684"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5532 = torch.prim.GetAttr %arg0["_tensor_constant683"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5533 = torch.prim.GetAttr %arg0["_tensor_constant682"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5534 = torch.prim.GetAttr %arg0["_param_constant72"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5535 = torch.prim.GetAttr %arg0["_tensor_constant675"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5536 = torch.prim.GetAttr %arg0["_tensor_constant674"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5537 = torch.prim.GetAttr %arg0["_tensor_constant671"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5538 = torch.prim.GetAttr %arg0["_tensor_constant670"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5539 = torch.prim.GetAttr %arg0["_tensor_constant669"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5540 = torch.prim.GetAttr %arg0["_tensor_constant668"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5541 = torch.prim.GetAttr %arg0["_tensor_constant667"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5542 = torch.prim.GetAttr %arg0["_param_constant71"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5543 = torch.prim.GetAttr %arg0["_tensor_constant660"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5544 = torch.prim.GetAttr %arg0["_tensor_constant659"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5545 = torch.prim.GetAttr %arg0["_tensor_constant656"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5546 = torch.prim.GetAttr %arg0["_tensor_constant655"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5547 = torch.prim.GetAttr %arg0["_tensor_constant654"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5548 = torch.prim.GetAttr %arg0["_tensor_constant653"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5549 = torch.prim.GetAttr %arg0["_tensor_constant652"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5550 = torch.prim.GetAttr %arg0["_param_constant70"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5551 = torch.prim.GetAttr %arg0["_tensor_constant645"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5552 = torch.prim.GetAttr %arg0["_tensor_constant644"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5553 = torch.prim.GetAttr %arg0["_tensor_constant641"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5554 = torch.prim.GetAttr %arg0["_tensor_constant640"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5555 = torch.prim.GetAttr %arg0["_tensor_constant639"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5556 = torch.prim.GetAttr %arg0["_tensor_constant638"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5557 = torch.prim.GetAttr %arg0["_tensor_constant637"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5558 = torch.prim.GetAttr %arg0["_param_constant69"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5559 = torch.prim.GetAttr %arg0["_tensor_constant630"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5560 = torch.prim.GetAttr %arg0["_tensor_constant629"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5561 = torch.prim.GetAttr %arg0["_tensor_constant626"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5562 = torch.prim.GetAttr %arg0["_tensor_constant625"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5563 = torch.prim.GetAttr %arg0["_tensor_constant624"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5564 = torch.prim.GetAttr %arg0["_tensor_constant623"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5565 = torch.prim.GetAttr %arg0["_tensor_constant622"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5566 = torch.prim.GetAttr %arg0["_param_constant68"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5567 = torch.prim.GetAttr %arg0["_param_constant67"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5568 = torch.prim.GetAttr %arg0["_param_constant66"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5569 = torch.prim.GetAttr %arg0["_param_constant65"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5570 = torch.prim.GetAttr %arg0["_param_constant64"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5571 = torch.prim.GetAttr %arg0["_tensor_constant615"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5572 = torch.prim.GetAttr %arg0["_tensor_constant614"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5573 = torch.prim.GetAttr %arg0["_tensor_constant611"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5574 = torch.prim.GetAttr %arg0["_tensor_constant610"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5575 = torch.prim.GetAttr %arg0["_tensor_constant609"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5576 = torch.prim.GetAttr %arg0["_tensor_constant608"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5577 = torch.prim.GetAttr %arg0["_tensor_constant607"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5578 = torch.prim.GetAttr %arg0["_param_constant63"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5579 = torch.prim.GetAttr %arg0["_param_constant62"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5580 = torch.prim.GetAttr %arg0["_param_constant61"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5581 = torch.prim.GetAttr %arg0["_tensor_constant600"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5582 = torch.prim.GetAttr %arg0["_tensor_constant599"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5583 = torch.prim.GetAttr %arg0["_tensor_constant596"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5584 = torch.prim.GetAttr %arg0["_tensor_constant595"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5585 = torch.prim.GetAttr %arg0["_tensor_constant594"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5586 = torch.prim.GetAttr %arg0["_tensor_constant593"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5587 = torch.prim.GetAttr %arg0["_tensor_constant592"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5588 = torch.prim.GetAttr %arg0["_param_constant60"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5589 = torch.prim.GetAttr %arg0["_tensor_constant586"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5590 = torch.prim.GetAttr %arg0["_tensor_constant585"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5591 = torch.prim.GetAttr %arg0["_param_constant59"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5592 = torch.prim.GetAttr %arg0["_tensor_constant579"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5593 = torch.prim.GetAttr %arg0["_tensor_constant578"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5594 = torch.prim.GetAttr %arg0["_param_constant58"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5595 = torch.prim.GetAttr %arg0["_tensor_constant571"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5596 = torch.prim.GetAttr %arg0["_tensor_constant570"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5597 = torch.prim.GetAttr %arg0["_tensor_constant567"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5598 = torch.prim.GetAttr %arg0["_tensor_constant566"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5599 = torch.prim.GetAttr %arg0["_tensor_constant565"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5600 = torch.prim.GetAttr %arg0["_tensor_constant564"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5601 = torch.prim.GetAttr %arg0["_tensor_constant563"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5602 = torch.prim.GetAttr %arg0["_tensor_constant557"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5603 = torch.prim.GetAttr %arg0["_tensor_constant556"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5604 = torch.prim.GetAttr %arg0["_tensor_constant553"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5605 = torch.prim.GetAttr %arg0["_tensor_constant552"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5606 = torch.prim.GetAttr %arg0["_tensor_constant551"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5607 = torch.prim.GetAttr %arg0["_tensor_constant550"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5608 = torch.prim.GetAttr %arg0["_tensor_constant549"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5609 = torch.prim.GetAttr %arg0["_tensor_constant543"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5610 = torch.prim.GetAttr %arg0["_tensor_constant542"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5611 = torch.prim.GetAttr %arg0["_tensor_constant539"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5612 = torch.prim.GetAttr %arg0["_tensor_constant538"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5613 = torch.prim.GetAttr %arg0["_tensor_constant537"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5614 = torch.prim.GetAttr %arg0["_tensor_constant536"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5615 = torch.prim.GetAttr %arg0["_tensor_constant535"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5616 = torch.prim.GetAttr %arg0["_param_constant57"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5617 = torch.prim.GetAttr %arg0["_tensor_constant528"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5618 = torch.prim.GetAttr %arg0["_tensor_constant527"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5619 = torch.prim.GetAttr %arg0["_tensor_constant524"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5620 = torch.prim.GetAttr %arg0["_tensor_constant523"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5621 = torch.prim.GetAttr %arg0["_tensor_constant522"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5622 = torch.prim.GetAttr %arg0["_tensor_constant521"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5623 = torch.prim.GetAttr %arg0["_tensor_constant520"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5624 = torch.prim.GetAttr %arg0["_param_constant56"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5625 = torch.prim.GetAttr %arg0["_tensor_constant513"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5626 = torch.prim.GetAttr %arg0["_tensor_constant512"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5627 = torch.prim.GetAttr %arg0["_tensor_constant509"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5628 = torch.prim.GetAttr %arg0["_tensor_constant508"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5629 = torch.prim.GetAttr %arg0["_tensor_constant507"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5630 = torch.prim.GetAttr %arg0["_tensor_constant506"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5631 = torch.prim.GetAttr %arg0["_tensor_constant505"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5632 = torch.prim.GetAttr %arg0["_param_constant55"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5633 = torch.prim.GetAttr %arg0["_tensor_constant498"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5634 = torch.prim.GetAttr %arg0["_tensor_constant497"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5635 = torch.prim.GetAttr %arg0["_tensor_constant494"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5636 = torch.prim.GetAttr %arg0["_tensor_constant493"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5637 = torch.prim.GetAttr %arg0["_tensor_constant492"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5638 = torch.prim.GetAttr %arg0["_tensor_constant491"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5639 = torch.prim.GetAttr %arg0["_tensor_constant490"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5640 = torch.prim.GetAttr %arg0["_param_constant54"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5641 = torch.prim.GetAttr %arg0["_tensor_constant483"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5642 = torch.prim.GetAttr %arg0["_tensor_constant482"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5643 = torch.prim.GetAttr %arg0["_tensor_constant479"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5644 = torch.prim.GetAttr %arg0["_tensor_constant478"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5645 = torch.prim.GetAttr %arg0["_tensor_constant477"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5646 = torch.prim.GetAttr %arg0["_tensor_constant476"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5647 = torch.prim.GetAttr %arg0["_tensor_constant475"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5648 = torch.prim.GetAttr %arg0["_param_constant53"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5649 = torch.prim.GetAttr %arg0["_tensor_constant468"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5650 = torch.prim.GetAttr %arg0["_tensor_constant467"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5651 = torch.prim.GetAttr %arg0["_tensor_constant464"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5652 = torch.prim.GetAttr %arg0["_tensor_constant463"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5653 = torch.prim.GetAttr %arg0["_tensor_constant462"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5654 = torch.prim.GetAttr %arg0["_tensor_constant461"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5655 = torch.prim.GetAttr %arg0["_tensor_constant460"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5656 = torch.prim.GetAttr %arg0["_param_constant52"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5657 = torch.prim.GetAttr %arg0["_tensor_constant453"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5658 = torch.prim.GetAttr %arg0["_tensor_constant452"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5659 = torch.prim.GetAttr %arg0["_tensor_constant449"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5660 = torch.prim.GetAttr %arg0["_tensor_constant448"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5661 = torch.prim.GetAttr %arg0["_tensor_constant447"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5662 = torch.prim.GetAttr %arg0["_tensor_constant446"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5663 = torch.prim.GetAttr %arg0["_tensor_constant445"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5664 = torch.prim.GetAttr %arg0["_param_constant51"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5665 = torch.prim.GetAttr %arg0["_tensor_constant438"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5666 = torch.prim.GetAttr %arg0["_tensor_constant437"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5667 = torch.prim.GetAttr %arg0["_tensor_constant434"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5668 = torch.prim.GetAttr %arg0["_tensor_constant433"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5669 = torch.prim.GetAttr %arg0["_tensor_constant432"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5670 = torch.prim.GetAttr %arg0["_tensor_constant431"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5671 = torch.prim.GetAttr %arg0["_tensor_constant430"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5672 = torch.prim.GetAttr %arg0["_param_constant50"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5673 = torch.prim.GetAttr %arg0["_tensor_constant423"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5674 = torch.prim.GetAttr %arg0["_tensor_constant422"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5675 = torch.prim.GetAttr %arg0["_tensor_constant419"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5676 = torch.prim.GetAttr %arg0["_tensor_constant418"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5677 = torch.prim.GetAttr %arg0["_tensor_constant417"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5678 = torch.prim.GetAttr %arg0["_tensor_constant416"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5679 = torch.prim.GetAttr %arg0["_tensor_constant415"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5680 = torch.prim.GetAttr %arg0["_param_constant49"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5681 = torch.prim.GetAttr %arg0["_param_constant48"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5682 = torch.prim.GetAttr %arg0["_param_constant47"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5683 = torch.prim.GetAttr %arg0["_param_constant46"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5684 = torch.prim.GetAttr %arg0["_param_constant45"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5685 = torch.prim.GetAttr %arg0["_tensor_constant408"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5686 = torch.prim.GetAttr %arg0["_tensor_constant407"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5687 = torch.prim.GetAttr %arg0["_tensor_constant404"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5688 = torch.prim.GetAttr %arg0["_tensor_constant403"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5689 = torch.prim.GetAttr %arg0["_tensor_constant402"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5690 = torch.prim.GetAttr %arg0["_tensor_constant401"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5691 = torch.prim.GetAttr %arg0["_tensor_constant400"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5692 = torch.prim.GetAttr %arg0["_param_constant44"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5693 = torch.prim.GetAttr %arg0["_param_constant43"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5694 = torch.prim.GetAttr %arg0["_param_constant42"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5695 = torch.prim.GetAttr %arg0["_tensor_constant393"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5696 = torch.prim.GetAttr %arg0["_tensor_constant392"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5697 = torch.prim.GetAttr %arg0["_tensor_constant389"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5698 = torch.prim.GetAttr %arg0["_tensor_constant388"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5699 = torch.prim.GetAttr %arg0["_tensor_constant387"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5700 = torch.prim.GetAttr %arg0["_tensor_constant386"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5701 = torch.prim.GetAttr %arg0["_tensor_constant385"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5702 = torch.prim.GetAttr %arg0["_param_constant41"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5703 = torch.prim.GetAttr %arg0["_tensor_constant378"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5704 = torch.prim.GetAttr %arg0["_tensor_constant377"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5705 = torch.prim.GetAttr %arg0["_tensor_constant374"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5706 = torch.prim.GetAttr %arg0["_tensor_constant373"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5707 = torch.prim.GetAttr %arg0["_tensor_constant372"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5708 = torch.prim.GetAttr %arg0["_tensor_constant371"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5709 = torch.prim.GetAttr %arg0["_tensor_constant370"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5710 = torch.prim.GetAttr %arg0["_param_constant40"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5711 = torch.prim.GetAttr %arg0["_tensor_constant364"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5712 = torch.prim.GetAttr %arg0["_tensor_constant363"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5713 = torch.prim.GetAttr %arg0["_param_constant39"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5714 = torch.prim.GetAttr %arg0["_tensor_constant357"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5715 = torch.prim.GetAttr %arg0["_tensor_constant356"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5716 = torch.prim.GetAttr %arg0["_param_constant38"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5717 = torch.prim.GetAttr %arg0["_tensor_constant349"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5718 = torch.prim.GetAttr %arg0["_tensor_constant348"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5719 = torch.prim.GetAttr %arg0["_tensor_constant345"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5720 = torch.prim.GetAttr %arg0["_tensor_constant344"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5721 = torch.prim.GetAttr %arg0["_tensor_constant343"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5722 = torch.prim.GetAttr %arg0["_tensor_constant342"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5723 = torch.prim.GetAttr %arg0["_tensor_constant341"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5724 = torch.prim.GetAttr %arg0["_tensor_constant335"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5725 = torch.prim.GetAttr %arg0["_tensor_constant334"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5726 = torch.prim.GetAttr %arg0["_tensor_constant331"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5727 = torch.prim.GetAttr %arg0["_tensor_constant330"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5728 = torch.prim.GetAttr %arg0["_tensor_constant329"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5729 = torch.prim.GetAttr %arg0["_tensor_constant328"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5730 = torch.prim.GetAttr %arg0["_tensor_constant327"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5731 = torch.prim.GetAttr %arg0["_tensor_constant321"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5732 = torch.prim.GetAttr %arg0["_tensor_constant320"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5733 = torch.prim.GetAttr %arg0["_tensor_constant317"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5734 = torch.prim.GetAttr %arg0["_tensor_constant316"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5735 = torch.prim.GetAttr %arg0["_tensor_constant315"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5736 = torch.prim.GetAttr %arg0["_tensor_constant314"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5737 = torch.prim.GetAttr %arg0["_tensor_constant313"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5738 = torch.prim.GetAttr %arg0["_param_constant37"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5739 = torch.prim.GetAttr %arg0["_tensor_constant306"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5740 = torch.prim.GetAttr %arg0["_tensor_constant305"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5741 = torch.prim.GetAttr %arg0["_tensor_constant302"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5742 = torch.prim.GetAttr %arg0["_tensor_constant301"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5743 = torch.prim.GetAttr %arg0["_tensor_constant300"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5744 = torch.prim.GetAttr %arg0["_tensor_constant299"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5745 = torch.prim.GetAttr %arg0["_tensor_constant298"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5746 = torch.prim.GetAttr %arg0["_param_constant36"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5747 = torch.prim.GetAttr %arg0["_tensor_constant291"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5748 = torch.prim.GetAttr %arg0["_tensor_constant290"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5749 = torch.prim.GetAttr %arg0["_tensor_constant287"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5750 = torch.prim.GetAttr %arg0["_tensor_constant286"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5751 = torch.prim.GetAttr %arg0["_tensor_constant285"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5752 = torch.prim.GetAttr %arg0["_tensor_constant284"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5753 = torch.prim.GetAttr %arg0["_tensor_constant283"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5754 = torch.prim.GetAttr %arg0["_param_constant35"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5755 = torch.prim.GetAttr %arg0["_tensor_constant276"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5756 = torch.prim.GetAttr %arg0["_tensor_constant275"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5757 = torch.prim.GetAttr %arg0["_tensor_constant272"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5758 = torch.prim.GetAttr %arg0["_tensor_constant271"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5759 = torch.prim.GetAttr %arg0["_tensor_constant270"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5760 = torch.prim.GetAttr %arg0["_tensor_constant269"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5761 = torch.prim.GetAttr %arg0["_tensor_constant268"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5762 = torch.prim.GetAttr %arg0["_param_constant34"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5763 = torch.prim.GetAttr %arg0["_tensor_constant261"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5764 = torch.prim.GetAttr %arg0["_tensor_constant260"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5765 = torch.prim.GetAttr %arg0["_tensor_constant257"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5766 = torch.prim.GetAttr %arg0["_tensor_constant256"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5767 = torch.prim.GetAttr %arg0["_tensor_constant255"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5768 = torch.prim.GetAttr %arg0["_tensor_constant254"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5769 = torch.prim.GetAttr %arg0["_tensor_constant253"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5770 = torch.prim.GetAttr %arg0["_param_constant33"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5771 = torch.prim.GetAttr %arg0["_tensor_constant246"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5772 = torch.prim.GetAttr %arg0["_tensor_constant245"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5773 = torch.prim.GetAttr %arg0["_tensor_constant242"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5774 = torch.prim.GetAttr %arg0["_tensor_constant241"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5775 = torch.prim.GetAttr %arg0["_tensor_constant240"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5776 = torch.prim.GetAttr %arg0["_tensor_constant239"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5777 = torch.prim.GetAttr %arg0["_tensor_constant238"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5778 = torch.prim.GetAttr %arg0["_param_constant32"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5779 = torch.prim.GetAttr %arg0["_tensor_constant231"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5780 = torch.prim.GetAttr %arg0["_tensor_constant230"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5781 = torch.prim.GetAttr %arg0["_tensor_constant227"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5782 = torch.prim.GetAttr %arg0["_tensor_constant226"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5783 = torch.prim.GetAttr %arg0["_tensor_constant225"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5784 = torch.prim.GetAttr %arg0["_tensor_constant224"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5785 = torch.prim.GetAttr %arg0["_tensor_constant223"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5786 = torch.prim.GetAttr %arg0["_param_constant31"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5787 = torch.prim.GetAttr %arg0["_tensor_constant216"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5788 = torch.prim.GetAttr %arg0["_tensor_constant215"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5789 = torch.prim.GetAttr %arg0["_tensor_constant212"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5790 = torch.prim.GetAttr %arg0["_tensor_constant211"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5791 = torch.prim.GetAttr %arg0["_tensor_constant210"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5792 = torch.prim.GetAttr %arg0["_tensor_constant209"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5793 = torch.prim.GetAttr %arg0["_tensor_constant208"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5794 = torch.prim.GetAttr %arg0["_param_constant30"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5795 = torch.prim.GetAttr %arg0["_param_constant29"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5796 = torch.prim.GetAttr %arg0["_param_constant28"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5797 = torch.prim.GetAttr %arg0["_param_constant27"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5798 = torch.prim.GetAttr %arg0["_param_constant26"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5799 = torch.prim.GetAttr %arg0["_tensor_constant201"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5800 = torch.prim.GetAttr %arg0["_tensor_constant200"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5801 = torch.prim.GetAttr %arg0["_tensor_constant197"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5802 = torch.prim.GetAttr %arg0["_tensor_constant196"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5803 = torch.prim.GetAttr %arg0["_tensor_constant195"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5804 = torch.prim.GetAttr %arg0["_tensor_constant194"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5805 = torch.prim.GetAttr %arg0["_tensor_constant193"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5806 = torch.prim.GetAttr %arg0["_param_constant25"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5807 = torch.prim.GetAttr %arg0["_param_constant24"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5808 = torch.prim.GetAttr %arg0["_param_constant23"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5809 = torch.prim.GetAttr %arg0["_tensor_constant186"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5810 = torch.prim.GetAttr %arg0["_tensor_constant185"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5811 = torch.prim.GetAttr %arg0["_tensor_constant182"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5812 = torch.prim.GetAttr %arg0["_tensor_constant181"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5813 = torch.prim.GetAttr %arg0["_tensor_constant180"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5814 = torch.prim.GetAttr %arg0["_tensor_constant179"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5815 = torch.prim.GetAttr %arg0["_tensor_constant178"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5816 = torch.prim.GetAttr %arg0["_param_constant22"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5817 = torch.prim.GetAttr %arg0["_tensor_constant172"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5818 = torch.prim.GetAttr %arg0["_tensor_constant171"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5819 = torch.prim.GetAttr %arg0["_param_constant21"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5820 = torch.prim.GetAttr %arg0["_tensor_constant165"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5821 = torch.prim.GetAttr %arg0["_tensor_constant164"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5822 = torch.prim.GetAttr %arg0["_param_constant20"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5823 = torch.prim.GetAttr %arg0["_tensor_constant157"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5824 = torch.prim.GetAttr %arg0["_tensor_constant156"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5825 = torch.prim.GetAttr %arg0["_tensor_constant153"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5826 = torch.prim.GetAttr %arg0["_tensor_constant152"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5827 = torch.prim.GetAttr %arg0["_tensor_constant151"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5828 = torch.prim.GetAttr %arg0["_tensor_constant150"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5829 = torch.prim.GetAttr %arg0["_tensor_constant149"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5830 = torch.prim.GetAttr %arg0["_tensor_constant143"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5831 = torch.prim.GetAttr %arg0["_tensor_constant142"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5832 = torch.prim.GetAttr %arg0["_tensor_constant139"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5833 = torch.prim.GetAttr %arg0["_tensor_constant138"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5834 = torch.prim.GetAttr %arg0["_tensor_constant137"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5835 = torch.prim.GetAttr %arg0["_tensor_constant136"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5836 = torch.prim.GetAttr %arg0["_tensor_constant135"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5837 = torch.prim.GetAttr %arg0["_tensor_constant129"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5838 = torch.prim.GetAttr %arg0["_tensor_constant128"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5839 = torch.prim.GetAttr %arg0["_tensor_constant125"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5840 = torch.prim.GetAttr %arg0["_tensor_constant124"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5841 = torch.prim.GetAttr %arg0["_tensor_constant123"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5842 = torch.prim.GetAttr %arg0["_tensor_constant122"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5843 = torch.prim.GetAttr %arg0["_tensor_constant121"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5844 = torch.prim.GetAttr %arg0["_param_constant19"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5845 = torch.prim.GetAttr %arg0["_tensor_constant114"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5846 = torch.prim.GetAttr %arg0["_tensor_constant113"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5847 = torch.prim.GetAttr %arg0["_tensor_constant110"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5848 = torch.prim.GetAttr %arg0["_tensor_constant109"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5849 = torch.prim.GetAttr %arg0["_tensor_constant108"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5850 = torch.prim.GetAttr %arg0["_tensor_constant107"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5851 = torch.prim.GetAttr %arg0["_tensor_constant106"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5852 = torch.prim.GetAttr %arg0["_param_constant18"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5853 = torch.prim.GetAttr %arg0["_tensor_constant99"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5854 = torch.prim.GetAttr %arg0["_tensor_constant98"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5855 = torch.prim.GetAttr %arg0["_tensor_constant95"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5856 = torch.prim.GetAttr %arg0["_tensor_constant94"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5857 = torch.prim.GetAttr %arg0["_tensor_constant93"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5858 = torch.prim.GetAttr %arg0["_tensor_constant92"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5859 = torch.prim.GetAttr %arg0["_tensor_constant91"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5860 = torch.prim.GetAttr %arg0["_param_constant17"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5861 = torch.prim.GetAttr %arg0["_tensor_constant84"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5862 = torch.prim.GetAttr %arg0["_tensor_constant83"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5863 = torch.prim.GetAttr %arg0["_tensor_constant80"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5864 = torch.prim.GetAttr %arg0["_tensor_constant79"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5865 = torch.prim.GetAttr %arg0["_tensor_constant78"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5866 = torch.prim.GetAttr %arg0["_tensor_constant77"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5867 = torch.prim.GetAttr %arg0["_tensor_constant76"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5868 = torch.prim.GetAttr %arg0["_param_constant16"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5869 = torch.prim.GetAttr %arg0["_tensor_constant69"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5870 = torch.prim.GetAttr %arg0["_tensor_constant68"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5871 = torch.prim.GetAttr %arg0["_tensor_constant65"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5872 = torch.prim.GetAttr %arg0["_tensor_constant64"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5873 = torch.prim.GetAttr %arg0["_tensor_constant63"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5874 = torch.prim.GetAttr %arg0["_tensor_constant62"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5875 = torch.prim.GetAttr %arg0["_tensor_constant61"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5876 = torch.prim.GetAttr %arg0["_param_constant15"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5877 = torch.prim.GetAttr %arg0["_tensor_constant54"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5878 = torch.prim.GetAttr %arg0["_tensor_constant53"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5879 = torch.prim.GetAttr %arg0["_tensor_constant50"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5880 = torch.prim.GetAttr %arg0["_tensor_constant49"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5881 = torch.prim.GetAttr %arg0["_tensor_constant48"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5882 = torch.prim.GetAttr %arg0["_tensor_constant47"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5883 = torch.prim.GetAttr %arg0["_tensor_constant46"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5884 = torch.prim.GetAttr %arg0["_param_constant14"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5885 = torch.prim.GetAttr %arg0["_tensor_constant39"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5886 = torch.prim.GetAttr %arg0["_tensor_constant38"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5887 = torch.prim.GetAttr %arg0["_tensor_constant35"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5888 = torch.prim.GetAttr %arg0["_tensor_constant34"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5889 = torch.prim.GetAttr %arg0["_tensor_constant33"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5890 = torch.prim.GetAttr %arg0["_tensor_constant32"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5891 = torch.prim.GetAttr %arg0["_tensor_constant31"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5892 = torch.prim.GetAttr %arg0["_param_constant13"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5893 = torch.prim.GetAttr %arg0["_tensor_constant24"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5894 = torch.prim.GetAttr %arg0["_tensor_constant23"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5895 = torch.prim.GetAttr %arg0["_tensor_constant20"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5896 = torch.prim.GetAttr %arg0["_tensor_constant19"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5897 = torch.prim.GetAttr %arg0["_tensor_constant18"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5898 = torch.prim.GetAttr %arg0["_tensor_constant17"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5899 = torch.prim.GetAttr %arg0["_tensor_constant16"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5900 = torch.prim.GetAttr %arg0["_param_constant12"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5901 = torch.prim.GetAttr %arg0["_param_constant11"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5902 = torch.prim.GetAttr %arg0["_param_constant10"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5903 = torch.prim.GetAttr %arg0["_param_constant9"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5904 = torch.prim.GetAttr %arg0["_param_constant8"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5905 = torch.prim.GetAttr %arg0["_tensor_constant9"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5906 = torch.prim.GetAttr %arg0["_tensor_constant8"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5907 = torch.prim.GetAttr %arg0["_tensor_constant5"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5908 = torch.prim.GetAttr %arg0["_tensor_constant4"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5909 = torch.prim.GetAttr %arg0["_tensor_constant3"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5910 = torch.prim.GetAttr %arg0["_tensor_constant2"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5911 = torch.prim.GetAttr %arg0["_tensor_constant1"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5912 = torch.prim.GetAttr %arg0["_param_constant7"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5913 = torch.prim.GetAttr %arg0["_param_constant6"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5914 = torch.prim.GetAttr %arg0["_param_constant5"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5915 = torch.prim.GetAttr %arg0["_param_constant4"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5916 = torch.prim.GetAttr %arg0["_param_constant3"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5917 = torch.prim.GetAttr %arg0["_param_constant2"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5918 = torch.prim.GetAttr %arg0["_param_constant1"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %5919 = torch.prim.GetAttr %arg0["_param_constant0"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0) | |
| %int6 = torch.constant.int 6 loc(#loc1) | |
| %none_0 = torch.constant.none loc(#loc0) | |
| %none_1 = torch.constant.none loc(#loc0) | |
| %none_2 = torch.constant.none loc(#loc0) | |
| %false_3 = torch.constant.bool false loc(#loc1) | |
| %none_4 = torch.constant.none loc(#loc0) | |
| %5920 = torch.aten._to_copy %3920, %int6, %none_0, %none_1, %none_2, %false_3, %none_4 : !torch.tensor<[],si64>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[],f32> loc(#loc1) | |
| %int0 = torch.constant.int 0 loc(#loc1) | |
| %5921 = torch.aten.unsqueeze %5920, %int0 : !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[1],f32> loc(#loc1) | |
| %int2 = torch.constant.int 2 loc(#loc1) | |
| %5922 = torch.prim.ListConstruct %int2 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_5 = torch.constant.bool false loc(#loc1) | |
| %5923 = torch.aten.expand %5921, %5922, %false_5 : !torch.tensor<[1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2],f32> loc(#loc1) | |
| %int0_6 = torch.constant.int 0 loc(#loc1) | |
| %int160 = torch.constant.int 160 loc(#loc1) | |
| %int6_7 = torch.constant.int 6 loc(#loc1) | |
| %none_8 = torch.constant.none loc(#loc0) | |
| %cpu = torch.constant.device "cpu" loc(#loc1) | |
| %false_9 = torch.constant.bool false loc(#loc1) | |
| %5924 = torch.aten.arange.start %int0_6, %int160, %int6_7, %none_8, %cpu, %false_9 : !torch.int, !torch.int, !torch.int, !torch.none, !torch.Device, !torch.bool -> !torch.tensor<[160],f32> loc(#loc1) | |
| %5925 = torch.tensor.literal(dense<-9.2103403719761836> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %5926 = torch.aten.mul.Tensor %5924, %5925 : !torch.tensor<[160],f32>, !torch.tensor<[],f64> -> !torch.tensor<[160],f32> loc(#loc1) | |
| %5927 = torch.tensor.literal(dense<160> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %5928 = torch.aten.div.Tensor %5926, %5927 : !torch.tensor<[160],f32>, !torch.tensor<[],si64> -> !torch.tensor<[160],f32> loc(#loc1) | |
| %5929 = torch.aten.exp %5928 : !torch.tensor<[160],f32> -> !torch.tensor<[160],f32> loc(#loc1) | |
| %int0_10 = torch.constant.int 0 loc(#loc1) | |
| %int0_11 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1 = torch.constant.int 1 loc(#loc1) | |
| %5930 = torch.aten.slice.Tensor %5923, %int0_10, %int0_11, %int9223372036854775807, %int1 : !torch.tensor<[2],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2],f32> loc(#loc1) | |
| %int1_12 = torch.constant.int 1 loc(#loc1) | |
| %5931 = torch.aten.unsqueeze %5930, %int1_12 : !torch.tensor<[2],f32>, !torch.int -> !torch.tensor<[2,1],f32> loc(#loc1) | |
| %int0_13 = torch.constant.int 0 loc(#loc1) | |
| %5932 = torch.aten.unsqueeze %5929, %int0_13 : !torch.tensor<[160],f32>, !torch.int -> !torch.tensor<[1,160],f32> loc(#loc1) | |
| %int1_14 = torch.constant.int 1 loc(#loc1) | |
| %int0_15 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_16 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_17 = torch.constant.int 1 loc(#loc1) | |
| %5933 = torch.aten.slice.Tensor %5932, %int1_14, %int0_15, %int9223372036854775807_16, %int1_17 : !torch.tensor<[1,160],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[1,160],f32> loc(#loc1) | |
| %5934 = torch.aten.mul.Tensor %5931, %5933 : !torch.tensor<[2,1],f32>, !torch.tensor<[1,160],f32> -> !torch.tensor<[2,160],f32> loc(#loc1) | |
| %5935 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %5936 = torch.aten.mul.Tensor %5934, %5935 : !torch.tensor<[2,160],f32>, !torch.tensor<[],si64> -> !torch.tensor<[2,160],f32> loc(#loc1) | |
| %5937 = torch.aten.sin %5936 : !torch.tensor<[2,160],f32> -> !torch.tensor<[2,160],f32> loc(#loc1) | |
| %5938 = torch.aten.cos %5936 : !torch.tensor<[2,160],f32> -> !torch.tensor<[2,160],f32> loc(#loc1) | |
| %5939 = torch.prim.ListConstruct %5937, %5938 : (!torch.tensor<[2,160],f32>, !torch.tensor<[2,160],f32>) -> !torch.list<tensor> loc(#loc0) | |
| %int-1 = torch.constant.int -1 loc(#loc1) | |
| %5940 = torch.aten.cat %5939, %int-1 : !torch.list<tensor>, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int0_18 = torch.constant.int 0 loc(#loc1) | |
| %int0_19 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_20 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_21 = torch.constant.int 1 loc(#loc1) | |
| %5941 = torch.aten.slice.Tensor %5940, %int0_18, %int0_19, %int9223372036854775807_20, %int1_21 : !torch.tensor<[2,320],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int1_22 = torch.constant.int 1 loc(#loc1) | |
| %int160_23 = torch.constant.int 160 loc(#loc1) | |
| %int9223372036854775807_24 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_25 = torch.constant.int 1 loc(#loc1) | |
| %5942 = torch.aten.slice.Tensor %5941, %int1_22, %int160_23, %int9223372036854775807_24, %int1_25 : !torch.tensor<[2,320],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,160],f32> loc(#loc1) | |
| %int0_26 = torch.constant.int 0 loc(#loc1) | |
| %int0_27 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_28 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_29 = torch.constant.int 1 loc(#loc1) | |
| %5943 = torch.aten.slice.Tensor %5940, %int0_26, %int0_27, %int9223372036854775807_28, %int1_29 : !torch.tensor<[2,320],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int1_30 = torch.constant.int 1 loc(#loc1) | |
| %int0_31 = torch.constant.int 0 loc(#loc1) | |
| %int160_32 = torch.constant.int 160 loc(#loc1) | |
| %int1_33 = torch.constant.int 1 loc(#loc1) | |
| %5944 = torch.aten.slice.Tensor %5943, %int1_30, %int0_31, %int160_32, %int1_33 : !torch.tensor<[2,320],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,160],f32> loc(#loc1) | |
| %5945 = torch.prim.ListConstruct %5942, %5944 : (!torch.tensor<[2,160],f32>, !torch.tensor<[2,160],f32>) -> !torch.list<tensor> loc(#loc0) | |
| %int-1_34 = torch.constant.int -1 loc(#loc1) | |
| %5946 = torch.aten.cat %5945, %int-1_34 : !torch.list<tensor>, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %5947 = torch.aten.t %5919 : !torch.tensor -> !torch.tensor<[320,1280],f32> loc(#loc1) | |
| %int1_35 = torch.constant.int 1 loc(#loc1) | |
| %int1_36 = torch.constant.int 1 loc(#loc1) | |
| %5948 = torch.aten.addmm %5918, %5946, %5947, %int1_35, %int1_36 : !torch.tensor, !torch.tensor<[2,320],f32>, !torch.tensor<[320,1280],f32>, !torch.int, !torch.int -> !torch.tensor<[2,1280],f32> loc(#loc1) | |
| %5949 = torch.aten.silu %5948 : !torch.tensor<[2,1280],f32> -> !torch.tensor<[2,1280],f32> loc(#loc1) | |
| %5950 = torch.aten.t %5917 : !torch.tensor -> !torch.tensor<[1280,1280],f32> loc(#loc1) | |
| %int1_37 = torch.constant.int 1 loc(#loc1) | |
| %int1_38 = torch.constant.int 1 loc(#loc1) | |
| %5951 = torch.aten.addmm %5916, %5949, %5950, %int1_37, %int1_38 : !torch.tensor, !torch.tensor<[2,1280],f32>, !torch.tensor<[1280,1280],f32>, !torch.int, !torch.int -> !torch.tensor<[2,1280],f32> loc(#loc1) | |
| %int1_39 = torch.constant.int 1 loc(#loc1) | |
| %int1_40 = torch.constant.int 1 loc(#loc1) | |
| %5952 = torch.prim.ListConstruct %int1_39, %int1_40 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_41 = torch.constant.int 1 loc(#loc1) | |
| %int1_42 = torch.constant.int 1 loc(#loc1) | |
| %5953 = torch.prim.ListConstruct %int1_41, %int1_42 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_43 = torch.constant.int 1 loc(#loc1) | |
| %int1_44 = torch.constant.int 1 loc(#loc1) | |
| %5954 = torch.prim.ListConstruct %int1_43, %int1_44 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_45 = torch.constant.bool false loc(#loc1) | |
| %int0_46 = torch.constant.int 0 loc(#loc1) | |
| %int0_47 = torch.constant.int 0 loc(#loc1) | |
| %5955 = torch.prim.ListConstruct %int0_46, %int0_47 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_48 = torch.constant.int 1 loc(#loc1) | |
| %false_49 = torch.constant.bool false loc(#loc1) | |
| %false_50 = torch.constant.bool false loc(#loc1) | |
| %true = torch.constant.bool true loc(#loc1) | |
| %true_51 = torch.constant.bool true loc(#loc1) | |
| %5956 = torch.aten._convolution %3919, %5915, %5914, %5952, %5953, %5954, %false_45, %5955, %int1_48, %false_49, %false_50, %true, %true_51 : !torch.tensor<[2,4,64,64],f32>, !torch.tensor, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int2_52 = torch.constant.int 2 loc(#loc1) | |
| %int32 = torch.constant.int 32 loc(#loc1) | |
| %int10 = torch.constant.int 10 loc(#loc1) | |
| %int4096 = torch.constant.int 4096 loc(#loc1) | |
| %5957 = torch.prim.ListConstruct %int2_52, %int32, %int10, %int4096 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %5958 = torch.aten.view %5956, %5957 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_53 = torch.constant.int 2 loc(#loc1) | |
| %int3 = torch.constant.int 3 loc(#loc1) | |
| %5959 = torch.prim.ListConstruct %int2_53, %int3 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_54 = torch.constant.int 0 loc(#loc1) | |
| %true_55 = torch.constant.bool true loc(#loc1) | |
| %5960 = torch.aten.var.correction %5958, %5959, %int0_54, %true_55 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_56 = torch.constant.int 2 loc(#loc1) | |
| %int3_57 = torch.constant.int 3 loc(#loc1) | |
| %5961 = torch.prim.ListConstruct %int2_56, %int3_57 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_58 = torch.constant.bool true loc(#loc1) | |
| %none_59 = torch.constant.none loc(#loc0) | |
| %5962 = torch.aten.mean.dim %5958, %5961, %true_58, %none_59 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %5963 = torch.tensor.literal(dense<1.000000e-05> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_60 = torch.constant.int 1 loc(#loc1) | |
| %5964 = torch.aten.add.Tensor %5960, %5963, %int1_60 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %5965 = torch.aten.rsqrt %5964 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_61 = torch.constant.int 1 loc(#loc1) | |
| %5966 = torch.aten.sub.Tensor %5958, %5962, %int1_61 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %5967 = torch.aten.mul.Tensor %5966, %5965 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_62 = torch.constant.int 2 loc(#loc1) | |
| %int320 = torch.constant.int 320 loc(#loc1) | |
| %int64 = torch.constant.int 64 loc(#loc1) | |
| %int64_63 = torch.constant.int 64 loc(#loc1) | |
| %5968 = torch.prim.ListConstruct %int2_62, %int320, %int64, %int64_63 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %5969 = torch.aten.view %5967, %5968 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int-1_64 = torch.constant.int -1 loc(#loc1) | |
| %5970 = torch.aten.unsqueeze %5913, %int-1_64 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_65 = torch.constant.int -1 loc(#loc1) | |
| %5971 = torch.aten.unsqueeze %5970, %int-1_65 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %5972 = torch.aten.mul.Tensor %5969, %5971 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,1,1],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int-1_66 = torch.constant.int -1 loc(#loc1) | |
| %5973 = torch.aten.unsqueeze %5912, %int-1_66 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_67 = torch.constant.int -1 loc(#loc1) | |
| %5974 = torch.aten.unsqueeze %5973, %int-1_67 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %int1_68 = torch.constant.int 1 loc(#loc1) | |
| %5975 = torch.aten.add.Tensor %5972, %5974, %int1_68 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,1,1],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %5976 = torch.aten.silu %5975 : !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %5977 = torch.aten.detach %5911 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %5978 = torch.aten.detach %5977 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_69 = torch.constant.int 2 loc(#loc1) | |
| %int20 = torch.constant.int 20 loc(#loc1) | |
| %int16 = torch.constant.int 16 loc(#loc1) | |
| %int64_70 = torch.constant.int 64 loc(#loc1) | |
| %int64_71 = torch.constant.int 64 loc(#loc1) | |
| %5979 = torch.prim.ListConstruct %int2_69, %int20, %int16, %int64_70, %int64_71 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %5980 = torch.aten.view %5976, %5979 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %5981 = torch.aten.abs %5980 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_72 = torch.constant.int 2 loc(#loc1) | |
| %true_73 = torch.constant.bool true loc(#loc1) | |
| %values, %indices = torch.aten.max.dim %5981, %int2_72, %true_73 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_74 = torch.constant.int 2 loc(#loc1) | |
| %int20_75 = torch.constant.int 20 loc(#loc1) | |
| %int1_76 = torch.constant.int 1 loc(#loc1) | |
| %int64_77 = torch.constant.int 64 loc(#loc1) | |
| %int64_78 = torch.constant.int 64 loc(#loc1) | |
| %5982 = torch.prim.ListConstruct %int2_74, %int20_75, %int1_76, %int64_77, %int64_78 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %5983 = torch.aten.view %values, %5982 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_79 = torch.constant.int 2 loc(#loc1) | |
| %int20_80 = torch.constant.int 20 loc(#loc1) | |
| %int16_81 = torch.constant.int 16 loc(#loc1) | |
| %int64_82 = torch.constant.int 64 loc(#loc1) | |
| %int64_83 = torch.constant.int 64 loc(#loc1) | |
| %5984 = torch.prim.ListConstruct %int2_79, %int20_80, %int16_81, %int64_82, %int64_83 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_84 = torch.constant.bool false loc(#loc1) | |
| %5985 = torch.aten.expand %5983, %5984, %false_84 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_85 = torch.constant.int 0 loc(#loc1) | |
| %5986 = torch.aten.clone %5985, %int0_85 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_86 = torch.constant.int 2 loc(#loc1) | |
| %int320_87 = torch.constant.int 320 loc(#loc1) | |
| %int64_88 = torch.constant.int 64 loc(#loc1) | |
| %int64_89 = torch.constant.int 64 loc(#loc1) | |
| %5987 = torch.prim.ListConstruct %int2_86, %int320_87, %int64_88, %int64_89 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %5988 = torch.aten._unsafe_view %5986, %5987 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %5989 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_90 = torch.constant.int 1 loc(#loc1) | |
| %5990 = torch.aten.sub.Tensor %5978, %5989, %int1_90 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %5991 = torch.aten.lift_fresh_copy %5910 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %5992 = torch.aten.pow.Tensor_Tensor %5991, %5990 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %5993 = torch.aten.neg %5992 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %5994 = torch.aten.neg %5993 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %5995 = torch.aten.div.Tensor %5988, %5994 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %5996 = torch.aten.detach %5909 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %5997 = torch.aten.detach %5996 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %5998 = torch.aten.div.Tensor %5976, %5995 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_91 = torch.constant.int 1 loc(#loc1) | |
| %5999 = torch.aten.add.Tensor %5998, %5997, %int1_91 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6000 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_92 = torch.constant.int 1 loc(#loc1) | |
| %6001 = torch.aten.sub.Tensor %5978, %6000, %int1_92 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6002 = torch.aten.lift_fresh_copy %5908 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6003 = torch.aten.pow.Tensor_Tensor %6002, %6001 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6004 = torch.aten.neg %6003 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6005 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_93 = torch.constant.int 1 loc(#loc1) | |
| %6006 = torch.aten.sub.Tensor %5978, %6005, %int1_93 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6007 = torch.aten.lift_fresh_copy %5907 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6008 = torch.aten.pow.Tensor_Tensor %6007, %6006 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6009 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_94 = torch.constant.int 1 loc(#loc1) | |
| %6010 = torch.aten.sub.Tensor %6008, %6009, %int1_94 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6011 = torch.aten.gt.Tensor %5999, %6010 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6012 = torch.aten.where.self %6011, %6010, %5999 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6013 = torch.aten.lt.Tensor %6012, %6004 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6014 = torch.aten.where.self %6013, %6004, %6012 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6015 = torch.operator "aten.round"(%6014) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_95 = torch.constant.int 1 loc(#loc1) | |
| %6016 = torch.aten.sub.Tensor %6015, %5997, %int1_95 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6017 = torch.aten.mul.Tensor %6016, %5995 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_96 = torch.constant.int 320 loc(#loc1) | |
| %int20_97 = torch.constant.int 20 loc(#loc1) | |
| %int16_98 = torch.constant.int 16 loc(#loc1) | |
| %int3_99 = torch.constant.int 3 loc(#loc1) | |
| %int3_100 = torch.constant.int 3 loc(#loc1) | |
| %6018 = torch.prim.ListConstruct %int320_96, %int20_97, %int16_98, %int3_99, %int3_100 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_101 = torch.constant.bool false loc(#loc1) | |
| %6019 = torch.aten.expand %5906, %6018, %false_101 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int0_102 = torch.constant.int 0 loc(#loc1) | |
| %6020 = torch.aten.clone %6019, %int0_102 : !torch.tensor<[320,20,16,3,3],f16>, !torch.int -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int320_103 = torch.constant.int 320 loc(#loc1) | |
| %int320_104 = torch.constant.int 320 loc(#loc1) | |
| %int3_105 = torch.constant.int 3 loc(#loc1) | |
| %int3_106 = torch.constant.int 3 loc(#loc1) | |
| %6021 = torch.prim.ListConstruct %int320_103, %int320_104, %int3_105, %int3_106 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6022 = torch.aten._unsafe_view %6020, %6021 : !torch.tensor<[320,20,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %6023 = torch.aten.mul.Tensor %5905, %6022 : !torch.tensor, !torch.tensor<[320,320,3,3],f16> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %int6_107 = torch.constant.int 6 loc(#loc1) | |
| %none_108 = torch.constant.none loc(#loc0) | |
| %none_109 = torch.constant.none loc(#loc0) | |
| %none_110 = torch.constant.none loc(#loc0) | |
| %false_111 = torch.constant.bool false loc(#loc1) | |
| %none_112 = torch.constant.none loc(#loc0) | |
| %6024 = torch.aten._to_copy %6023, %int6_107, %none_108, %none_109, %none_110, %false_111, %none_112 : !torch.tensor<[320,320,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,3,3],f32> loc(#loc1) | |
| %int1_113 = torch.constant.int 1 loc(#loc1) | |
| %int1_114 = torch.constant.int 1 loc(#loc1) | |
| %6025 = torch.prim.ListConstruct %int1_113, %int1_114 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_115 = torch.constant.int 1 loc(#loc1) | |
| %int1_116 = torch.constant.int 1 loc(#loc1) | |
| %6026 = torch.prim.ListConstruct %int1_115, %int1_116 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_117 = torch.constant.int 1 loc(#loc1) | |
| %int1_118 = torch.constant.int 1 loc(#loc1) | |
| %6027 = torch.prim.ListConstruct %int1_117, %int1_118 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_119 = torch.constant.bool false loc(#loc1) | |
| %int0_120 = torch.constant.int 0 loc(#loc1) | |
| %int0_121 = torch.constant.int 0 loc(#loc1) | |
| %6028 = torch.prim.ListConstruct %int0_120, %int0_121 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_122 = torch.constant.int 1 loc(#loc1) | |
| %false_123 = torch.constant.bool false loc(#loc1) | |
| %false_124 = torch.constant.bool false loc(#loc1) | |
| %true_125 = torch.constant.bool true loc(#loc1) | |
| %true_126 = torch.constant.bool true loc(#loc1) | |
| %6029 = torch.aten._convolution %6017, %6024, %5904, %6025, %6026, %6027, %false_119, %6028, %int1_122, %false_123, %false_124, %true_125, %true_126 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6030 = torch.aten.silu %5951 : !torch.tensor<[2,1280],f32> -> !torch.tensor<[2,1280],f32> loc(#loc1) | |
| %6031 = torch.aten.t %5903 : !torch.tensor -> !torch.tensor<[1280,320],f32> loc(#loc1) | |
| %int1_127 = torch.constant.int 1 loc(#loc1) | |
| %int1_128 = torch.constant.int 1 loc(#loc1) | |
| %6032 = torch.aten.addmm %5902, %6030, %6031, %int1_127, %int1_128 : !torch.tensor, !torch.tensor<[2,1280],f32>, !torch.tensor<[1280,320],f32>, !torch.int, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int0_129 = torch.constant.int 0 loc(#loc1) | |
| %int0_130 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_131 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_132 = torch.constant.int 1 loc(#loc1) | |
| %6033 = torch.aten.slice.Tensor %6032, %int0_129, %int0_130, %int9223372036854775807_131, %int1_132 : !torch.tensor<[2,320],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int1_133 = torch.constant.int 1 loc(#loc1) | |
| %int0_134 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_135 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_136 = torch.constant.int 1 loc(#loc1) | |
| %6034 = torch.aten.slice.Tensor %6033, %int1_133, %int0_134, %int9223372036854775807_135, %int1_136 : !torch.tensor<[2,320],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int2_137 = torch.constant.int 2 loc(#loc1) | |
| %6035 = torch.aten.unsqueeze %6034, %int2_137 : !torch.tensor<[2,320],f32>, !torch.int -> !torch.tensor<[2,320,1],f32> loc(#loc1) | |
| %int3_138 = torch.constant.int 3 loc(#loc1) | |
| %6036 = torch.aten.unsqueeze %6035, %int3_138 : !torch.tensor<[2,320,1],f32>, !torch.int -> !torch.tensor<[2,320,1,1],f32> loc(#loc1) | |
| %int1_139 = torch.constant.int 1 loc(#loc1) | |
| %6037 = torch.aten.add.Tensor %6029, %6036, %int1_139 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,1,1],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int2_140 = torch.constant.int 2 loc(#loc1) | |
| %int32_141 = torch.constant.int 32 loc(#loc1) | |
| %int10_142 = torch.constant.int 10 loc(#loc1) | |
| %int4096_143 = torch.constant.int 4096 loc(#loc1) | |
| %6038 = torch.prim.ListConstruct %int2_140, %int32_141, %int10_142, %int4096_143 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6039 = torch.aten.view %6037, %6038 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_144 = torch.constant.int 2 loc(#loc1) | |
| %int3_145 = torch.constant.int 3 loc(#loc1) | |
| %6040 = torch.prim.ListConstruct %int2_144, %int3_145 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_146 = torch.constant.int 0 loc(#loc1) | |
| %true_147 = torch.constant.bool true loc(#loc1) | |
| %6041 = torch.aten.var.correction %6039, %6040, %int0_146, %true_147 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_148 = torch.constant.int 2 loc(#loc1) | |
| %int3_149 = torch.constant.int 3 loc(#loc1) | |
| %6042 = torch.prim.ListConstruct %int2_148, %int3_149 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_150 = torch.constant.bool true loc(#loc1) | |
| %none_151 = torch.constant.none loc(#loc0) | |
| %6043 = torch.aten.mean.dim %6039, %6042, %true_150, %none_151 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %6044 = torch.tensor.literal(dense<1.000000e-05> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_152 = torch.constant.int 1 loc(#loc1) | |
| %6045 = torch.aten.add.Tensor %6041, %6044, %int1_152 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %6046 = torch.aten.rsqrt %6045 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_153 = torch.constant.int 1 loc(#loc1) | |
| %6047 = torch.aten.sub.Tensor %6039, %6043, %int1_153 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %6048 = torch.aten.mul.Tensor %6047, %6046 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_154 = torch.constant.int 2 loc(#loc1) | |
| %int320_155 = torch.constant.int 320 loc(#loc1) | |
| %int64_156 = torch.constant.int 64 loc(#loc1) | |
| %int64_157 = torch.constant.int 64 loc(#loc1) | |
| %6049 = torch.prim.ListConstruct %int2_154, %int320_155, %int64_156, %int64_157 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6050 = torch.aten.view %6048, %6049 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int-1_158 = torch.constant.int -1 loc(#loc1) | |
| %6051 = torch.aten.unsqueeze %5901, %int-1_158 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_159 = torch.constant.int -1 loc(#loc1) | |
| %6052 = torch.aten.unsqueeze %6051, %int-1_159 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %6053 = torch.aten.mul.Tensor %6050, %6052 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,1,1],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int-1_160 = torch.constant.int -1 loc(#loc1) | |
| %6054 = torch.aten.unsqueeze %5900, %int-1_160 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_161 = torch.constant.int -1 loc(#loc1) | |
| %6055 = torch.aten.unsqueeze %6054, %int-1_161 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %int1_162 = torch.constant.int 1 loc(#loc1) | |
| %6056 = torch.aten.add.Tensor %6053, %6055, %int1_162 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,1,1],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6057 = torch.aten.silu %6056 : !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6058 = torch.aten.detach %5899 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6059 = torch.aten.detach %6058 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_163 = torch.constant.int 2 loc(#loc1) | |
| %int20_164 = torch.constant.int 20 loc(#loc1) | |
| %int16_165 = torch.constant.int 16 loc(#loc1) | |
| %int64_166 = torch.constant.int 64 loc(#loc1) | |
| %int64_167 = torch.constant.int 64 loc(#loc1) | |
| %6060 = torch.prim.ListConstruct %int2_163, %int20_164, %int16_165, %int64_166, %int64_167 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6061 = torch.aten.view %6057, %6060 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %6062 = torch.aten.abs %6061 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_168 = torch.constant.int 2 loc(#loc1) | |
| %true_169 = torch.constant.bool true loc(#loc1) | |
| %values_170, %indices_171 = torch.aten.max.dim %6062, %int2_168, %true_169 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_172 = torch.constant.int 2 loc(#loc1) | |
| %int20_173 = torch.constant.int 20 loc(#loc1) | |
| %int1_174 = torch.constant.int 1 loc(#loc1) | |
| %int64_175 = torch.constant.int 64 loc(#loc1) | |
| %int64_176 = torch.constant.int 64 loc(#loc1) | |
| %6063 = torch.prim.ListConstruct %int2_172, %int20_173, %int1_174, %int64_175, %int64_176 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6064 = torch.aten.view %values_170, %6063 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_177 = torch.constant.int 2 loc(#loc1) | |
| %int20_178 = torch.constant.int 20 loc(#loc1) | |
| %int16_179 = torch.constant.int 16 loc(#loc1) | |
| %int64_180 = torch.constant.int 64 loc(#loc1) | |
| %int64_181 = torch.constant.int 64 loc(#loc1) | |
| %6065 = torch.prim.ListConstruct %int2_177, %int20_178, %int16_179, %int64_180, %int64_181 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_182 = torch.constant.bool false loc(#loc1) | |
| %6066 = torch.aten.expand %6064, %6065, %false_182 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_183 = torch.constant.int 0 loc(#loc1) | |
| %6067 = torch.aten.clone %6066, %int0_183 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_184 = torch.constant.int 2 loc(#loc1) | |
| %int320_185 = torch.constant.int 320 loc(#loc1) | |
| %int64_186 = torch.constant.int 64 loc(#loc1) | |
| %int64_187 = torch.constant.int 64 loc(#loc1) | |
| %6068 = torch.prim.ListConstruct %int2_184, %int320_185, %int64_186, %int64_187 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6069 = torch.aten._unsafe_view %6067, %6068 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6070 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_188 = torch.constant.int 1 loc(#loc1) | |
| %6071 = torch.aten.sub.Tensor %6059, %6070, %int1_188 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6072 = torch.aten.lift_fresh_copy %5898 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6073 = torch.aten.pow.Tensor_Tensor %6072, %6071 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6074 = torch.aten.neg %6073 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6075 = torch.aten.neg %6074 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6076 = torch.aten.div.Tensor %6069, %6075 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6077 = torch.aten.detach %5897 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6078 = torch.aten.detach %6077 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6079 = torch.aten.div.Tensor %6057, %6076 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_189 = torch.constant.int 1 loc(#loc1) | |
| %6080 = torch.aten.add.Tensor %6079, %6078, %int1_189 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6081 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_190 = torch.constant.int 1 loc(#loc1) | |
| %6082 = torch.aten.sub.Tensor %6059, %6081, %int1_190 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6083 = torch.aten.lift_fresh_copy %5896 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6084 = torch.aten.pow.Tensor_Tensor %6083, %6082 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6085 = torch.aten.neg %6084 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6086 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_191 = torch.constant.int 1 loc(#loc1) | |
| %6087 = torch.aten.sub.Tensor %6059, %6086, %int1_191 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6088 = torch.aten.lift_fresh_copy %5895 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6089 = torch.aten.pow.Tensor_Tensor %6088, %6087 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6090 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_192 = torch.constant.int 1 loc(#loc1) | |
| %6091 = torch.aten.sub.Tensor %6089, %6090, %int1_192 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6092 = torch.aten.gt.Tensor %6080, %6091 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6093 = torch.aten.where.self %6092, %6091, %6080 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6094 = torch.aten.lt.Tensor %6093, %6085 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6095 = torch.aten.where.self %6094, %6085, %6093 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6096 = torch.operator "aten.round"(%6095) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_193 = torch.constant.int 1 loc(#loc1) | |
| %6097 = torch.aten.sub.Tensor %6096, %6078, %int1_193 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6098 = torch.aten.mul.Tensor %6097, %6076 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_194 = torch.constant.int 320 loc(#loc1) | |
| %int20_195 = torch.constant.int 20 loc(#loc1) | |
| %int16_196 = torch.constant.int 16 loc(#loc1) | |
| %int3_197 = torch.constant.int 3 loc(#loc1) | |
| %int3_198 = torch.constant.int 3 loc(#loc1) | |
| %6099 = torch.prim.ListConstruct %int320_194, %int20_195, %int16_196, %int3_197, %int3_198 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_199 = torch.constant.bool false loc(#loc1) | |
| %6100 = torch.aten.expand %5894, %6099, %false_199 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int0_200 = torch.constant.int 0 loc(#loc1) | |
| %6101 = torch.aten.clone %6100, %int0_200 : !torch.tensor<[320,20,16,3,3],f16>, !torch.int -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int320_201 = torch.constant.int 320 loc(#loc1) | |
| %int320_202 = torch.constant.int 320 loc(#loc1) | |
| %int3_203 = torch.constant.int 3 loc(#loc1) | |
| %int3_204 = torch.constant.int 3 loc(#loc1) | |
| %6102 = torch.prim.ListConstruct %int320_201, %int320_202, %int3_203, %int3_204 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6103 = torch.aten._unsafe_view %6101, %6102 : !torch.tensor<[320,20,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %6104 = torch.aten.mul.Tensor %5893, %6103 : !torch.tensor, !torch.tensor<[320,320,3,3],f16> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %int6_205 = torch.constant.int 6 loc(#loc1) | |
| %none_206 = torch.constant.none loc(#loc0) | |
| %none_207 = torch.constant.none loc(#loc0) | |
| %none_208 = torch.constant.none loc(#loc0) | |
| %false_209 = torch.constant.bool false loc(#loc1) | |
| %none_210 = torch.constant.none loc(#loc0) | |
| %6105 = torch.aten._to_copy %6104, %int6_205, %none_206, %none_207, %none_208, %false_209, %none_210 : !torch.tensor<[320,320,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,3,3],f32> loc(#loc1) | |
| %int1_211 = torch.constant.int 1 loc(#loc1) | |
| %int1_212 = torch.constant.int 1 loc(#loc1) | |
| %6106 = torch.prim.ListConstruct %int1_211, %int1_212 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_213 = torch.constant.int 1 loc(#loc1) | |
| %int1_214 = torch.constant.int 1 loc(#loc1) | |
| %6107 = torch.prim.ListConstruct %int1_213, %int1_214 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_215 = torch.constant.int 1 loc(#loc1) | |
| %int1_216 = torch.constant.int 1 loc(#loc1) | |
| %6108 = torch.prim.ListConstruct %int1_215, %int1_216 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_217 = torch.constant.bool false loc(#loc1) | |
| %int0_218 = torch.constant.int 0 loc(#loc1) | |
| %int0_219 = torch.constant.int 0 loc(#loc1) | |
| %6109 = torch.prim.ListConstruct %int0_218, %int0_219 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_220 = torch.constant.int 1 loc(#loc1) | |
| %false_221 = torch.constant.bool false loc(#loc1) | |
| %false_222 = torch.constant.bool false loc(#loc1) | |
| %true_223 = torch.constant.bool true loc(#loc1) | |
| %true_224 = torch.constant.bool true loc(#loc1) | |
| %6110 = torch.aten._convolution %6098, %6105, %5892, %6106, %6107, %6108, %false_217, %6109, %int1_220, %false_221, %false_222, %true_223, %true_224 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_225 = torch.constant.int 1 loc(#loc1) | |
| %6111 = torch.aten.add.Tensor %5956, %6110, %int1_225 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6112 = torch.tensor.literal(dense<1.000000e+00> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %6113 = torch.aten.div.Tensor %6111, %6112 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f64> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int2_226 = torch.constant.int 2 loc(#loc1) | |
| %int32_227 = torch.constant.int 32 loc(#loc1) | |
| %int10_228 = torch.constant.int 10 loc(#loc1) | |
| %int4096_229 = torch.constant.int 4096 loc(#loc1) | |
| %6114 = torch.prim.ListConstruct %int2_226, %int32_227, %int10_228, %int4096_229 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6115 = torch.aten.view %6113, %6114 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_230 = torch.constant.int 2 loc(#loc1) | |
| %int3_231 = torch.constant.int 3 loc(#loc1) | |
| %6116 = torch.prim.ListConstruct %int2_230, %int3_231 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_232 = torch.constant.int 0 loc(#loc1) | |
| %true_233 = torch.constant.bool true loc(#loc1) | |
| %6117 = torch.aten.var.correction %6115, %6116, %int0_232, %true_233 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_234 = torch.constant.int 2 loc(#loc1) | |
| %int3_235 = torch.constant.int 3 loc(#loc1) | |
| %6118 = torch.prim.ListConstruct %int2_234, %int3_235 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_236 = torch.constant.bool true loc(#loc1) | |
| %none_237 = torch.constant.none loc(#loc0) | |
| %6119 = torch.aten.mean.dim %6115, %6118, %true_236, %none_237 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %6120 = torch.tensor.literal(dense<9.9999999999999995E-7> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_238 = torch.constant.int 1 loc(#loc1) | |
| %6121 = torch.aten.add.Tensor %6117, %6120, %int1_238 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %6122 = torch.aten.rsqrt %6121 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_239 = torch.constant.int 1 loc(#loc1) | |
| %6123 = torch.aten.sub.Tensor %6115, %6119, %int1_239 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %6124 = torch.aten.mul.Tensor %6123, %6122 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_240 = torch.constant.int 2 loc(#loc1) | |
| %int320_241 = torch.constant.int 320 loc(#loc1) | |
| %int64_242 = torch.constant.int 64 loc(#loc1) | |
| %int64_243 = torch.constant.int 64 loc(#loc1) | |
| %6125 = torch.prim.ListConstruct %int2_240, %int320_241, %int64_242, %int64_243 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6126 = torch.aten.view %6124, %6125 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6127 = torch.aten.detach %5891 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6128 = torch.aten.detach %6127 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_244 = torch.constant.int 2 loc(#loc1) | |
| %int20_245 = torch.constant.int 20 loc(#loc1) | |
| %int16_246 = torch.constant.int 16 loc(#loc1) | |
| %int64_247 = torch.constant.int 64 loc(#loc1) | |
| %int64_248 = torch.constant.int 64 loc(#loc1) | |
| %6129 = torch.prim.ListConstruct %int2_244, %int20_245, %int16_246, %int64_247, %int64_248 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6130 = torch.aten.view %6126, %6129 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %6131 = torch.aten.abs %6130 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_249 = torch.constant.int 2 loc(#loc1) | |
| %true_250 = torch.constant.bool true loc(#loc1) | |
| %values_251, %indices_252 = torch.aten.max.dim %6131, %int2_249, %true_250 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_253 = torch.constant.int 2 loc(#loc1) | |
| %int20_254 = torch.constant.int 20 loc(#loc1) | |
| %int1_255 = torch.constant.int 1 loc(#loc1) | |
| %int64_256 = torch.constant.int 64 loc(#loc1) | |
| %int64_257 = torch.constant.int 64 loc(#loc1) | |
| %6132 = torch.prim.ListConstruct %int2_253, %int20_254, %int1_255, %int64_256, %int64_257 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6133 = torch.aten.view %values_251, %6132 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_258 = torch.constant.int 2 loc(#loc1) | |
| %int20_259 = torch.constant.int 20 loc(#loc1) | |
| %int16_260 = torch.constant.int 16 loc(#loc1) | |
| %int64_261 = torch.constant.int 64 loc(#loc1) | |
| %int64_262 = torch.constant.int 64 loc(#loc1) | |
| %6134 = torch.prim.ListConstruct %int2_258, %int20_259, %int16_260, %int64_261, %int64_262 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_263 = torch.constant.bool false loc(#loc1) | |
| %6135 = torch.aten.expand %6133, %6134, %false_263 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_264 = torch.constant.int 0 loc(#loc1) | |
| %6136 = torch.aten.clone %6135, %int0_264 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_265 = torch.constant.int 2 loc(#loc1) | |
| %int320_266 = torch.constant.int 320 loc(#loc1) | |
| %int64_267 = torch.constant.int 64 loc(#loc1) | |
| %int64_268 = torch.constant.int 64 loc(#loc1) | |
| %6137 = torch.prim.ListConstruct %int2_265, %int320_266, %int64_267, %int64_268 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6138 = torch.aten._unsafe_view %6136, %6137 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6139 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_269 = torch.constant.int 1 loc(#loc1) | |
| %6140 = torch.aten.sub.Tensor %6128, %6139, %int1_269 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6141 = torch.aten.lift_fresh_copy %5890 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6142 = torch.aten.pow.Tensor_Tensor %6141, %6140 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6143 = torch.aten.neg %6142 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6144 = torch.aten.neg %6143 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6145 = torch.aten.div.Tensor %6138, %6144 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6146 = torch.aten.detach %5889 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6147 = torch.aten.detach %6146 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6148 = torch.aten.div.Tensor %6126, %6145 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_270 = torch.constant.int 1 loc(#loc1) | |
| %6149 = torch.aten.add.Tensor %6148, %6147, %int1_270 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6150 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_271 = torch.constant.int 1 loc(#loc1) | |
| %6151 = torch.aten.sub.Tensor %6128, %6150, %int1_271 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6152 = torch.aten.lift_fresh_copy %5888 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6153 = torch.aten.pow.Tensor_Tensor %6152, %6151 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6154 = torch.aten.neg %6153 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6155 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_272 = torch.constant.int 1 loc(#loc1) | |
| %6156 = torch.aten.sub.Tensor %6128, %6155, %int1_272 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6157 = torch.aten.lift_fresh_copy %5887 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6158 = torch.aten.pow.Tensor_Tensor %6157, %6156 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6159 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_273 = torch.constant.int 1 loc(#loc1) | |
| %6160 = torch.aten.sub.Tensor %6158, %6159, %int1_273 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6161 = torch.aten.gt.Tensor %6149, %6160 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6162 = torch.aten.where.self %6161, %6160, %6149 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6163 = torch.aten.lt.Tensor %6162, %6154 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6164 = torch.aten.where.self %6163, %6154, %6162 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6165 = torch.operator "aten.round"(%6164) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_274 = torch.constant.int 1 loc(#loc1) | |
| %6166 = torch.aten.sub.Tensor %6165, %6147, %int1_274 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6167 = torch.aten.mul.Tensor %6166, %6145 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_275 = torch.constant.int 320 loc(#loc1) | |
| %int20_276 = torch.constant.int 20 loc(#loc1) | |
| %int16_277 = torch.constant.int 16 loc(#loc1) | |
| %int1_278 = torch.constant.int 1 loc(#loc1) | |
| %int1_279 = torch.constant.int 1 loc(#loc1) | |
| %6168 = torch.prim.ListConstruct %int320_275, %int20_276, %int16_277, %int1_278, %int1_279 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_280 = torch.constant.bool false loc(#loc1) | |
| %6169 = torch.aten.expand %5886, %6168, %false_280 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,1,1],f16> loc(#loc1) | |
| %int0_281 = torch.constant.int 0 loc(#loc1) | |
| %6170 = torch.aten.clone %6169, %int0_281 : !torch.tensor<[320,20,16,1,1],f16>, !torch.int -> !torch.tensor<[320,20,16,1,1],f16> loc(#loc1) | |
| %int320_282 = torch.constant.int 320 loc(#loc1) | |
| %int320_283 = torch.constant.int 320 loc(#loc1) | |
| %int1_284 = torch.constant.int 1 loc(#loc1) | |
| %int1_285 = torch.constant.int 1 loc(#loc1) | |
| %6171 = torch.prim.ListConstruct %int320_282, %int320_283, %int1_284, %int1_285 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6172 = torch.aten._unsafe_view %6170, %6171 : !torch.tensor<[320,20,16,1,1],f16>, !torch.list<int> -> !torch.tensor<[320,320,1,1],f16> loc(#loc1) | |
| %6173 = torch.aten.mul.Tensor %5885, %6172 : !torch.tensor, !torch.tensor<[320,320,1,1],f16> -> !torch.tensor<[320,320,1,1],f16> loc(#loc1) | |
| %int6_286 = torch.constant.int 6 loc(#loc1) | |
| %none_287 = torch.constant.none loc(#loc0) | |
| %none_288 = torch.constant.none loc(#loc0) | |
| %none_289 = torch.constant.none loc(#loc0) | |
| %false_290 = torch.constant.bool false loc(#loc1) | |
| %none_291 = torch.constant.none loc(#loc0) | |
| %6174 = torch.aten._to_copy %6173, %int6_286, %none_287, %none_288, %none_289, %false_290, %none_291 : !torch.tensor<[320,320,1,1],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,1,1],f32> loc(#loc1) | |
| %int1_292 = torch.constant.int 1 loc(#loc1) | |
| %int1_293 = torch.constant.int 1 loc(#loc1) | |
| %6175 = torch.prim.ListConstruct %int1_292, %int1_293 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_294 = torch.constant.int 0 loc(#loc1) | |
| %int0_295 = torch.constant.int 0 loc(#loc1) | |
| %6176 = torch.prim.ListConstruct %int0_294, %int0_295 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_296 = torch.constant.int 1 loc(#loc1) | |
| %int1_297 = torch.constant.int 1 loc(#loc1) | |
| %6177 = torch.prim.ListConstruct %int1_296, %int1_297 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_298 = torch.constant.bool false loc(#loc1) | |
| %int0_299 = torch.constant.int 0 loc(#loc1) | |
| %int0_300 = torch.constant.int 0 loc(#loc1) | |
| %6178 = torch.prim.ListConstruct %int0_299, %int0_300 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_301 = torch.constant.int 1 loc(#loc1) | |
| %false_302 = torch.constant.bool false loc(#loc1) | |
| %false_303 = torch.constant.bool false loc(#loc1) | |
| %true_304 = torch.constant.bool true loc(#loc1) | |
| %true_305 = torch.constant.bool true loc(#loc1) | |
| %6179 = torch.aten._convolution %6167, %6174, %5884, %6175, %6176, %6177, %false_298, %6178, %int1_301, %false_302, %false_303, %true_304, %true_305 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,1,1],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int0_306 = torch.constant.int 0 loc(#loc1) | |
| %int2_307 = torch.constant.int 2 loc(#loc1) | |
| %int3_308 = torch.constant.int 3 loc(#loc1) | |
| %int1_309 = torch.constant.int 1 loc(#loc1) | |
| %6180 = torch.prim.ListConstruct %int0_306, %int2_307, %int3_308, %int1_309 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6181 = torch.aten.permute %6179, %6180 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,64,64,320],f32> loc(#loc1) | |
| %int2_310 = torch.constant.int 2 loc(#loc1) | |
| %int4096_311 = torch.constant.int 4096 loc(#loc1) | |
| %int320_312 = torch.constant.int 320 loc(#loc1) | |
| %6182 = torch.prim.ListConstruct %int2_310, %int4096_311, %int320_312 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720 = torch.constant.int 1310720 loc(#loc1) | |
| %int1_313 = torch.constant.int 1 loc(#loc1) | |
| %int4096_314 = torch.constant.int 4096 loc(#loc1) | |
| %6183 = torch.prim.ListConstruct %int1310720, %int1_313, %int4096_314 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6184 = torch.aten._reshape_alias %6181, %6182, %6183 : !torch.tensor<[2,64,64,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_315 = torch.constant.int 320 loc(#loc1) | |
| %6185 = torch.prim.ListConstruct %int320_315 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_316 = torch.constant.none loc(#loc0) | |
| %none_317 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0, %result1, %result2 = torch.aten.native_layer_norm %6184, %6185, %none_316, %none_317, %float1.000000e-05 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,1],f32>, !torch.tensor<[2,4096,1],f32> loc(#loc1) | |
| %6186 = torch.aten.detach %5883 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6187 = torch.aten.detach %6186 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_318 = torch.constant.int 2 loc(#loc1) | |
| %int4096_319 = torch.constant.int 4096 loc(#loc1) | |
| %int20_320 = torch.constant.int 20 loc(#loc1) | |
| %int16_321 = torch.constant.int 16 loc(#loc1) | |
| %6188 = torch.prim.ListConstruct %int2_318, %int4096_319, %int20_320, %int16_321 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6189 = torch.aten.view %result0, %6188 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %6190 = torch.aten.abs %6189 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_322 = torch.constant.int 3 loc(#loc1) | |
| %true_323 = torch.constant.bool true loc(#loc1) | |
| %values_324, %indices_325 = torch.aten.max.dim %6190, %int3_322, %true_323 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_326 = torch.constant.int 2 loc(#loc1) | |
| %int4096_327 = torch.constant.int 4096 loc(#loc1) | |
| %int20_328 = torch.constant.int 20 loc(#loc1) | |
| %int1_329 = torch.constant.int 1 loc(#loc1) | |
| %6191 = torch.prim.ListConstruct %int2_326, %int4096_327, %int20_328, %int1_329 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6192 = torch.aten.view %values_324, %6191 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_330 = torch.constant.int 2 loc(#loc1) | |
| %int4096_331 = torch.constant.int 4096 loc(#loc1) | |
| %int20_332 = torch.constant.int 20 loc(#loc1) | |
| %int16_333 = torch.constant.int 16 loc(#loc1) | |
| %6193 = torch.prim.ListConstruct %int2_330, %int4096_331, %int20_332, %int16_333 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_334 = torch.constant.bool false loc(#loc1) | |
| %6194 = torch.aten.expand %6192, %6193, %false_334 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_335 = torch.constant.int 0 loc(#loc1) | |
| %6195 = torch.aten.clone %6194, %int0_335 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_336 = torch.constant.int 2 loc(#loc1) | |
| %int4096_337 = torch.constant.int 4096 loc(#loc1) | |
| %int320_338 = torch.constant.int 320 loc(#loc1) | |
| %6196 = torch.prim.ListConstruct %int2_336, %int4096_337, %int320_338 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6197 = torch.aten._unsafe_view %6195, %6196 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6198 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_339 = torch.constant.int 1 loc(#loc1) | |
| %6199 = torch.aten.sub.Tensor %6187, %6198, %int1_339 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6200 = torch.aten.lift_fresh_copy %5882 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6201 = torch.aten.pow.Tensor_Tensor %6200, %6199 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6202 = torch.aten.neg %6201 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6203 = torch.aten.neg %6202 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6204 = torch.aten.div.Tensor %6197, %6203 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6205 = torch.aten.detach %5881 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6206 = torch.aten.detach %6205 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6207 = torch.aten.div.Tensor %result0, %6204 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_340 = torch.constant.int 1 loc(#loc1) | |
| %6208 = torch.aten.add.Tensor %6207, %6206, %int1_340 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6209 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_341 = torch.constant.int 1 loc(#loc1) | |
| %6210 = torch.aten.sub.Tensor %6187, %6209, %int1_341 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6211 = torch.aten.lift_fresh_copy %5880 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6212 = torch.aten.pow.Tensor_Tensor %6211, %6210 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6213 = torch.aten.neg %6212 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6214 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_342 = torch.constant.int 1 loc(#loc1) | |
| %6215 = torch.aten.sub.Tensor %6187, %6214, %int1_342 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6216 = torch.aten.lift_fresh_copy %5879 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6217 = torch.aten.pow.Tensor_Tensor %6216, %6215 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6218 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_343 = torch.constant.int 1 loc(#loc1) | |
| %6219 = torch.aten.sub.Tensor %6217, %6218, %int1_343 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6220 = torch.aten.gt.Tensor %6208, %6219 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6221 = torch.aten.where.self %6220, %6219, %6208 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6222 = torch.aten.lt.Tensor %6221, %6213 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6223 = torch.aten.where.self %6222, %6213, %6221 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6224 = torch.operator "aten.round"(%6223) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_344 = torch.constant.int 1 loc(#loc1) | |
| %6225 = torch.aten.sub.Tensor %6224, %6206, %int1_344 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6226 = torch.aten.mul.Tensor %6225, %6204 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_345 = torch.constant.int 320 loc(#loc1) | |
| %int20_346 = torch.constant.int 20 loc(#loc1) | |
| %int16_347 = torch.constant.int 16 loc(#loc1) | |
| %6227 = torch.prim.ListConstruct %int320_345, %int20_346, %int16_347 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_348 = torch.constant.bool false loc(#loc1) | |
| %6228 = torch.aten.expand %5878, %6227, %false_348 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_349 = torch.constant.int 0 loc(#loc1) | |
| %6229 = torch.aten.clone %6228, %int0_349 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_350 = torch.constant.int 320 loc(#loc1) | |
| %int320_351 = torch.constant.int 320 loc(#loc1) | |
| %6230 = torch.prim.ListConstruct %int320_350, %int320_351 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6231 = torch.aten._unsafe_view %6229, %6230 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %6232 = torch.aten.mul.Tensor %5877, %6231 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_352 = torch.constant.int 6 loc(#loc1) | |
| %none_353 = torch.constant.none loc(#loc0) | |
| %none_354 = torch.constant.none loc(#loc0) | |
| %none_355 = torch.constant.none loc(#loc0) | |
| %false_356 = torch.constant.bool false loc(#loc1) | |
| %none_357 = torch.constant.none loc(#loc0) | |
| %6233 = torch.aten._to_copy %6232, %int6_352, %none_353, %none_354, %none_355, %false_356, %none_357 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %6234 = torch.aten.t %6233 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192 = torch.constant.int 8192 loc(#loc1) | |
| %int320_358 = torch.constant.int 320 loc(#loc1) | |
| %6235 = torch.prim.ListConstruct %int8192, %int320_358 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6236 = torch.aten.view %6226, %6235 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_359 = torch.constant.int 1 loc(#loc1) | |
| %int1_360 = torch.constant.int 1 loc(#loc1) | |
| %6237 = torch.aten.addmm %5876, %6236, %6234, %int1_359, %int1_360 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_361 = torch.constant.int 2 loc(#loc1) | |
| %int4096_362 = torch.constant.int 4096 loc(#loc1) | |
| %int320_363 = torch.constant.int 320 loc(#loc1) | |
| %6238 = torch.prim.ListConstruct %int2_361, %int4096_362, %int320_363 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6239 = torch.aten.view %6237, %6238 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6240 = torch.aten.detach %5875 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6241 = torch.aten.detach %6240 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_364 = torch.constant.int 2 loc(#loc1) | |
| %int4096_365 = torch.constant.int 4096 loc(#loc1) | |
| %int20_366 = torch.constant.int 20 loc(#loc1) | |
| %int16_367 = torch.constant.int 16 loc(#loc1) | |
| %6242 = torch.prim.ListConstruct %int2_364, %int4096_365, %int20_366, %int16_367 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6243 = torch.aten.view %result0, %6242 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %6244 = torch.aten.abs %6243 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_368 = torch.constant.int 3 loc(#loc1) | |
| %true_369 = torch.constant.bool true loc(#loc1) | |
| %values_370, %indices_371 = torch.aten.max.dim %6244, %int3_368, %true_369 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_372 = torch.constant.int 2 loc(#loc1) | |
| %int4096_373 = torch.constant.int 4096 loc(#loc1) | |
| %int20_374 = torch.constant.int 20 loc(#loc1) | |
| %int1_375 = torch.constant.int 1 loc(#loc1) | |
| %6245 = torch.prim.ListConstruct %int2_372, %int4096_373, %int20_374, %int1_375 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6246 = torch.aten.view %values_370, %6245 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_376 = torch.constant.int 2 loc(#loc1) | |
| %int4096_377 = torch.constant.int 4096 loc(#loc1) | |
| %int20_378 = torch.constant.int 20 loc(#loc1) | |
| %int16_379 = torch.constant.int 16 loc(#loc1) | |
| %6247 = torch.prim.ListConstruct %int2_376, %int4096_377, %int20_378, %int16_379 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_380 = torch.constant.bool false loc(#loc1) | |
| %6248 = torch.aten.expand %6246, %6247, %false_380 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_381 = torch.constant.int 0 loc(#loc1) | |
| %6249 = torch.aten.clone %6248, %int0_381 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_382 = torch.constant.int 2 loc(#loc1) | |
| %int4096_383 = torch.constant.int 4096 loc(#loc1) | |
| %int320_384 = torch.constant.int 320 loc(#loc1) | |
| %6250 = torch.prim.ListConstruct %int2_382, %int4096_383, %int320_384 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6251 = torch.aten._unsafe_view %6249, %6250 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6252 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_385 = torch.constant.int 1 loc(#loc1) | |
| %6253 = torch.aten.sub.Tensor %6241, %6252, %int1_385 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6254 = torch.aten.lift_fresh_copy %5874 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6255 = torch.aten.pow.Tensor_Tensor %6254, %6253 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6256 = torch.aten.neg %6255 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6257 = torch.aten.neg %6256 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6258 = torch.aten.div.Tensor %6251, %6257 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6259 = torch.aten.detach %5873 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6260 = torch.aten.detach %6259 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6261 = torch.aten.div.Tensor %result0, %6258 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_386 = torch.constant.int 1 loc(#loc1) | |
| %6262 = torch.aten.add.Tensor %6261, %6260, %int1_386 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6263 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_387 = torch.constant.int 1 loc(#loc1) | |
| %6264 = torch.aten.sub.Tensor %6241, %6263, %int1_387 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6265 = torch.aten.lift_fresh_copy %5872 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6266 = torch.aten.pow.Tensor_Tensor %6265, %6264 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6267 = torch.aten.neg %6266 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6268 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_388 = torch.constant.int 1 loc(#loc1) | |
| %6269 = torch.aten.sub.Tensor %6241, %6268, %int1_388 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6270 = torch.aten.lift_fresh_copy %5871 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6271 = torch.aten.pow.Tensor_Tensor %6270, %6269 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6272 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_389 = torch.constant.int 1 loc(#loc1) | |
| %6273 = torch.aten.sub.Tensor %6271, %6272, %int1_389 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6274 = torch.aten.gt.Tensor %6262, %6273 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6275 = torch.aten.where.self %6274, %6273, %6262 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6276 = torch.aten.lt.Tensor %6275, %6267 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6277 = torch.aten.where.self %6276, %6267, %6275 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6278 = torch.operator "aten.round"(%6277) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_390 = torch.constant.int 1 loc(#loc1) | |
| %6279 = torch.aten.sub.Tensor %6278, %6260, %int1_390 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6280 = torch.aten.mul.Tensor %6279, %6258 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_391 = torch.constant.int 320 loc(#loc1) | |
| %int20_392 = torch.constant.int 20 loc(#loc1) | |
| %int16_393 = torch.constant.int 16 loc(#loc1) | |
| %6281 = torch.prim.ListConstruct %int320_391, %int20_392, %int16_393 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_394 = torch.constant.bool false loc(#loc1) | |
| %6282 = torch.aten.expand %5870, %6281, %false_394 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_395 = torch.constant.int 0 loc(#loc1) | |
| %6283 = torch.aten.clone %6282, %int0_395 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_396 = torch.constant.int 320 loc(#loc1) | |
| %int320_397 = torch.constant.int 320 loc(#loc1) | |
| %6284 = torch.prim.ListConstruct %int320_396, %int320_397 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6285 = torch.aten._unsafe_view %6283, %6284 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %6286 = torch.aten.mul.Tensor %5869, %6285 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_398 = torch.constant.int 6 loc(#loc1) | |
| %none_399 = torch.constant.none loc(#loc0) | |
| %none_400 = torch.constant.none loc(#loc0) | |
| %none_401 = torch.constant.none loc(#loc0) | |
| %false_402 = torch.constant.bool false loc(#loc1) | |
| %none_403 = torch.constant.none loc(#loc0) | |
| %6287 = torch.aten._to_copy %6286, %int6_398, %none_399, %none_400, %none_401, %false_402, %none_403 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %6288 = torch.aten.t %6287 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_404 = torch.constant.int 8192 loc(#loc1) | |
| %int320_405 = torch.constant.int 320 loc(#loc1) | |
| %6289 = torch.prim.ListConstruct %int8192_404, %int320_405 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6290 = torch.aten.view %6280, %6289 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_406 = torch.constant.int 1 loc(#loc1) | |
| %int1_407 = torch.constant.int 1 loc(#loc1) | |
| %6291 = torch.aten.addmm %5868, %6290, %6288, %int1_406, %int1_407 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_408 = torch.constant.int 2 loc(#loc1) | |
| %int4096_409 = torch.constant.int 4096 loc(#loc1) | |
| %int320_410 = torch.constant.int 320 loc(#loc1) | |
| %6292 = torch.prim.ListConstruct %int2_408, %int4096_409, %int320_410 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6293 = torch.aten.view %6291, %6292 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6294 = torch.aten.detach %5867 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6295 = torch.aten.detach %6294 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_411 = torch.constant.int 2 loc(#loc1) | |
| %int4096_412 = torch.constant.int 4096 loc(#loc1) | |
| %int20_413 = torch.constant.int 20 loc(#loc1) | |
| %int16_414 = torch.constant.int 16 loc(#loc1) | |
| %6296 = torch.prim.ListConstruct %int2_411, %int4096_412, %int20_413, %int16_414 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6297 = torch.aten.view %result0, %6296 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %6298 = torch.aten.abs %6297 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_415 = torch.constant.int 3 loc(#loc1) | |
| %true_416 = torch.constant.bool true loc(#loc1) | |
| %values_417, %indices_418 = torch.aten.max.dim %6298, %int3_415, %true_416 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_419 = torch.constant.int 2 loc(#loc1) | |
| %int4096_420 = torch.constant.int 4096 loc(#loc1) | |
| %int20_421 = torch.constant.int 20 loc(#loc1) | |
| %int1_422 = torch.constant.int 1 loc(#loc1) | |
| %6299 = torch.prim.ListConstruct %int2_419, %int4096_420, %int20_421, %int1_422 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6300 = torch.aten.view %values_417, %6299 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_423 = torch.constant.int 2 loc(#loc1) | |
| %int4096_424 = torch.constant.int 4096 loc(#loc1) | |
| %int20_425 = torch.constant.int 20 loc(#loc1) | |
| %int16_426 = torch.constant.int 16 loc(#loc1) | |
| %6301 = torch.prim.ListConstruct %int2_423, %int4096_424, %int20_425, %int16_426 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_427 = torch.constant.bool false loc(#loc1) | |
| %6302 = torch.aten.expand %6300, %6301, %false_427 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_428 = torch.constant.int 0 loc(#loc1) | |
| %6303 = torch.aten.clone %6302, %int0_428 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_429 = torch.constant.int 2 loc(#loc1) | |
| %int4096_430 = torch.constant.int 4096 loc(#loc1) | |
| %int320_431 = torch.constant.int 320 loc(#loc1) | |
| %6304 = torch.prim.ListConstruct %int2_429, %int4096_430, %int320_431 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6305 = torch.aten._unsafe_view %6303, %6304 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6306 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_432 = torch.constant.int 1 loc(#loc1) | |
| %6307 = torch.aten.sub.Tensor %6295, %6306, %int1_432 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6308 = torch.aten.lift_fresh_copy %5866 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6309 = torch.aten.pow.Tensor_Tensor %6308, %6307 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6310 = torch.aten.neg %6309 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6311 = torch.aten.neg %6310 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6312 = torch.aten.div.Tensor %6305, %6311 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6313 = torch.aten.detach %5865 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6314 = torch.aten.detach %6313 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6315 = torch.aten.div.Tensor %result0, %6312 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_433 = torch.constant.int 1 loc(#loc1) | |
| %6316 = torch.aten.add.Tensor %6315, %6314, %int1_433 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6317 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_434 = torch.constant.int 1 loc(#loc1) | |
| %6318 = torch.aten.sub.Tensor %6295, %6317, %int1_434 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6319 = torch.aten.lift_fresh_copy %5864 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6320 = torch.aten.pow.Tensor_Tensor %6319, %6318 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6321 = torch.aten.neg %6320 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6322 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_435 = torch.constant.int 1 loc(#loc1) | |
| %6323 = torch.aten.sub.Tensor %6295, %6322, %int1_435 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6324 = torch.aten.lift_fresh_copy %5863 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6325 = torch.aten.pow.Tensor_Tensor %6324, %6323 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6326 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_436 = torch.constant.int 1 loc(#loc1) | |
| %6327 = torch.aten.sub.Tensor %6325, %6326, %int1_436 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6328 = torch.aten.gt.Tensor %6316, %6327 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6329 = torch.aten.where.self %6328, %6327, %6316 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6330 = torch.aten.lt.Tensor %6329, %6321 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6331 = torch.aten.where.self %6330, %6321, %6329 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6332 = torch.operator "aten.round"(%6331) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_437 = torch.constant.int 1 loc(#loc1) | |
| %6333 = torch.aten.sub.Tensor %6332, %6314, %int1_437 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6334 = torch.aten.mul.Tensor %6333, %6312 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_438 = torch.constant.int 320 loc(#loc1) | |
| %int20_439 = torch.constant.int 20 loc(#loc1) | |
| %int16_440 = torch.constant.int 16 loc(#loc1) | |
| %6335 = torch.prim.ListConstruct %int320_438, %int20_439, %int16_440 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_441 = torch.constant.bool false loc(#loc1) | |
| %6336 = torch.aten.expand %5862, %6335, %false_441 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_442 = torch.constant.int 0 loc(#loc1) | |
| %6337 = torch.aten.clone %6336, %int0_442 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_443 = torch.constant.int 320 loc(#loc1) | |
| %int320_444 = torch.constant.int 320 loc(#loc1) | |
| %6338 = torch.prim.ListConstruct %int320_443, %int320_444 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6339 = torch.aten._unsafe_view %6337, %6338 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %6340 = torch.aten.mul.Tensor %5861, %6339 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_445 = torch.constant.int 6 loc(#loc1) | |
| %none_446 = torch.constant.none loc(#loc0) | |
| %none_447 = torch.constant.none loc(#loc0) | |
| %none_448 = torch.constant.none loc(#loc0) | |
| %false_449 = torch.constant.bool false loc(#loc1) | |
| %none_450 = torch.constant.none loc(#loc0) | |
| %6341 = torch.aten._to_copy %6340, %int6_445, %none_446, %none_447, %none_448, %false_449, %none_450 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %6342 = torch.aten.t %6341 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_451 = torch.constant.int 8192 loc(#loc1) | |
| %int320_452 = torch.constant.int 320 loc(#loc1) | |
| %6343 = torch.prim.ListConstruct %int8192_451, %int320_452 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6344 = torch.aten.view %6334, %6343 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_453 = torch.constant.int 1 loc(#loc1) | |
| %int1_454 = torch.constant.int 1 loc(#loc1) | |
| %6345 = torch.aten.addmm %5860, %6344, %6342, %int1_453, %int1_454 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_455 = torch.constant.int 2 loc(#loc1) | |
| %int4096_456 = torch.constant.int 4096 loc(#loc1) | |
| %int320_457 = torch.constant.int 320 loc(#loc1) | |
| %6346 = torch.prim.ListConstruct %int2_455, %int4096_456, %int320_457 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6347 = torch.aten.view %6345, %6346 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int2_458 = torch.constant.int 2 loc(#loc1) | |
| %int4096_459 = torch.constant.int 4096 loc(#loc1) | |
| %int8 = torch.constant.int 8 loc(#loc1) | |
| %int40 = torch.constant.int 40 loc(#loc1) | |
| %6348 = torch.prim.ListConstruct %int2_458, %int4096_459, %int8, %int40 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_460 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_461 = torch.constant.int 320 loc(#loc1) | |
| %int40_462 = torch.constant.int 40 loc(#loc1) | |
| %int1_463 = torch.constant.int 1 loc(#loc1) | |
| %6349 = torch.prim.ListConstruct %int1310720_460, %int320_461, %int40_462, %int1_463 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6350 = torch.aten._reshape_alias %6239, %6348, %6349 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_464 = torch.constant.int 0 loc(#loc1) | |
| %int2_465 = torch.constant.int 2 loc(#loc1) | |
| %int1_466 = torch.constant.int 1 loc(#loc1) | |
| %int3_467 = torch.constant.int 3 loc(#loc1) | |
| %6351 = torch.prim.ListConstruct %int0_464, %int2_465, %int1_466, %int3_467 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6352 = torch.aten.permute %6350, %6351 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_468 = torch.constant.int 0 loc(#loc1) | |
| %6353 = torch.aten.clone %6352, %int0_468 : !torch.tensor<[2,8,4096,40],f32>, !torch.int -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int16_469 = torch.constant.int 16 loc(#loc1) | |
| %int4096_470 = torch.constant.int 4096 loc(#loc1) | |
| %int40_471 = torch.constant.int 40 loc(#loc1) | |
| %6354 = torch.prim.ListConstruct %int16_469, %int4096_470, %int40_471 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6355 = torch.aten._unsafe_view %6353, %6354 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_472 = torch.constant.int 2 loc(#loc1) | |
| %int4096_473 = torch.constant.int 4096 loc(#loc1) | |
| %int8_474 = torch.constant.int 8 loc(#loc1) | |
| %int40_475 = torch.constant.int 40 loc(#loc1) | |
| %6356 = torch.prim.ListConstruct %int2_472, %int4096_473, %int8_474, %int40_475 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_476 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_477 = torch.constant.int 320 loc(#loc1) | |
| %int40_478 = torch.constant.int 40 loc(#loc1) | |
| %int1_479 = torch.constant.int 1 loc(#loc1) | |
| %6357 = torch.prim.ListConstruct %int1310720_476, %int320_477, %int40_478, %int1_479 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6358 = torch.aten._reshape_alias %6293, %6356, %6357 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_480 = torch.constant.int 0 loc(#loc1) | |
| %int2_481 = torch.constant.int 2 loc(#loc1) | |
| %int1_482 = torch.constant.int 1 loc(#loc1) | |
| %int3_483 = torch.constant.int 3 loc(#loc1) | |
| %6359 = torch.prim.ListConstruct %int0_480, %int2_481, %int1_482, %int3_483 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6360 = torch.aten.permute %6358, %6359 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_484 = torch.constant.int 0 loc(#loc1) | |
| %6361 = torch.aten.clone %6360, %int0_484 : !torch.tensor<[2,8,4096,40],f32>, !torch.int -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int16_485 = torch.constant.int 16 loc(#loc1) | |
| %int4096_486 = torch.constant.int 4096 loc(#loc1) | |
| %int40_487 = torch.constant.int 40 loc(#loc1) | |
| %6362 = torch.prim.ListConstruct %int16_485, %int4096_486, %int40_487 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6363 = torch.aten._unsafe_view %6361, %6362 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_488 = torch.constant.int 2 loc(#loc1) | |
| %int4096_489 = torch.constant.int 4096 loc(#loc1) | |
| %int8_490 = torch.constant.int 8 loc(#loc1) | |
| %int40_491 = torch.constant.int 40 loc(#loc1) | |
| %6364 = torch.prim.ListConstruct %int2_488, %int4096_489, %int8_490, %int40_491 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_492 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_493 = torch.constant.int 320 loc(#loc1) | |
| %int40_494 = torch.constant.int 40 loc(#loc1) | |
| %int1_495 = torch.constant.int 1 loc(#loc1) | |
| %6365 = torch.prim.ListConstruct %int1310720_492, %int320_493, %int40_494, %int1_495 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6366 = torch.aten._reshape_alias %6347, %6364, %6365 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_496 = torch.constant.int 0 loc(#loc1) | |
| %int2_497 = torch.constant.int 2 loc(#loc1) | |
| %int1_498 = torch.constant.int 1 loc(#loc1) | |
| %int3_499 = torch.constant.int 3 loc(#loc1) | |
| %6367 = torch.prim.ListConstruct %int0_496, %int2_497, %int1_498, %int3_499 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6368 = torch.aten.permute %6366, %6367 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_500 = torch.constant.int 0 loc(#loc1) | |
| %6369 = torch.aten.clone %6368, %int0_500 : !torch.tensor<[2,8,4096,40],f32>, !torch.int -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int16_501 = torch.constant.int 16 loc(#loc1) | |
| %int4096_502 = torch.constant.int 4096 loc(#loc1) | |
| %int40_503 = torch.constant.int 40 loc(#loc1) | |
| %6370 = torch.prim.ListConstruct %int16_501, %int4096_502, %int40_503 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6371 = torch.aten._unsafe_view %6369, %6370 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int-1_504 = torch.constant.int -1 loc(#loc1) | |
| %int-2 = torch.constant.int -2 loc(#loc1) | |
| %6372 = torch.aten.transpose.int %6363, %int-1_504, %int-2 : !torch.tensor<[16,4096,40],f32>, !torch.int, !torch.int -> !torch.tensor<[16,40,4096],f32> loc(#loc1) | |
| %int16_505 = torch.constant.int 16 loc(#loc1) | |
| %int4096_506 = torch.constant.int 4096 loc(#loc1) | |
| %int40_507 = torch.constant.int 40 loc(#loc1) | |
| %6373 = torch.prim.ListConstruct %int16_505, %int4096_506, %int40_507 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_508 = torch.constant.bool false loc(#loc1) | |
| %6374 = torch.aten.expand %6355, %6373, %false_508 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_509 = torch.constant.int 16 loc(#loc1) | |
| %int4096_510 = torch.constant.int 4096 loc(#loc1) | |
| %int40_511 = torch.constant.int 40 loc(#loc1) | |
| %6375 = torch.prim.ListConstruct %int16_509, %int4096_510, %int40_511 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int163840 = torch.constant.int 163840 loc(#loc1) | |
| %int40_512 = torch.constant.int 40 loc(#loc1) | |
| %int1_513 = torch.constant.int 1 loc(#loc1) | |
| %6376 = torch.prim.ListConstruct %int163840, %int40_512, %int1_513 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6377 = torch.aten._reshape_alias %6374, %6375, %6376 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_514 = torch.constant.int 16 loc(#loc1) | |
| %int40_515 = torch.constant.int 40 loc(#loc1) | |
| %int4096_516 = torch.constant.int 4096 loc(#loc1) | |
| %6378 = torch.prim.ListConstruct %int16_514, %int40_515, %int4096_516 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_517 = torch.constant.bool false loc(#loc1) | |
| %6379 = torch.aten.expand %6372, %6378, %false_517 : !torch.tensor<[16,40,4096],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,40,4096],f32> loc(#loc1) | |
| %int16_518 = torch.constant.int 16 loc(#loc1) | |
| %int40_519 = torch.constant.int 40 loc(#loc1) | |
| %int4096_520 = torch.constant.int 4096 loc(#loc1) | |
| %6380 = torch.prim.ListConstruct %int16_518, %int40_519, %int4096_520 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int163840_521 = torch.constant.int 163840 loc(#loc1) | |
| %int1_522 = torch.constant.int 1 loc(#loc1) | |
| %int40_523 = torch.constant.int 40 loc(#loc1) | |
| %6381 = torch.prim.ListConstruct %int163840_521, %int1_522, %int40_523 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6382 = torch.aten._reshape_alias %6379, %6380, %6381 : !torch.tensor<[16,40,4096],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,40,4096],f32> loc(#loc1) | |
| %6383 = torch.aten.bmm %6377, %6382 : !torch.tensor<[16,4096,40],f32>, !torch.tensor<[16,40,4096],f32> -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int16_524 = torch.constant.int 16 loc(#loc1) | |
| %int4096_525 = torch.constant.int 4096 loc(#loc1) | |
| %int4096_526 = torch.constant.int 4096 loc(#loc1) | |
| %6384 = torch.prim.ListConstruct %int16_524, %int4096_525, %int4096_526 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6385 = torch.aten._unsafe_view %6383, %6384 : !torch.tensor<[16,4096,4096],f32>, !torch.list<int> -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %6386 = torch.tensor.literal(dense<0.15811388300841897> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %6387 = torch.aten.mul.Tensor %6385, %6386 : !torch.tensor<[16,4096,4096],f32>, !torch.tensor<[],f64> -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int-1_527 = torch.constant.int -1 loc(#loc1) | |
| %false_528 = torch.constant.bool false loc(#loc1) | |
| %6388 = torch.aten._softmax %6387, %int-1_527, %false_528 : !torch.tensor<[16,4096,4096],f32>, !torch.int, !torch.bool -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int16_529 = torch.constant.int 16 loc(#loc1) | |
| %int4096_530 = torch.constant.int 4096 loc(#loc1) | |
| %int4096_531 = torch.constant.int 4096 loc(#loc1) | |
| %6389 = torch.prim.ListConstruct %int16_529, %int4096_530, %int4096_531 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_532 = torch.constant.bool false loc(#loc1) | |
| %6390 = torch.aten.expand %6388, %6389, %false_532 : !torch.tensor<[16,4096,4096],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int16_533 = torch.constant.int 16 loc(#loc1) | |
| %int4096_534 = torch.constant.int 4096 loc(#loc1) | |
| %int4096_535 = torch.constant.int 4096 loc(#loc1) | |
| %6391 = torch.prim.ListConstruct %int16_533, %int4096_534, %int4096_535 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int16777216 = torch.constant.int 16777216 loc(#loc1) | |
| %int4096_536 = torch.constant.int 4096 loc(#loc1) | |
| %int1_537 = torch.constant.int 1 loc(#loc1) | |
| %6392 = torch.prim.ListConstruct %int16777216, %int4096_536, %int1_537 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6393 = torch.aten._reshape_alias %6390, %6391, %6392 : !torch.tensor<[16,4096,4096],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int16_538 = torch.constant.int 16 loc(#loc1) | |
| %int4096_539 = torch.constant.int 4096 loc(#loc1) | |
| %int40_540 = torch.constant.int 40 loc(#loc1) | |
| %6394 = torch.prim.ListConstruct %int16_538, %int4096_539, %int40_540 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_541 = torch.constant.bool false loc(#loc1) | |
| %6395 = torch.aten.expand %6371, %6394, %false_541 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_542 = torch.constant.int 16 loc(#loc1) | |
| %int4096_543 = torch.constant.int 4096 loc(#loc1) | |
| %int40_544 = torch.constant.int 40 loc(#loc1) | |
| %6396 = torch.prim.ListConstruct %int16_542, %int4096_543, %int40_544 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int163840_545 = torch.constant.int 163840 loc(#loc1) | |
| %int40_546 = torch.constant.int 40 loc(#loc1) | |
| %int1_547 = torch.constant.int 1 loc(#loc1) | |
| %6397 = torch.prim.ListConstruct %int163840_545, %int40_546, %int1_547 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6398 = torch.aten._reshape_alias %6395, %6396, %6397 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %6399 = torch.aten.bmm %6393, %6398 : !torch.tensor<[16,4096,4096],f32>, !torch.tensor<[16,4096,40],f32> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_548 = torch.constant.int 16 loc(#loc1) | |
| %int4096_549 = torch.constant.int 4096 loc(#loc1) | |
| %int40_550 = torch.constant.int 40 loc(#loc1) | |
| %6400 = torch.prim.ListConstruct %int16_548, %int4096_549, %int40_550 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6401 = torch.aten._unsafe_view %6399, %6400 : !torch.tensor<[16,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_551 = torch.constant.int 2 loc(#loc1) | |
| %int8_552 = torch.constant.int 8 loc(#loc1) | |
| %int4096_553 = torch.constant.int 4096 loc(#loc1) | |
| %int40_554 = torch.constant.int 40 loc(#loc1) | |
| %6402 = torch.prim.ListConstruct %int2_551, %int8_552, %int4096_553, %int40_554 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_555 = torch.constant.int 1310720 loc(#loc1) | |
| %int163840_556 = torch.constant.int 163840 loc(#loc1) | |
| %int40_557 = torch.constant.int 40 loc(#loc1) | |
| %int1_558 = torch.constant.int 1 loc(#loc1) | |
| %6403 = torch.prim.ListConstruct %int1310720_555, %int163840_556, %int40_557, %int1_558 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6404 = torch.aten._reshape_alias %6401, %6402, %6403 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_559 = torch.constant.int 0 loc(#loc1) | |
| %int2_560 = torch.constant.int 2 loc(#loc1) | |
| %int1_561 = torch.constant.int 1 loc(#loc1) | |
| %int3_562 = torch.constant.int 3 loc(#loc1) | |
| %6405 = torch.prim.ListConstruct %int0_559, %int2_560, %int1_561, %int3_562 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6406 = torch.aten.permute %6404, %6405 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_563 = torch.constant.int 0 loc(#loc1) | |
| %6407 = torch.aten.clone %6406, %int0_563 : !torch.tensor<[2,4096,8,40],f32>, !torch.int -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int2_564 = torch.constant.int 2 loc(#loc1) | |
| %int4096_565 = torch.constant.int 4096 loc(#loc1) | |
| %int320_566 = torch.constant.int 320 loc(#loc1) | |
| %6408 = torch.prim.ListConstruct %int2_564, %int4096_565, %int320_566 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6409 = torch.aten._unsafe_view %6407, %6408 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6410 = torch.aten.detach %5859 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6411 = torch.aten.detach %6410 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_567 = torch.constant.int 2 loc(#loc1) | |
| %int4096_568 = torch.constant.int 4096 loc(#loc1) | |
| %int20_569 = torch.constant.int 20 loc(#loc1) | |
| %int16_570 = torch.constant.int 16 loc(#loc1) | |
| %6412 = torch.prim.ListConstruct %int2_567, %int4096_568, %int20_569, %int16_570 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6413 = torch.aten.view %6409, %6412 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %6414 = torch.aten.abs %6413 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_571 = torch.constant.int 3 loc(#loc1) | |
| %true_572 = torch.constant.bool true loc(#loc1) | |
| %values_573, %indices_574 = torch.aten.max.dim %6414, %int3_571, %true_572 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_575 = torch.constant.int 2 loc(#loc1) | |
| %int4096_576 = torch.constant.int 4096 loc(#loc1) | |
| %int20_577 = torch.constant.int 20 loc(#loc1) | |
| %int1_578 = torch.constant.int 1 loc(#loc1) | |
| %6415 = torch.prim.ListConstruct %int2_575, %int4096_576, %int20_577, %int1_578 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6416 = torch.aten.view %values_573, %6415 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_579 = torch.constant.int 2 loc(#loc1) | |
| %int4096_580 = torch.constant.int 4096 loc(#loc1) | |
| %int20_581 = torch.constant.int 20 loc(#loc1) | |
| %int16_582 = torch.constant.int 16 loc(#loc1) | |
| %6417 = torch.prim.ListConstruct %int2_579, %int4096_580, %int20_581, %int16_582 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_583 = torch.constant.bool false loc(#loc1) | |
| %6418 = torch.aten.expand %6416, %6417, %false_583 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_584 = torch.constant.int 0 loc(#loc1) | |
| %6419 = torch.aten.clone %6418, %int0_584 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_585 = torch.constant.int 2 loc(#loc1) | |
| %int4096_586 = torch.constant.int 4096 loc(#loc1) | |
| %int320_587 = torch.constant.int 320 loc(#loc1) | |
| %6420 = torch.prim.ListConstruct %int2_585, %int4096_586, %int320_587 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6421 = torch.aten._unsafe_view %6419, %6420 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6422 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_588 = torch.constant.int 1 loc(#loc1) | |
| %6423 = torch.aten.sub.Tensor %6411, %6422, %int1_588 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6424 = torch.aten.lift_fresh_copy %5858 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6425 = torch.aten.pow.Tensor_Tensor %6424, %6423 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6426 = torch.aten.neg %6425 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6427 = torch.aten.neg %6426 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6428 = torch.aten.div.Tensor %6421, %6427 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6429 = torch.aten.detach %5857 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6430 = torch.aten.detach %6429 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6431 = torch.aten.div.Tensor %6409, %6428 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_589 = torch.constant.int 1 loc(#loc1) | |
| %6432 = torch.aten.add.Tensor %6431, %6430, %int1_589 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6433 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_590 = torch.constant.int 1 loc(#loc1) | |
| %6434 = torch.aten.sub.Tensor %6411, %6433, %int1_590 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6435 = torch.aten.lift_fresh_copy %5856 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6436 = torch.aten.pow.Tensor_Tensor %6435, %6434 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6437 = torch.aten.neg %6436 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6438 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_591 = torch.constant.int 1 loc(#loc1) | |
| %6439 = torch.aten.sub.Tensor %6411, %6438, %int1_591 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6440 = torch.aten.lift_fresh_copy %5855 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6441 = torch.aten.pow.Tensor_Tensor %6440, %6439 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6442 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_592 = torch.constant.int 1 loc(#loc1) | |
| %6443 = torch.aten.sub.Tensor %6441, %6442, %int1_592 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6444 = torch.aten.gt.Tensor %6432, %6443 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6445 = torch.aten.where.self %6444, %6443, %6432 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6446 = torch.aten.lt.Tensor %6445, %6437 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6447 = torch.aten.where.self %6446, %6437, %6445 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6448 = torch.operator "aten.round"(%6447) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_593 = torch.constant.int 1 loc(#loc1) | |
| %6449 = torch.aten.sub.Tensor %6448, %6430, %int1_593 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6450 = torch.aten.mul.Tensor %6449, %6428 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_594 = torch.constant.int 320 loc(#loc1) | |
| %int20_595 = torch.constant.int 20 loc(#loc1) | |
| %int16_596 = torch.constant.int 16 loc(#loc1) | |
| %6451 = torch.prim.ListConstruct %int320_594, %int20_595, %int16_596 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_597 = torch.constant.bool false loc(#loc1) | |
| %6452 = torch.aten.expand %5854, %6451, %false_597 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_598 = torch.constant.int 0 loc(#loc1) | |
| %6453 = torch.aten.clone %6452, %int0_598 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_599 = torch.constant.int 320 loc(#loc1) | |
| %int320_600 = torch.constant.int 320 loc(#loc1) | |
| %6454 = torch.prim.ListConstruct %int320_599, %int320_600 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6455 = torch.aten._unsafe_view %6453, %6454 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %6456 = torch.aten.mul.Tensor %5853, %6455 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_601 = torch.constant.int 6 loc(#loc1) | |
| %none_602 = torch.constant.none loc(#loc0) | |
| %none_603 = torch.constant.none loc(#loc0) | |
| %none_604 = torch.constant.none loc(#loc0) | |
| %false_605 = torch.constant.bool false loc(#loc1) | |
| %none_606 = torch.constant.none loc(#loc0) | |
| %6457 = torch.aten._to_copy %6456, %int6_601, %none_602, %none_603, %none_604, %false_605, %none_606 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %6458 = torch.aten.t %6457 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_607 = torch.constant.int 8192 loc(#loc1) | |
| %int320_608 = torch.constant.int 320 loc(#loc1) | |
| %6459 = torch.prim.ListConstruct %int8192_607, %int320_608 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6460 = torch.aten.view %6450, %6459 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_609 = torch.constant.int 1 loc(#loc1) | |
| %int1_610 = torch.constant.int 1 loc(#loc1) | |
| %6461 = torch.aten.addmm %5852, %6460, %6458, %int1_609, %int1_610 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_611 = torch.constant.int 2 loc(#loc1) | |
| %int4096_612 = torch.constant.int 4096 loc(#loc1) | |
| %int320_613 = torch.constant.int 320 loc(#loc1) | |
| %6462 = torch.prim.ListConstruct %int2_611, %int4096_612, %int320_613 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6463 = torch.aten.view %6461, %6462 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_614 = torch.constant.int 1 loc(#loc1) | |
| %6464 = torch.aten.add.Tensor %6463, %6184, %int1_614 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_615 = torch.constant.int 320 loc(#loc1) | |
| %6465 = torch.prim.ListConstruct %int320_615 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_616 = torch.constant.none loc(#loc0) | |
| %none_617 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05_618 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0_619, %result1_620, %result2_621 = torch.aten.native_layer_norm %6464, %6465, %none_616, %none_617, %float1.000000e-05_618 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,1],f32>, !torch.tensor<[2,4096,1],f32> loc(#loc1) | |
| %6466 = torch.aten.detach %5851 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6467 = torch.aten.detach %6466 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_622 = torch.constant.int 2 loc(#loc1) | |
| %int4096_623 = torch.constant.int 4096 loc(#loc1) | |
| %int20_624 = torch.constant.int 20 loc(#loc1) | |
| %int16_625 = torch.constant.int 16 loc(#loc1) | |
| %6468 = torch.prim.ListConstruct %int2_622, %int4096_623, %int20_624, %int16_625 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6469 = torch.aten.view %result0_619, %6468 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %6470 = torch.aten.abs %6469 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_626 = torch.constant.int 3 loc(#loc1) | |
| %true_627 = torch.constant.bool true loc(#loc1) | |
| %values_628, %indices_629 = torch.aten.max.dim %6470, %int3_626, %true_627 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_630 = torch.constant.int 2 loc(#loc1) | |
| %int4096_631 = torch.constant.int 4096 loc(#loc1) | |
| %int20_632 = torch.constant.int 20 loc(#loc1) | |
| %int1_633 = torch.constant.int 1 loc(#loc1) | |
| %6471 = torch.prim.ListConstruct %int2_630, %int4096_631, %int20_632, %int1_633 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6472 = torch.aten.view %values_628, %6471 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_634 = torch.constant.int 2 loc(#loc1) | |
| %int4096_635 = torch.constant.int 4096 loc(#loc1) | |
| %int20_636 = torch.constant.int 20 loc(#loc1) | |
| %int16_637 = torch.constant.int 16 loc(#loc1) | |
| %6473 = torch.prim.ListConstruct %int2_634, %int4096_635, %int20_636, %int16_637 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_638 = torch.constant.bool false loc(#loc1) | |
| %6474 = torch.aten.expand %6472, %6473, %false_638 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_639 = torch.constant.int 0 loc(#loc1) | |
| %6475 = torch.aten.clone %6474, %int0_639 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_640 = torch.constant.int 2 loc(#loc1) | |
| %int4096_641 = torch.constant.int 4096 loc(#loc1) | |
| %int320_642 = torch.constant.int 320 loc(#loc1) | |
| %6476 = torch.prim.ListConstruct %int2_640, %int4096_641, %int320_642 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6477 = torch.aten._unsafe_view %6475, %6476 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6478 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_643 = torch.constant.int 1 loc(#loc1) | |
| %6479 = torch.aten.sub.Tensor %6467, %6478, %int1_643 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6480 = torch.aten.lift_fresh_copy %5850 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6481 = torch.aten.pow.Tensor_Tensor %6480, %6479 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6482 = torch.aten.neg %6481 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6483 = torch.aten.neg %6482 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6484 = torch.aten.div.Tensor %6477, %6483 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6485 = torch.aten.detach %5849 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6486 = torch.aten.detach %6485 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6487 = torch.aten.div.Tensor %result0_619, %6484 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_644 = torch.constant.int 1 loc(#loc1) | |
| %6488 = torch.aten.add.Tensor %6487, %6486, %int1_644 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6489 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_645 = torch.constant.int 1 loc(#loc1) | |
| %6490 = torch.aten.sub.Tensor %6467, %6489, %int1_645 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6491 = torch.aten.lift_fresh_copy %5848 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6492 = torch.aten.pow.Tensor_Tensor %6491, %6490 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6493 = torch.aten.neg %6492 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6494 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_646 = torch.constant.int 1 loc(#loc1) | |
| %6495 = torch.aten.sub.Tensor %6467, %6494, %int1_646 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6496 = torch.aten.lift_fresh_copy %5847 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6497 = torch.aten.pow.Tensor_Tensor %6496, %6495 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6498 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_647 = torch.constant.int 1 loc(#loc1) | |
| %6499 = torch.aten.sub.Tensor %6497, %6498, %int1_647 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6500 = torch.aten.gt.Tensor %6488, %6499 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6501 = torch.aten.where.self %6500, %6499, %6488 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6502 = torch.aten.lt.Tensor %6501, %6493 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6503 = torch.aten.where.self %6502, %6493, %6501 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6504 = torch.operator "aten.round"(%6503) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_648 = torch.constant.int 1 loc(#loc1) | |
| %6505 = torch.aten.sub.Tensor %6504, %6486, %int1_648 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6506 = torch.aten.mul.Tensor %6505, %6484 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_649 = torch.constant.int 320 loc(#loc1) | |
| %int20_650 = torch.constant.int 20 loc(#loc1) | |
| %int16_651 = torch.constant.int 16 loc(#loc1) | |
| %6507 = torch.prim.ListConstruct %int320_649, %int20_650, %int16_651 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_652 = torch.constant.bool false loc(#loc1) | |
| %6508 = torch.aten.expand %5846, %6507, %false_652 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_653 = torch.constant.int 0 loc(#loc1) | |
| %6509 = torch.aten.clone %6508, %int0_653 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_654 = torch.constant.int 320 loc(#loc1) | |
| %int320_655 = torch.constant.int 320 loc(#loc1) | |
| %6510 = torch.prim.ListConstruct %int320_654, %int320_655 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6511 = torch.aten._unsafe_view %6509, %6510 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %6512 = torch.aten.mul.Tensor %5845, %6511 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_656 = torch.constant.int 6 loc(#loc1) | |
| %none_657 = torch.constant.none loc(#loc0) | |
| %none_658 = torch.constant.none loc(#loc0) | |
| %none_659 = torch.constant.none loc(#loc0) | |
| %false_660 = torch.constant.bool false loc(#loc1) | |
| %none_661 = torch.constant.none loc(#loc0) | |
| %6513 = torch.aten._to_copy %6512, %int6_656, %none_657, %none_658, %none_659, %false_660, %none_661 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %6514 = torch.aten.t %6513 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_662 = torch.constant.int 8192 loc(#loc1) | |
| %int320_663 = torch.constant.int 320 loc(#loc1) | |
| %6515 = torch.prim.ListConstruct %int8192_662, %int320_663 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6516 = torch.aten.view %6506, %6515 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_664 = torch.constant.int 1 loc(#loc1) | |
| %int1_665 = torch.constant.int 1 loc(#loc1) | |
| %6517 = torch.aten.addmm %5844, %6516, %6514, %int1_664, %int1_665 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_666 = torch.constant.int 2 loc(#loc1) | |
| %int4096_667 = torch.constant.int 4096 loc(#loc1) | |
| %int320_668 = torch.constant.int 320 loc(#loc1) | |
| %6518 = torch.prim.ListConstruct %int2_666, %int4096_667, %int320_668 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6519 = torch.aten.view %6517, %6518 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6520 = torch.aten.detach %5843 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6521 = torch.aten.detach %6520 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_669 = torch.constant.int 2 loc(#loc1) | |
| %int77 = torch.constant.int 77 loc(#loc1) | |
| %int48 = torch.constant.int 48 loc(#loc1) | |
| %int16_670 = torch.constant.int 16 loc(#loc1) | |
| %6522 = torch.prim.ListConstruct %int2_669, %int77, %int48, %int16_670 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6523 = torch.aten.view %3921, %6522 : !torch.tensor<[2,77,768],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %6524 = torch.aten.abs %6523 : !torch.tensor<[2,77,48,16],f32> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int3_671 = torch.constant.int 3 loc(#loc1) | |
| %true_672 = torch.constant.bool true loc(#loc1) | |
| %values_673, %indices_674 = torch.aten.max.dim %6524, %int3_671, %true_672 : !torch.tensor<[2,77,48,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,77,48,1],f32>, !torch.tensor<[2,77,48,1],si64> loc(#loc1) | |
| %int2_675 = torch.constant.int 2 loc(#loc1) | |
| %int77_676 = torch.constant.int 77 loc(#loc1) | |
| %int48_677 = torch.constant.int 48 loc(#loc1) | |
| %int1_678 = torch.constant.int 1 loc(#loc1) | |
| %6525 = torch.prim.ListConstruct %int2_675, %int77_676, %int48_677, %int1_678 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6526 = torch.aten.view %values_673, %6525 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,1],f32> loc(#loc1) | |
| %int2_679 = torch.constant.int 2 loc(#loc1) | |
| %int77_680 = torch.constant.int 77 loc(#loc1) | |
| %int48_681 = torch.constant.int 48 loc(#loc1) | |
| %int16_682 = torch.constant.int 16 loc(#loc1) | |
| %6527 = torch.prim.ListConstruct %int2_679, %int77_680, %int48_681, %int16_682 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_683 = torch.constant.bool false loc(#loc1) | |
| %6528 = torch.aten.expand %6526, %6527, %false_683 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int0_684 = torch.constant.int 0 loc(#loc1) | |
| %6529 = torch.aten.clone %6528, %int0_684 : !torch.tensor<[2,77,48,16],f32>, !torch.int -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int2_685 = torch.constant.int 2 loc(#loc1) | |
| %int77_686 = torch.constant.int 77 loc(#loc1) | |
| %int768 = torch.constant.int 768 loc(#loc1) | |
| %6530 = torch.prim.ListConstruct %int2_685, %int77_686, %int768 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6531 = torch.aten._unsafe_view %6529, %6530 : !torch.tensor<[2,77,48,16],f32>, !torch.list<int> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6532 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_687 = torch.constant.int 1 loc(#loc1) | |
| %6533 = torch.aten.sub.Tensor %6521, %6532, %int1_687 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6534 = torch.aten.lift_fresh_copy %5842 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6535 = torch.aten.pow.Tensor_Tensor %6534, %6533 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6536 = torch.aten.neg %6535 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6537 = torch.aten.neg %6536 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6538 = torch.aten.div.Tensor %6531, %6537 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6539 = torch.aten.detach %5841 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6540 = torch.aten.detach %6539 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6541 = torch.aten.div.Tensor %3921, %6538 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_688 = torch.constant.int 1 loc(#loc1) | |
| %6542 = torch.aten.add.Tensor %6541, %6540, %int1_688 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6543 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_689 = torch.constant.int 1 loc(#loc1) | |
| %6544 = torch.aten.sub.Tensor %6521, %6543, %int1_689 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6545 = torch.aten.lift_fresh_copy %5840 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6546 = torch.aten.pow.Tensor_Tensor %6545, %6544 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6547 = torch.aten.neg %6546 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6548 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_690 = torch.constant.int 1 loc(#loc1) | |
| %6549 = torch.aten.sub.Tensor %6521, %6548, %int1_690 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6550 = torch.aten.lift_fresh_copy %5839 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6551 = torch.aten.pow.Tensor_Tensor %6550, %6549 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6552 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_691 = torch.constant.int 1 loc(#loc1) | |
| %6553 = torch.aten.sub.Tensor %6551, %6552, %int1_691 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6554 = torch.aten.gt.Tensor %6542, %6553 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %6555 = torch.aten.where.self %6554, %6553, %6542 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6556 = torch.aten.lt.Tensor %6555, %6547 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %6557 = torch.aten.where.self %6556, %6547, %6555 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6558 = torch.operator "aten.round"(%6557) : (!torch.tensor<[2,77,768],f32>) -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_692 = torch.constant.int 1 loc(#loc1) | |
| %6559 = torch.aten.sub.Tensor %6558, %6540, %int1_692 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6560 = torch.aten.mul.Tensor %6559, %6538 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int320_693 = torch.constant.int 320 loc(#loc1) | |
| %int48_694 = torch.constant.int 48 loc(#loc1) | |
| %int16_695 = torch.constant.int 16 loc(#loc1) | |
| %6561 = torch.prim.ListConstruct %int320_693, %int48_694, %int16_695 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_696 = torch.constant.bool false loc(#loc1) | |
| %6562 = torch.aten.expand %5838, %6561, %false_696 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,48,16],f16> loc(#loc1) | |
| %int0_697 = torch.constant.int 0 loc(#loc1) | |
| %6563 = torch.aten.clone %6562, %int0_697 : !torch.tensor<[320,48,16],f16>, !torch.int -> !torch.tensor<[320,48,16],f16> loc(#loc1) | |
| %int320_698 = torch.constant.int 320 loc(#loc1) | |
| %int768_699 = torch.constant.int 768 loc(#loc1) | |
| %6564 = torch.prim.ListConstruct %int320_698, %int768_699 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6565 = torch.aten._unsafe_view %6563, %6564 : !torch.tensor<[320,48,16],f16>, !torch.list<int> -> !torch.tensor<[320,768],f16> loc(#loc1) | |
| %6566 = torch.aten.mul.Tensor %5837, %6565 : !torch.tensor, !torch.tensor<[320,768],f16> -> !torch.tensor<[320,768],f16> loc(#loc1) | |
| %int6_700 = torch.constant.int 6 loc(#loc1) | |
| %none_701 = torch.constant.none loc(#loc0) | |
| %none_702 = torch.constant.none loc(#loc0) | |
| %none_703 = torch.constant.none loc(#loc0) | |
| %false_704 = torch.constant.bool false loc(#loc1) | |
| %none_705 = torch.constant.none loc(#loc0) | |
| %6567 = torch.aten._to_copy %6566, %int6_700, %none_701, %none_702, %none_703, %false_704, %none_705 : !torch.tensor<[320,768],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,768],f32> loc(#loc1) | |
| %6568 = torch.aten.t %6567 : !torch.tensor<[320,768],f32> -> !torch.tensor<[768,320],f32> loc(#loc1) | |
| %int154 = torch.constant.int 154 loc(#loc1) | |
| %int768_706 = torch.constant.int 768 loc(#loc1) | |
| %6569 = torch.prim.ListConstruct %int154, %int768_706 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int768_707 = torch.constant.int 768 loc(#loc1) | |
| %int1_708 = torch.constant.int 1 loc(#loc1) | |
| %6570 = torch.prim.ListConstruct %int768_707, %int1_708 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6571 = torch.aten._reshape_alias %6560, %6569, %6570 : !torch.tensor<[2,77,768],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[154,768],f32> loc(#loc1) | |
| %6572 = torch.aten.mm %6571, %6568 : !torch.tensor<[154,768],f32>, !torch.tensor<[768,320],f32> -> !torch.tensor<[154,320],f32> loc(#loc1) | |
| %int2_709 = torch.constant.int 2 loc(#loc1) | |
| %int77_710 = torch.constant.int 77 loc(#loc1) | |
| %int320_711 = torch.constant.int 320 loc(#loc1) | |
| %6573 = torch.prim.ListConstruct %int2_709, %int77_710, %int320_711 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6574 = torch.aten._unsafe_view %6572, %6573 : !torch.tensor<[154,320],f32>, !torch.list<int> -> !torch.tensor<[2,77,320],f32> loc(#loc1) | |
| %6575 = torch.aten.detach %5836 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6576 = torch.aten.detach %6575 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_712 = torch.constant.int 2 loc(#loc1) | |
| %int77_713 = torch.constant.int 77 loc(#loc1) | |
| %int48_714 = torch.constant.int 48 loc(#loc1) | |
| %int16_715 = torch.constant.int 16 loc(#loc1) | |
| %6577 = torch.prim.ListConstruct %int2_712, %int77_713, %int48_714, %int16_715 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6578 = torch.aten.view %3921, %6577 : !torch.tensor<[2,77,768],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %6579 = torch.aten.abs %6578 : !torch.tensor<[2,77,48,16],f32> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int3_716 = torch.constant.int 3 loc(#loc1) | |
| %true_717 = torch.constant.bool true loc(#loc1) | |
| %values_718, %indices_719 = torch.aten.max.dim %6579, %int3_716, %true_717 : !torch.tensor<[2,77,48,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,77,48,1],f32>, !torch.tensor<[2,77,48,1],si64> loc(#loc1) | |
| %int2_720 = torch.constant.int 2 loc(#loc1) | |
| %int77_721 = torch.constant.int 77 loc(#loc1) | |
| %int48_722 = torch.constant.int 48 loc(#loc1) | |
| %int1_723 = torch.constant.int 1 loc(#loc1) | |
| %6580 = torch.prim.ListConstruct %int2_720, %int77_721, %int48_722, %int1_723 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6581 = torch.aten.view %values_718, %6580 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,1],f32> loc(#loc1) | |
| %int2_724 = torch.constant.int 2 loc(#loc1) | |
| %int77_725 = torch.constant.int 77 loc(#loc1) | |
| %int48_726 = torch.constant.int 48 loc(#loc1) | |
| %int16_727 = torch.constant.int 16 loc(#loc1) | |
| %6582 = torch.prim.ListConstruct %int2_724, %int77_725, %int48_726, %int16_727 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_728 = torch.constant.bool false loc(#loc1) | |
| %6583 = torch.aten.expand %6581, %6582, %false_728 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int0_729 = torch.constant.int 0 loc(#loc1) | |
| %6584 = torch.aten.clone %6583, %int0_729 : !torch.tensor<[2,77,48,16],f32>, !torch.int -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int2_730 = torch.constant.int 2 loc(#loc1) | |
| %int77_731 = torch.constant.int 77 loc(#loc1) | |
| %int768_732 = torch.constant.int 768 loc(#loc1) | |
| %6585 = torch.prim.ListConstruct %int2_730, %int77_731, %int768_732 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6586 = torch.aten._unsafe_view %6584, %6585 : !torch.tensor<[2,77,48,16],f32>, !torch.list<int> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6587 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_733 = torch.constant.int 1 loc(#loc1) | |
| %6588 = torch.aten.sub.Tensor %6576, %6587, %int1_733 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6589 = torch.aten.lift_fresh_copy %5835 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6590 = torch.aten.pow.Tensor_Tensor %6589, %6588 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6591 = torch.aten.neg %6590 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6592 = torch.aten.neg %6591 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6593 = torch.aten.div.Tensor %6586, %6592 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6594 = torch.aten.detach %5834 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6595 = torch.aten.detach %6594 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6596 = torch.aten.div.Tensor %3921, %6593 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_734 = torch.constant.int 1 loc(#loc1) | |
| %6597 = torch.aten.add.Tensor %6596, %6595, %int1_734 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6598 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_735 = torch.constant.int 1 loc(#loc1) | |
| %6599 = torch.aten.sub.Tensor %6576, %6598, %int1_735 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6600 = torch.aten.lift_fresh_copy %5833 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6601 = torch.aten.pow.Tensor_Tensor %6600, %6599 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6602 = torch.aten.neg %6601 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6603 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_736 = torch.constant.int 1 loc(#loc1) | |
| %6604 = torch.aten.sub.Tensor %6576, %6603, %int1_736 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6605 = torch.aten.lift_fresh_copy %5832 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6606 = torch.aten.pow.Tensor_Tensor %6605, %6604 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6607 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_737 = torch.constant.int 1 loc(#loc1) | |
| %6608 = torch.aten.sub.Tensor %6606, %6607, %int1_737 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6609 = torch.aten.gt.Tensor %6597, %6608 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %6610 = torch.aten.where.self %6609, %6608, %6597 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6611 = torch.aten.lt.Tensor %6610, %6602 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %6612 = torch.aten.where.self %6611, %6602, %6610 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6613 = torch.operator "aten.round"(%6612) : (!torch.tensor<[2,77,768],f32>) -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_738 = torch.constant.int 1 loc(#loc1) | |
| %6614 = torch.aten.sub.Tensor %6613, %6595, %int1_738 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %6615 = torch.aten.mul.Tensor %6614, %6593 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int320_739 = torch.constant.int 320 loc(#loc1) | |
| %int48_740 = torch.constant.int 48 loc(#loc1) | |
| %int16_741 = torch.constant.int 16 loc(#loc1) | |
| %6616 = torch.prim.ListConstruct %int320_739, %int48_740, %int16_741 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_742 = torch.constant.bool false loc(#loc1) | |
| %6617 = torch.aten.expand %5831, %6616, %false_742 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,48,16],f16> loc(#loc1) | |
| %int0_743 = torch.constant.int 0 loc(#loc1) | |
| %6618 = torch.aten.clone %6617, %int0_743 : !torch.tensor<[320,48,16],f16>, !torch.int -> !torch.tensor<[320,48,16],f16> loc(#loc1) | |
| %int320_744 = torch.constant.int 320 loc(#loc1) | |
| %int768_745 = torch.constant.int 768 loc(#loc1) | |
| %6619 = torch.prim.ListConstruct %int320_744, %int768_745 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6620 = torch.aten._unsafe_view %6618, %6619 : !torch.tensor<[320,48,16],f16>, !torch.list<int> -> !torch.tensor<[320,768],f16> loc(#loc1) | |
| %6621 = torch.aten.mul.Tensor %5830, %6620 : !torch.tensor, !torch.tensor<[320,768],f16> -> !torch.tensor<[320,768],f16> loc(#loc1) | |
| %int6_746 = torch.constant.int 6 loc(#loc1) | |
| %none_747 = torch.constant.none loc(#loc0) | |
| %none_748 = torch.constant.none loc(#loc0) | |
| %none_749 = torch.constant.none loc(#loc0) | |
| %false_750 = torch.constant.bool false loc(#loc1) | |
| %none_751 = torch.constant.none loc(#loc0) | |
| %6622 = torch.aten._to_copy %6621, %int6_746, %none_747, %none_748, %none_749, %false_750, %none_751 : !torch.tensor<[320,768],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,768],f32> loc(#loc1) | |
| %6623 = torch.aten.t %6622 : !torch.tensor<[320,768],f32> -> !torch.tensor<[768,320],f32> loc(#loc1) | |
| %int154_752 = torch.constant.int 154 loc(#loc1) | |
| %int768_753 = torch.constant.int 768 loc(#loc1) | |
| %6624 = torch.prim.ListConstruct %int154_752, %int768_753 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int768_754 = torch.constant.int 768 loc(#loc1) | |
| %int1_755 = torch.constant.int 1 loc(#loc1) | |
| %6625 = torch.prim.ListConstruct %int768_754, %int1_755 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6626 = torch.aten._reshape_alias %6615, %6624, %6625 : !torch.tensor<[2,77,768],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[154,768],f32> loc(#loc1) | |
| %6627 = torch.aten.mm %6626, %6623 : !torch.tensor<[154,768],f32>, !torch.tensor<[768,320],f32> -> !torch.tensor<[154,320],f32> loc(#loc1) | |
| %int2_756 = torch.constant.int 2 loc(#loc1) | |
| %int77_757 = torch.constant.int 77 loc(#loc1) | |
| %int320_758 = torch.constant.int 320 loc(#loc1) | |
| %6628 = torch.prim.ListConstruct %int2_756, %int77_757, %int320_758 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6629 = torch.aten._unsafe_view %6627, %6628 : !torch.tensor<[154,320],f32>, !torch.list<int> -> !torch.tensor<[2,77,320],f32> loc(#loc1) | |
| %int2_759 = torch.constant.int 2 loc(#loc1) | |
| %int4096_760 = torch.constant.int 4096 loc(#loc1) | |
| %int8_761 = torch.constant.int 8 loc(#loc1) | |
| %int40_762 = torch.constant.int 40 loc(#loc1) | |
| %6630 = torch.prim.ListConstruct %int2_759, %int4096_760, %int8_761, %int40_762 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_763 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_764 = torch.constant.int 320 loc(#loc1) | |
| %int40_765 = torch.constant.int 40 loc(#loc1) | |
| %int1_766 = torch.constant.int 1 loc(#loc1) | |
| %6631 = torch.prim.ListConstruct %int1310720_763, %int320_764, %int40_765, %int1_766 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6632 = torch.aten._reshape_alias %6519, %6630, %6631 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_767 = torch.constant.int 0 loc(#loc1) | |
| %int2_768 = torch.constant.int 2 loc(#loc1) | |
| %int1_769 = torch.constant.int 1 loc(#loc1) | |
| %int3_770 = torch.constant.int 3 loc(#loc1) | |
| %6633 = torch.prim.ListConstruct %int0_767, %int2_768, %int1_769, %int3_770 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6634 = torch.aten.permute %6632, %6633 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_771 = torch.constant.int 0 loc(#loc1) | |
| %6635 = torch.aten.clone %6634, %int0_771 : !torch.tensor<[2,8,4096,40],f32>, !torch.int -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int16_772 = torch.constant.int 16 loc(#loc1) | |
| %int4096_773 = torch.constant.int 4096 loc(#loc1) | |
| %int40_774 = torch.constant.int 40 loc(#loc1) | |
| %6636 = torch.prim.ListConstruct %int16_772, %int4096_773, %int40_774 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6637 = torch.aten._unsafe_view %6635, %6636 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_775 = torch.constant.int 2 loc(#loc1) | |
| %int77_776 = torch.constant.int 77 loc(#loc1) | |
| %int8_777 = torch.constant.int 8 loc(#loc1) | |
| %int40_778 = torch.constant.int 40 loc(#loc1) | |
| %6638 = torch.prim.ListConstruct %int2_775, %int77_776, %int8_777, %int40_778 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int24640 = torch.constant.int 24640 loc(#loc1) | |
| %int320_779 = torch.constant.int 320 loc(#loc1) | |
| %int40_780 = torch.constant.int 40 loc(#loc1) | |
| %int1_781 = torch.constant.int 1 loc(#loc1) | |
| %6639 = torch.prim.ListConstruct %int24640, %int320_779, %int40_780, %int1_781 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6640 = torch.aten._reshape_alias %6574, %6638, %6639 : !torch.tensor<[2,77,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,77,8,40],f32> loc(#loc1) | |
| %int0_782 = torch.constant.int 0 loc(#loc1) | |
| %int2_783 = torch.constant.int 2 loc(#loc1) | |
| %int1_784 = torch.constant.int 1 loc(#loc1) | |
| %int3_785 = torch.constant.int 3 loc(#loc1) | |
| %6641 = torch.prim.ListConstruct %int0_782, %int2_783, %int1_784, %int3_785 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6642 = torch.aten.permute %6640, %6641 : !torch.tensor<[2,77,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,77,40],f32> loc(#loc1) | |
| %int0_786 = torch.constant.int 0 loc(#loc1) | |
| %6643 = torch.aten.clone %6642, %int0_786 : !torch.tensor<[2,8,77,40],f32>, !torch.int -> !torch.tensor<[2,8,77,40],f32> loc(#loc1) | |
| %int16_787 = torch.constant.int 16 loc(#loc1) | |
| %int77_788 = torch.constant.int 77 loc(#loc1) | |
| %int40_789 = torch.constant.int 40 loc(#loc1) | |
| %6644 = torch.prim.ListConstruct %int16_787, %int77_788, %int40_789 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6645 = torch.aten._unsafe_view %6643, %6644 : !torch.tensor<[2,8,77,40],f32>, !torch.list<int> -> !torch.tensor<[16,77,40],f32> loc(#loc1) | |
| %int2_790 = torch.constant.int 2 loc(#loc1) | |
| %int77_791 = torch.constant.int 77 loc(#loc1) | |
| %int8_792 = torch.constant.int 8 loc(#loc1) | |
| %int40_793 = torch.constant.int 40 loc(#loc1) | |
| %6646 = torch.prim.ListConstruct %int2_790, %int77_791, %int8_792, %int40_793 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int24640_794 = torch.constant.int 24640 loc(#loc1) | |
| %int320_795 = torch.constant.int 320 loc(#loc1) | |
| %int40_796 = torch.constant.int 40 loc(#loc1) | |
| %int1_797 = torch.constant.int 1 loc(#loc1) | |
| %6647 = torch.prim.ListConstruct %int24640_794, %int320_795, %int40_796, %int1_797 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6648 = torch.aten._reshape_alias %6629, %6646, %6647 : !torch.tensor<[2,77,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,77,8,40],f32> loc(#loc1) | |
| %int0_798 = torch.constant.int 0 loc(#loc1) | |
| %int2_799 = torch.constant.int 2 loc(#loc1) | |
| %int1_800 = torch.constant.int 1 loc(#loc1) | |
| %int3_801 = torch.constant.int 3 loc(#loc1) | |
| %6649 = torch.prim.ListConstruct %int0_798, %int2_799, %int1_800, %int3_801 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6650 = torch.aten.permute %6648, %6649 : !torch.tensor<[2,77,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,77,40],f32> loc(#loc1) | |
| %int0_802 = torch.constant.int 0 loc(#loc1) | |
| %6651 = torch.aten.clone %6650, %int0_802 : !torch.tensor<[2,8,77,40],f32>, !torch.int -> !torch.tensor<[2,8,77,40],f32> loc(#loc1) | |
| %int16_803 = torch.constant.int 16 loc(#loc1) | |
| %int77_804 = torch.constant.int 77 loc(#loc1) | |
| %int40_805 = torch.constant.int 40 loc(#loc1) | |
| %6652 = torch.prim.ListConstruct %int16_803, %int77_804, %int40_805 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6653 = torch.aten._unsafe_view %6651, %6652 : !torch.tensor<[2,8,77,40],f32>, !torch.list<int> -> !torch.tensor<[16,77,40],f32> loc(#loc1) | |
| %int-1_806 = torch.constant.int -1 loc(#loc1) | |
| %int-2_807 = torch.constant.int -2 loc(#loc1) | |
| %6654 = torch.aten.transpose.int %6645, %int-1_806, %int-2_807 : !torch.tensor<[16,77,40],f32>, !torch.int, !torch.int -> !torch.tensor<[16,40,77],f32> loc(#loc1) | |
| %int16_808 = torch.constant.int 16 loc(#loc1) | |
| %int4096_809 = torch.constant.int 4096 loc(#loc1) | |
| %int40_810 = torch.constant.int 40 loc(#loc1) | |
| %6655 = torch.prim.ListConstruct %int16_808, %int4096_809, %int40_810 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_811 = torch.constant.bool false loc(#loc1) | |
| %6656 = torch.aten.expand %6637, %6655, %false_811 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_812 = torch.constant.int 16 loc(#loc1) | |
| %int4096_813 = torch.constant.int 4096 loc(#loc1) | |
| %int40_814 = torch.constant.int 40 loc(#loc1) | |
| %6657 = torch.prim.ListConstruct %int16_812, %int4096_813, %int40_814 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int163840_815 = torch.constant.int 163840 loc(#loc1) | |
| %int40_816 = torch.constant.int 40 loc(#loc1) | |
| %int1_817 = torch.constant.int 1 loc(#loc1) | |
| %6658 = torch.prim.ListConstruct %int163840_815, %int40_816, %int1_817 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6659 = torch.aten._reshape_alias %6656, %6657, %6658 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_818 = torch.constant.int 16 loc(#loc1) | |
| %int40_819 = torch.constant.int 40 loc(#loc1) | |
| %int77_820 = torch.constant.int 77 loc(#loc1) | |
| %6660 = torch.prim.ListConstruct %int16_818, %int40_819, %int77_820 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_821 = torch.constant.bool false loc(#loc1) | |
| %6661 = torch.aten.expand %6654, %6660, %false_821 : !torch.tensor<[16,40,77],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,40,77],f32> loc(#loc1) | |
| %int16_822 = torch.constant.int 16 loc(#loc1) | |
| %int40_823 = torch.constant.int 40 loc(#loc1) | |
| %int77_824 = torch.constant.int 77 loc(#loc1) | |
| %6662 = torch.prim.ListConstruct %int16_822, %int40_823, %int77_824 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int3080 = torch.constant.int 3080 loc(#loc1) | |
| %int1_825 = torch.constant.int 1 loc(#loc1) | |
| %int40_826 = torch.constant.int 40 loc(#loc1) | |
| %6663 = torch.prim.ListConstruct %int3080, %int1_825, %int40_826 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6664 = torch.aten._reshape_alias %6661, %6662, %6663 : !torch.tensor<[16,40,77],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,40,77],f32> loc(#loc1) | |
| %6665 = torch.aten.bmm %6659, %6664 : !torch.tensor<[16,4096,40],f32>, !torch.tensor<[16,40,77],f32> -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int16_827 = torch.constant.int 16 loc(#loc1) | |
| %int4096_828 = torch.constant.int 4096 loc(#loc1) | |
| %int77_829 = torch.constant.int 77 loc(#loc1) | |
| %6666 = torch.prim.ListConstruct %int16_827, %int4096_828, %int77_829 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6667 = torch.aten._unsafe_view %6665, %6666 : !torch.tensor<[16,4096,77],f32>, !torch.list<int> -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %6668 = torch.tensor.literal(dense<0.15811388300841897> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %6669 = torch.aten.mul.Tensor %6667, %6668 : !torch.tensor<[16,4096,77],f32>, !torch.tensor<[],f64> -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int-1_830 = torch.constant.int -1 loc(#loc1) | |
| %false_831 = torch.constant.bool false loc(#loc1) | |
| %6670 = torch.aten._softmax %6669, %int-1_830, %false_831 : !torch.tensor<[16,4096,77],f32>, !torch.int, !torch.bool -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int16_832 = torch.constant.int 16 loc(#loc1) | |
| %int4096_833 = torch.constant.int 4096 loc(#loc1) | |
| %int77_834 = torch.constant.int 77 loc(#loc1) | |
| %6671 = torch.prim.ListConstruct %int16_832, %int4096_833, %int77_834 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_835 = torch.constant.bool false loc(#loc1) | |
| %6672 = torch.aten.expand %6670, %6671, %false_835 : !torch.tensor<[16,4096,77],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int16_836 = torch.constant.int 16 loc(#loc1) | |
| %int4096_837 = torch.constant.int 4096 loc(#loc1) | |
| %int77_838 = torch.constant.int 77 loc(#loc1) | |
| %6673 = torch.prim.ListConstruct %int16_836, %int4096_837, %int77_838 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int315392 = torch.constant.int 315392 loc(#loc1) | |
| %int77_839 = torch.constant.int 77 loc(#loc1) | |
| %int1_840 = torch.constant.int 1 loc(#loc1) | |
| %6674 = torch.prim.ListConstruct %int315392, %int77_839, %int1_840 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6675 = torch.aten._reshape_alias %6672, %6673, %6674 : !torch.tensor<[16,4096,77],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int16_841 = torch.constant.int 16 loc(#loc1) | |
| %int77_842 = torch.constant.int 77 loc(#loc1) | |
| %int40_843 = torch.constant.int 40 loc(#loc1) | |
| %6676 = torch.prim.ListConstruct %int16_841, %int77_842, %int40_843 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_844 = torch.constant.bool false loc(#loc1) | |
| %6677 = torch.aten.expand %6653, %6676, %false_844 : !torch.tensor<[16,77,40],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,77,40],f32> loc(#loc1) | |
| %int16_845 = torch.constant.int 16 loc(#loc1) | |
| %int77_846 = torch.constant.int 77 loc(#loc1) | |
| %int40_847 = torch.constant.int 40 loc(#loc1) | |
| %6678 = torch.prim.ListConstruct %int16_845, %int77_846, %int40_847 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int3080_848 = torch.constant.int 3080 loc(#loc1) | |
| %int40_849 = torch.constant.int 40 loc(#loc1) | |
| %int1_850 = torch.constant.int 1 loc(#loc1) | |
| %6679 = torch.prim.ListConstruct %int3080_848, %int40_849, %int1_850 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6680 = torch.aten._reshape_alias %6677, %6678, %6679 : !torch.tensor<[16,77,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,77,40],f32> loc(#loc1) | |
| %6681 = torch.aten.bmm %6675, %6680 : !torch.tensor<[16,4096,77],f32>, !torch.tensor<[16,77,40],f32> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_851 = torch.constant.int 16 loc(#loc1) | |
| %int4096_852 = torch.constant.int 4096 loc(#loc1) | |
| %int40_853 = torch.constant.int 40 loc(#loc1) | |
| %6682 = torch.prim.ListConstruct %int16_851, %int4096_852, %int40_853 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6683 = torch.aten._unsafe_view %6681, %6682 : !torch.tensor<[16,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_854 = torch.constant.int 2 loc(#loc1) | |
| %int8_855 = torch.constant.int 8 loc(#loc1) | |
| %int4096_856 = torch.constant.int 4096 loc(#loc1) | |
| %int40_857 = torch.constant.int 40 loc(#loc1) | |
| %6684 = torch.prim.ListConstruct %int2_854, %int8_855, %int4096_856, %int40_857 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_858 = torch.constant.int 1310720 loc(#loc1) | |
| %int163840_859 = torch.constant.int 163840 loc(#loc1) | |
| %int40_860 = torch.constant.int 40 loc(#loc1) | |
| %int1_861 = torch.constant.int 1 loc(#loc1) | |
| %6685 = torch.prim.ListConstruct %int1310720_858, %int163840_859, %int40_860, %int1_861 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6686 = torch.aten._reshape_alias %6683, %6684, %6685 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_862 = torch.constant.int 0 loc(#loc1) | |
| %int2_863 = torch.constant.int 2 loc(#loc1) | |
| %int1_864 = torch.constant.int 1 loc(#loc1) | |
| %int3_865 = torch.constant.int 3 loc(#loc1) | |
| %6687 = torch.prim.ListConstruct %int0_862, %int2_863, %int1_864, %int3_865 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6688 = torch.aten.permute %6686, %6687 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_866 = torch.constant.int 0 loc(#loc1) | |
| %6689 = torch.aten.clone %6688, %int0_866 : !torch.tensor<[2,4096,8,40],f32>, !torch.int -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int2_867 = torch.constant.int 2 loc(#loc1) | |
| %int4096_868 = torch.constant.int 4096 loc(#loc1) | |
| %int320_869 = torch.constant.int 320 loc(#loc1) | |
| %6690 = torch.prim.ListConstruct %int2_867, %int4096_868, %int320_869 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6691 = torch.aten._unsafe_view %6689, %6690 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6692 = torch.aten.detach %5829 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6693 = torch.aten.detach %6692 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_870 = torch.constant.int 2 loc(#loc1) | |
| %int4096_871 = torch.constant.int 4096 loc(#loc1) | |
| %int20_872 = torch.constant.int 20 loc(#loc1) | |
| %int16_873 = torch.constant.int 16 loc(#loc1) | |
| %6694 = torch.prim.ListConstruct %int2_870, %int4096_871, %int20_872, %int16_873 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6695 = torch.aten.view %6691, %6694 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %6696 = torch.aten.abs %6695 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_874 = torch.constant.int 3 loc(#loc1) | |
| %true_875 = torch.constant.bool true loc(#loc1) | |
| %values_876, %indices_877 = torch.aten.max.dim %6696, %int3_874, %true_875 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_878 = torch.constant.int 2 loc(#loc1) | |
| %int4096_879 = torch.constant.int 4096 loc(#loc1) | |
| %int20_880 = torch.constant.int 20 loc(#loc1) | |
| %int1_881 = torch.constant.int 1 loc(#loc1) | |
| %6697 = torch.prim.ListConstruct %int2_878, %int4096_879, %int20_880, %int1_881 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6698 = torch.aten.view %values_876, %6697 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_882 = torch.constant.int 2 loc(#loc1) | |
| %int4096_883 = torch.constant.int 4096 loc(#loc1) | |
| %int20_884 = torch.constant.int 20 loc(#loc1) | |
| %int16_885 = torch.constant.int 16 loc(#loc1) | |
| %6699 = torch.prim.ListConstruct %int2_882, %int4096_883, %int20_884, %int16_885 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_886 = torch.constant.bool false loc(#loc1) | |
| %6700 = torch.aten.expand %6698, %6699, %false_886 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_887 = torch.constant.int 0 loc(#loc1) | |
| %6701 = torch.aten.clone %6700, %int0_887 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_888 = torch.constant.int 2 loc(#loc1) | |
| %int4096_889 = torch.constant.int 4096 loc(#loc1) | |
| %int320_890 = torch.constant.int 320 loc(#loc1) | |
| %6702 = torch.prim.ListConstruct %int2_888, %int4096_889, %int320_890 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6703 = torch.aten._unsafe_view %6701, %6702 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6704 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_891 = torch.constant.int 1 loc(#loc1) | |
| %6705 = torch.aten.sub.Tensor %6693, %6704, %int1_891 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6706 = torch.aten.lift_fresh_copy %5828 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6707 = torch.aten.pow.Tensor_Tensor %6706, %6705 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6708 = torch.aten.neg %6707 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6709 = torch.aten.neg %6708 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6710 = torch.aten.div.Tensor %6703, %6709 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6711 = torch.aten.detach %5827 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6712 = torch.aten.detach %6711 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6713 = torch.aten.div.Tensor %6691, %6710 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_892 = torch.constant.int 1 loc(#loc1) | |
| %6714 = torch.aten.add.Tensor %6713, %6712, %int1_892 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6715 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_893 = torch.constant.int 1 loc(#loc1) | |
| %6716 = torch.aten.sub.Tensor %6693, %6715, %int1_893 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6717 = torch.aten.lift_fresh_copy %5826 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6718 = torch.aten.pow.Tensor_Tensor %6717, %6716 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6719 = torch.aten.neg %6718 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6720 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_894 = torch.constant.int 1 loc(#loc1) | |
| %6721 = torch.aten.sub.Tensor %6693, %6720, %int1_894 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6722 = torch.aten.lift_fresh_copy %5825 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6723 = torch.aten.pow.Tensor_Tensor %6722, %6721 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6724 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_895 = torch.constant.int 1 loc(#loc1) | |
| %6725 = torch.aten.sub.Tensor %6723, %6724, %int1_895 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6726 = torch.aten.gt.Tensor %6714, %6725 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6727 = torch.aten.where.self %6726, %6725, %6714 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6728 = torch.aten.lt.Tensor %6727, %6719 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %6729 = torch.aten.where.self %6728, %6719, %6727 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6730 = torch.operator "aten.round"(%6729) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_896 = torch.constant.int 1 loc(#loc1) | |
| %6731 = torch.aten.sub.Tensor %6730, %6712, %int1_896 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %6732 = torch.aten.mul.Tensor %6731, %6710 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_897 = torch.constant.int 320 loc(#loc1) | |
| %int20_898 = torch.constant.int 20 loc(#loc1) | |
| %int16_899 = torch.constant.int 16 loc(#loc1) | |
| %6733 = torch.prim.ListConstruct %int320_897, %int20_898, %int16_899 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_900 = torch.constant.bool false loc(#loc1) | |
| %6734 = torch.aten.expand %5824, %6733, %false_900 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_901 = torch.constant.int 0 loc(#loc1) | |
| %6735 = torch.aten.clone %6734, %int0_901 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_902 = torch.constant.int 320 loc(#loc1) | |
| %int320_903 = torch.constant.int 320 loc(#loc1) | |
| %6736 = torch.prim.ListConstruct %int320_902, %int320_903 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6737 = torch.aten._unsafe_view %6735, %6736 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %6738 = torch.aten.mul.Tensor %5823, %6737 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_904 = torch.constant.int 6 loc(#loc1) | |
| %none_905 = torch.constant.none loc(#loc0) | |
| %none_906 = torch.constant.none loc(#loc0) | |
| %none_907 = torch.constant.none loc(#loc0) | |
| %false_908 = torch.constant.bool false loc(#loc1) | |
| %none_909 = torch.constant.none loc(#loc0) | |
| %6739 = torch.aten._to_copy %6738, %int6_904, %none_905, %none_906, %none_907, %false_908, %none_909 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %6740 = torch.aten.t %6739 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_910 = torch.constant.int 8192 loc(#loc1) | |
| %int320_911 = torch.constant.int 320 loc(#loc1) | |
| %6741 = torch.prim.ListConstruct %int8192_910, %int320_911 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6742 = torch.aten.view %6732, %6741 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_912 = torch.constant.int 1 loc(#loc1) | |
| %int1_913 = torch.constant.int 1 loc(#loc1) | |
| %6743 = torch.aten.addmm %5822, %6742, %6740, %int1_912, %int1_913 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_914 = torch.constant.int 2 loc(#loc1) | |
| %int4096_915 = torch.constant.int 4096 loc(#loc1) | |
| %int320_916 = torch.constant.int 320 loc(#loc1) | |
| %6744 = torch.prim.ListConstruct %int2_914, %int4096_915, %int320_916 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6745 = torch.aten.view %6743, %6744 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_917 = torch.constant.int 1 loc(#loc1) | |
| %6746 = torch.aten.add.Tensor %6745, %6464, %int1_917 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_918 = torch.constant.int 320 loc(#loc1) | |
| %6747 = torch.prim.ListConstruct %int320_918 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_919 = torch.constant.none loc(#loc0) | |
| %none_920 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05_921 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0_922, %result1_923, %result2_924 = torch.aten.native_layer_norm %6746, %6747, %none_919, %none_920, %float1.000000e-05_921 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,1],f32>, !torch.tensor<[2,4096,1],f32> loc(#loc1) | |
| %int2560 = torch.constant.int 2560 loc(#loc1) | |
| %int20_925 = torch.constant.int 20 loc(#loc1) | |
| %int16_926 = torch.constant.int 16 loc(#loc1) | |
| %6748 = torch.prim.ListConstruct %int2560, %int20_925, %int16_926 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_927 = torch.constant.bool false loc(#loc1) | |
| %6749 = torch.aten.expand %5821, %6748, %false_927 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[2560,20,16],f16> loc(#loc1) | |
| %int0_928 = torch.constant.int 0 loc(#loc1) | |
| %6750 = torch.aten.clone %6749, %int0_928 : !torch.tensor<[2560,20,16],f16>, !torch.int -> !torch.tensor<[2560,20,16],f16> loc(#loc1) | |
| %int2560_929 = torch.constant.int 2560 loc(#loc1) | |
| %int320_930 = torch.constant.int 320 loc(#loc1) | |
| %6751 = torch.prim.ListConstruct %int2560_929, %int320_930 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6752 = torch.aten._unsafe_view %6750, %6751 : !torch.tensor<[2560,20,16],f16>, !torch.list<int> -> !torch.tensor<[2560,320],f16> loc(#loc1) | |
| %6753 = torch.aten.mul.Tensor %5820, %6752 : !torch.tensor, !torch.tensor<[2560,320],f16> -> !torch.tensor<[2560,320],f16> loc(#loc1) | |
| %int6_931 = torch.constant.int 6 loc(#loc1) | |
| %none_932 = torch.constant.none loc(#loc0) | |
| %none_933 = torch.constant.none loc(#loc0) | |
| %none_934 = torch.constant.none loc(#loc0) | |
| %false_935 = torch.constant.bool false loc(#loc1) | |
| %none_936 = torch.constant.none loc(#loc0) | |
| %6754 = torch.aten._to_copy %6753, %int6_931, %none_932, %none_933, %none_934, %false_935, %none_936 : !torch.tensor<[2560,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[2560,320],f32> loc(#loc1) | |
| %6755 = torch.aten.t %6754 : !torch.tensor<[2560,320],f32> -> !torch.tensor<[320,2560],f32> loc(#loc1) | |
| %int8192_937 = torch.constant.int 8192 loc(#loc1) | |
| %int320_938 = torch.constant.int 320 loc(#loc1) | |
| %6756 = torch.prim.ListConstruct %int8192_937, %int320_938 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6757 = torch.aten.view %result0_922, %6756 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_939 = torch.constant.int 1 loc(#loc1) | |
| %int1_940 = torch.constant.int 1 loc(#loc1) | |
| %6758 = torch.aten.addmm %5819, %6757, %6755, %int1_939, %int1_940 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,2560],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,2560],f32> loc(#loc1) | |
| %int2_941 = torch.constant.int 2 loc(#loc1) | |
| %int4096_942 = torch.constant.int 4096 loc(#loc1) | |
| %int2560_943 = torch.constant.int 2560 loc(#loc1) | |
| %6759 = torch.prim.ListConstruct %int2_941, %int4096_942, %int2560_943 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6760 = torch.aten.view %6758, %6759 : !torch.tensor<[8192,2560],f32>, !torch.list<int> -> !torch.tensor<[2,4096,2560],f32> loc(#loc1) | |
| %int-1_944 = torch.constant.int -1 loc(#loc1) | |
| %int0_945 = torch.constant.int 0 loc(#loc1) | |
| %int1280 = torch.constant.int 1280 loc(#loc1) | |
| %int1_946 = torch.constant.int 1 loc(#loc1) | |
| %6761 = torch.aten.slice.Tensor %6760, %int-1_944, %int0_945, %int1280, %int1_946 : !torch.tensor<[2,4096,2560],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,4096,1280],f32> loc(#loc1) | |
| %int-1_947 = torch.constant.int -1 loc(#loc1) | |
| %int1280_948 = torch.constant.int 1280 loc(#loc1) | |
| %int2560_949 = torch.constant.int 2560 loc(#loc1) | |
| %int1_950 = torch.constant.int 1 loc(#loc1) | |
| %6762 = torch.aten.slice.Tensor %6760, %int-1_947, %int1280_948, %int2560_949, %int1_950 : !torch.tensor<[2,4096,2560],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,4096,1280],f32> loc(#loc1) | |
| %str = torch.constant.str "none" loc(#loc1) | |
| %6763 = torch.aten.gelu %6762, %str : !torch.tensor<[2,4096,1280],f32>, !torch.str -> !torch.tensor<[2,4096,1280],f32> loc(#loc1) | |
| %6764 = torch.aten.mul.Tensor %6761, %6763 : !torch.tensor<[2,4096,1280],f32>, !torch.tensor<[2,4096,1280],f32> -> !torch.tensor<[2,4096,1280],f32> loc(#loc1) | |
| %int320_951 = torch.constant.int 320 loc(#loc1) | |
| %int80 = torch.constant.int 80 loc(#loc1) | |
| %int16_952 = torch.constant.int 16 loc(#loc1) | |
| %6765 = torch.prim.ListConstruct %int320_951, %int80, %int16_952 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_953 = torch.constant.bool false loc(#loc1) | |
| %6766 = torch.aten.expand %5818, %6765, %false_953 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,80,16],f16> loc(#loc1) | |
| %int0_954 = torch.constant.int 0 loc(#loc1) | |
| %6767 = torch.aten.clone %6766, %int0_954 : !torch.tensor<[320,80,16],f16>, !torch.int -> !torch.tensor<[320,80,16],f16> loc(#loc1) | |
| %int320_955 = torch.constant.int 320 loc(#loc1) | |
| %int1280_956 = torch.constant.int 1280 loc(#loc1) | |
| %6768 = torch.prim.ListConstruct %int320_955, %int1280_956 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6769 = torch.aten._unsafe_view %6767, %6768 : !torch.tensor<[320,80,16],f16>, !torch.list<int> -> !torch.tensor<[320,1280],f16> loc(#loc1) | |
| %6770 = torch.aten.mul.Tensor %5817, %6769 : !torch.tensor, !torch.tensor<[320,1280],f16> -> !torch.tensor<[320,1280],f16> loc(#loc1) | |
| %int6_957 = torch.constant.int 6 loc(#loc1) | |
| %none_958 = torch.constant.none loc(#loc0) | |
| %none_959 = torch.constant.none loc(#loc0) | |
| %none_960 = torch.constant.none loc(#loc0) | |
| %false_961 = torch.constant.bool false loc(#loc1) | |
| %none_962 = torch.constant.none loc(#loc0) | |
| %6771 = torch.aten._to_copy %6770, %int6_957, %none_958, %none_959, %none_960, %false_961, %none_962 : !torch.tensor<[320,1280],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,1280],f32> loc(#loc1) | |
| %6772 = torch.aten.t %6771 : !torch.tensor<[320,1280],f32> -> !torch.tensor<[1280,320],f32> loc(#loc1) | |
| %int8192_963 = torch.constant.int 8192 loc(#loc1) | |
| %int1280_964 = torch.constant.int 1280 loc(#loc1) | |
| %6773 = torch.prim.ListConstruct %int8192_963, %int1280_964 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6774 = torch.aten.view %6764, %6773 : !torch.tensor<[2,4096,1280],f32>, !torch.list<int> -> !torch.tensor<[8192,1280],f32> loc(#loc1) | |
| %int1_965 = torch.constant.int 1 loc(#loc1) | |
| %int1_966 = torch.constant.int 1 loc(#loc1) | |
| %6775 = torch.aten.addmm %5816, %6774, %6772, %int1_965, %int1_966 : !torch.tensor, !torch.tensor<[8192,1280],f32>, !torch.tensor<[1280,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_967 = torch.constant.int 2 loc(#loc1) | |
| %int4096_968 = torch.constant.int 4096 loc(#loc1) | |
| %int320_969 = torch.constant.int 320 loc(#loc1) | |
| %6776 = torch.prim.ListConstruct %int2_967, %int4096_968, %int320_969 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6777 = torch.aten.view %6775, %6776 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_970 = torch.constant.int 1 loc(#loc1) | |
| %6778 = torch.aten.add.Tensor %6777, %6746, %int1_970 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int2_971 = torch.constant.int 2 loc(#loc1) | |
| %int64_972 = torch.constant.int 64 loc(#loc1) | |
| %int64_973 = torch.constant.int 64 loc(#loc1) | |
| %int320_974 = torch.constant.int 320 loc(#loc1) | |
| %6779 = torch.prim.ListConstruct %int2_971, %int64_972, %int64_973, %int320_974 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_975 = torch.constant.int 1310720 loc(#loc1) | |
| %int20480 = torch.constant.int 20480 loc(#loc1) | |
| %int320_976 = torch.constant.int 320 loc(#loc1) | |
| %int1_977 = torch.constant.int 1 loc(#loc1) | |
| %6780 = torch.prim.ListConstruct %int1310720_975, %int20480, %int320_976, %int1_977 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6781 = torch.aten._reshape_alias %6778, %6779, %6780 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,64,64,320],f32> loc(#loc1) | |
| %int0_978 = torch.constant.int 0 loc(#loc1) | |
| %int3_979 = torch.constant.int 3 loc(#loc1) | |
| %int1_980 = torch.constant.int 1 loc(#loc1) | |
| %int2_981 = torch.constant.int 2 loc(#loc1) | |
| %6782 = torch.prim.ListConstruct %int0_978, %int3_979, %int1_980, %int2_981 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6783 = torch.aten.permute %6781, %6782 : !torch.tensor<[2,64,64,320],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6784 = torch.aten.detach %5815 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6785 = torch.aten.detach %6784 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_982 = torch.constant.int 2 loc(#loc1) | |
| %int20_983 = torch.constant.int 20 loc(#loc1) | |
| %int16_984 = torch.constant.int 16 loc(#loc1) | |
| %int64_985 = torch.constant.int 64 loc(#loc1) | |
| %int64_986 = torch.constant.int 64 loc(#loc1) | |
| %6786 = torch.prim.ListConstruct %int2_982, %int20_983, %int16_984, %int64_985, %int64_986 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6787 = torch.aten.view %6783, %6786 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %6788 = torch.aten.abs %6787 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_987 = torch.constant.int 2 loc(#loc1) | |
| %true_988 = torch.constant.bool true loc(#loc1) | |
| %values_989, %indices_990 = torch.aten.max.dim %6788, %int2_987, %true_988 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_991 = torch.constant.int 2 loc(#loc1) | |
| %int20_992 = torch.constant.int 20 loc(#loc1) | |
| %int1_993 = torch.constant.int 1 loc(#loc1) | |
| %int64_994 = torch.constant.int 64 loc(#loc1) | |
| %int64_995 = torch.constant.int 64 loc(#loc1) | |
| %6789 = torch.prim.ListConstruct %int2_991, %int20_992, %int1_993, %int64_994, %int64_995 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6790 = torch.aten.view %values_989, %6789 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_996 = torch.constant.int 2 loc(#loc1) | |
| %int20_997 = torch.constant.int 20 loc(#loc1) | |
| %int16_998 = torch.constant.int 16 loc(#loc1) | |
| %int64_999 = torch.constant.int 64 loc(#loc1) | |
| %int64_1000 = torch.constant.int 64 loc(#loc1) | |
| %6791 = torch.prim.ListConstruct %int2_996, %int20_997, %int16_998, %int64_999, %int64_1000 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1001 = torch.constant.bool false loc(#loc1) | |
| %6792 = torch.aten.expand %6790, %6791, %false_1001 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_1002 = torch.constant.int 0 loc(#loc1) | |
| %6793 = torch.aten.clone %6792, %int0_1002 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_1003 = torch.constant.int 2 loc(#loc1) | |
| %int320_1004 = torch.constant.int 320 loc(#loc1) | |
| %int64_1005 = torch.constant.int 64 loc(#loc1) | |
| %int64_1006 = torch.constant.int 64 loc(#loc1) | |
| %6794 = torch.prim.ListConstruct %int2_1003, %int320_1004, %int64_1005, %int64_1006 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6795 = torch.aten._unsafe_view %6793, %6794 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6796 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1007 = torch.constant.int 1 loc(#loc1) | |
| %6797 = torch.aten.sub.Tensor %6785, %6796, %int1_1007 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6798 = torch.aten.lift_fresh_copy %5814 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6799 = torch.aten.pow.Tensor_Tensor %6798, %6797 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6800 = torch.aten.neg %6799 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6801 = torch.aten.neg %6800 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6802 = torch.aten.div.Tensor %6795, %6801 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6803 = torch.aten.detach %5813 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6804 = torch.aten.detach %6803 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6805 = torch.aten.div.Tensor %6783, %6802 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1008 = torch.constant.int 1 loc(#loc1) | |
| %6806 = torch.aten.add.Tensor %6805, %6804, %int1_1008 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6807 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1009 = torch.constant.int 1 loc(#loc1) | |
| %6808 = torch.aten.sub.Tensor %6785, %6807, %int1_1009 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6809 = torch.aten.lift_fresh_copy %5812 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6810 = torch.aten.pow.Tensor_Tensor %6809, %6808 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6811 = torch.aten.neg %6810 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6812 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1010 = torch.constant.int 1 loc(#loc1) | |
| %6813 = torch.aten.sub.Tensor %6785, %6812, %int1_1010 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6814 = torch.aten.lift_fresh_copy %5811 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6815 = torch.aten.pow.Tensor_Tensor %6814, %6813 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6816 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1011 = torch.constant.int 1 loc(#loc1) | |
| %6817 = torch.aten.sub.Tensor %6815, %6816, %int1_1011 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6818 = torch.aten.gt.Tensor %6806, %6817 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6819 = torch.aten.where.self %6818, %6817, %6806 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6820 = torch.aten.lt.Tensor %6819, %6811 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6821 = torch.aten.where.self %6820, %6811, %6819 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6822 = torch.operator "aten.round"(%6821) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1012 = torch.constant.int 1 loc(#loc1) | |
| %6823 = torch.aten.sub.Tensor %6822, %6804, %int1_1012 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6824 = torch.aten.mul.Tensor %6823, %6802 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_1013 = torch.constant.int 320 loc(#loc1) | |
| %int20_1014 = torch.constant.int 20 loc(#loc1) | |
| %int16_1015 = torch.constant.int 16 loc(#loc1) | |
| %int1_1016 = torch.constant.int 1 loc(#loc1) | |
| %int1_1017 = torch.constant.int 1 loc(#loc1) | |
| %6825 = torch.prim.ListConstruct %int320_1013, %int20_1014, %int16_1015, %int1_1016, %int1_1017 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1018 = torch.constant.bool false loc(#loc1) | |
| %6826 = torch.aten.expand %5810, %6825, %false_1018 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,1,1],f16> loc(#loc1) | |
| %int0_1019 = torch.constant.int 0 loc(#loc1) | |
| %6827 = torch.aten.clone %6826, %int0_1019 : !torch.tensor<[320,20,16,1,1],f16>, !torch.int -> !torch.tensor<[320,20,16,1,1],f16> loc(#loc1) | |
| %int320_1020 = torch.constant.int 320 loc(#loc1) | |
| %int320_1021 = torch.constant.int 320 loc(#loc1) | |
| %int1_1022 = torch.constant.int 1 loc(#loc1) | |
| %int1_1023 = torch.constant.int 1 loc(#loc1) | |
| %6828 = torch.prim.ListConstruct %int320_1020, %int320_1021, %int1_1022, %int1_1023 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6829 = torch.aten._unsafe_view %6827, %6828 : !torch.tensor<[320,20,16,1,1],f16>, !torch.list<int> -> !torch.tensor<[320,320,1,1],f16> loc(#loc1) | |
| %6830 = torch.aten.mul.Tensor %5809, %6829 : !torch.tensor, !torch.tensor<[320,320,1,1],f16> -> !torch.tensor<[320,320,1,1],f16> loc(#loc1) | |
| %int6_1024 = torch.constant.int 6 loc(#loc1) | |
| %none_1025 = torch.constant.none loc(#loc0) | |
| %none_1026 = torch.constant.none loc(#loc0) | |
| %none_1027 = torch.constant.none loc(#loc0) | |
| %false_1028 = torch.constant.bool false loc(#loc1) | |
| %none_1029 = torch.constant.none loc(#loc0) | |
| %6831 = torch.aten._to_copy %6830, %int6_1024, %none_1025, %none_1026, %none_1027, %false_1028, %none_1029 : !torch.tensor<[320,320,1,1],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,1,1],f32> loc(#loc1) | |
| %int1_1030 = torch.constant.int 1 loc(#loc1) | |
| %int1_1031 = torch.constant.int 1 loc(#loc1) | |
| %6832 = torch.prim.ListConstruct %int1_1030, %int1_1031 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_1032 = torch.constant.int 0 loc(#loc1) | |
| %int0_1033 = torch.constant.int 0 loc(#loc1) | |
| %6833 = torch.prim.ListConstruct %int0_1032, %int0_1033 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1034 = torch.constant.int 1 loc(#loc1) | |
| %int1_1035 = torch.constant.int 1 loc(#loc1) | |
| %6834 = torch.prim.ListConstruct %int1_1034, %int1_1035 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1036 = torch.constant.bool false loc(#loc1) | |
| %int0_1037 = torch.constant.int 0 loc(#loc1) | |
| %int0_1038 = torch.constant.int 0 loc(#loc1) | |
| %6835 = torch.prim.ListConstruct %int0_1037, %int0_1038 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1039 = torch.constant.int 1 loc(#loc1) | |
| %false_1040 = torch.constant.bool false loc(#loc1) | |
| %false_1041 = torch.constant.bool false loc(#loc1) | |
| %true_1042 = torch.constant.bool true loc(#loc1) | |
| %true_1043 = torch.constant.bool true loc(#loc1) | |
| %6836 = torch.aten._convolution %6824, %6831, %5808, %6832, %6833, %6834, %false_1036, %6835, %int1_1039, %false_1040, %false_1041, %true_1042, %true_1043 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,1,1],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1044 = torch.constant.int 1 loc(#loc1) | |
| %6837 = torch.aten.add.Tensor %6836, %6113, %int1_1044 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int2_1045 = torch.constant.int 2 loc(#loc1) | |
| %int32_1046 = torch.constant.int 32 loc(#loc1) | |
| %int10_1047 = torch.constant.int 10 loc(#loc1) | |
| %int4096_1048 = torch.constant.int 4096 loc(#loc1) | |
| %6838 = torch.prim.ListConstruct %int2_1045, %int32_1046, %int10_1047, %int4096_1048 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6839 = torch.aten.view %6837, %6838 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_1049 = torch.constant.int 2 loc(#loc1) | |
| %int3_1050 = torch.constant.int 3 loc(#loc1) | |
| %6840 = torch.prim.ListConstruct %int2_1049, %int3_1050 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_1051 = torch.constant.int 0 loc(#loc1) | |
| %true_1052 = torch.constant.bool true loc(#loc1) | |
| %6841 = torch.aten.var.correction %6839, %6840, %int0_1051, %true_1052 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_1053 = torch.constant.int 2 loc(#loc1) | |
| %int3_1054 = torch.constant.int 3 loc(#loc1) | |
| %6842 = torch.prim.ListConstruct %int2_1053, %int3_1054 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_1055 = torch.constant.bool true loc(#loc1) | |
| %none_1056 = torch.constant.none loc(#loc0) | |
| %6843 = torch.aten.mean.dim %6839, %6842, %true_1055, %none_1056 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %6844 = torch.tensor.literal(dense<1.000000e-05> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_1057 = torch.constant.int 1 loc(#loc1) | |
| %6845 = torch.aten.add.Tensor %6841, %6844, %int1_1057 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %6846 = torch.aten.rsqrt %6845 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_1058 = torch.constant.int 1 loc(#loc1) | |
| %6847 = torch.aten.sub.Tensor %6839, %6843, %int1_1058 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %6848 = torch.aten.mul.Tensor %6847, %6846 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_1059 = torch.constant.int 2 loc(#loc1) | |
| %int320_1060 = torch.constant.int 320 loc(#loc1) | |
| %int64_1061 = torch.constant.int 64 loc(#loc1) | |
| %int64_1062 = torch.constant.int 64 loc(#loc1) | |
| %6849 = torch.prim.ListConstruct %int2_1059, %int320_1060, %int64_1061, %int64_1062 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6850 = torch.aten.view %6848, %6849 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int-1_1063 = torch.constant.int -1 loc(#loc1) | |
| %6851 = torch.aten.unsqueeze %5807, %int-1_1063 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_1064 = torch.constant.int -1 loc(#loc1) | |
| %6852 = torch.aten.unsqueeze %6851, %int-1_1064 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %6853 = torch.aten.mul.Tensor %6850, %6852 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,1,1],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int-1_1065 = torch.constant.int -1 loc(#loc1) | |
| %6854 = torch.aten.unsqueeze %5806, %int-1_1065 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_1066 = torch.constant.int -1 loc(#loc1) | |
| %6855 = torch.aten.unsqueeze %6854, %int-1_1066 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %int1_1067 = torch.constant.int 1 loc(#loc1) | |
| %6856 = torch.aten.add.Tensor %6853, %6855, %int1_1067 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,1,1],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6857 = torch.aten.silu %6856 : !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6858 = torch.aten.detach %5805 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6859 = torch.aten.detach %6858 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1068 = torch.constant.int 2 loc(#loc1) | |
| %int20_1069 = torch.constant.int 20 loc(#loc1) | |
| %int16_1070 = torch.constant.int 16 loc(#loc1) | |
| %int64_1071 = torch.constant.int 64 loc(#loc1) | |
| %int64_1072 = torch.constant.int 64 loc(#loc1) | |
| %6860 = torch.prim.ListConstruct %int2_1068, %int20_1069, %int16_1070, %int64_1071, %int64_1072 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6861 = torch.aten.view %6857, %6860 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %6862 = torch.aten.abs %6861 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_1073 = torch.constant.int 2 loc(#loc1) | |
| %true_1074 = torch.constant.bool true loc(#loc1) | |
| %values_1075, %indices_1076 = torch.aten.max.dim %6862, %int2_1073, %true_1074 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_1077 = torch.constant.int 2 loc(#loc1) | |
| %int20_1078 = torch.constant.int 20 loc(#loc1) | |
| %int1_1079 = torch.constant.int 1 loc(#loc1) | |
| %int64_1080 = torch.constant.int 64 loc(#loc1) | |
| %int64_1081 = torch.constant.int 64 loc(#loc1) | |
| %6863 = torch.prim.ListConstruct %int2_1077, %int20_1078, %int1_1079, %int64_1080, %int64_1081 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6864 = torch.aten.view %values_1075, %6863 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_1082 = torch.constant.int 2 loc(#loc1) | |
| %int20_1083 = torch.constant.int 20 loc(#loc1) | |
| %int16_1084 = torch.constant.int 16 loc(#loc1) | |
| %int64_1085 = torch.constant.int 64 loc(#loc1) | |
| %int64_1086 = torch.constant.int 64 loc(#loc1) | |
| %6865 = torch.prim.ListConstruct %int2_1082, %int20_1083, %int16_1084, %int64_1085, %int64_1086 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1087 = torch.constant.bool false loc(#loc1) | |
| %6866 = torch.aten.expand %6864, %6865, %false_1087 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_1088 = torch.constant.int 0 loc(#loc1) | |
| %6867 = torch.aten.clone %6866, %int0_1088 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_1089 = torch.constant.int 2 loc(#loc1) | |
| %int320_1090 = torch.constant.int 320 loc(#loc1) | |
| %int64_1091 = torch.constant.int 64 loc(#loc1) | |
| %int64_1092 = torch.constant.int 64 loc(#loc1) | |
| %6868 = torch.prim.ListConstruct %int2_1089, %int320_1090, %int64_1091, %int64_1092 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6869 = torch.aten._unsafe_view %6867, %6868 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6870 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1093 = torch.constant.int 1 loc(#loc1) | |
| %6871 = torch.aten.sub.Tensor %6859, %6870, %int1_1093 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6872 = torch.aten.lift_fresh_copy %5804 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6873 = torch.aten.pow.Tensor_Tensor %6872, %6871 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6874 = torch.aten.neg %6873 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6875 = torch.aten.neg %6874 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6876 = torch.aten.div.Tensor %6869, %6875 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6877 = torch.aten.detach %5803 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6878 = torch.aten.detach %6877 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6879 = torch.aten.div.Tensor %6857, %6876 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1094 = torch.constant.int 1 loc(#loc1) | |
| %6880 = torch.aten.add.Tensor %6879, %6878, %int1_1094 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6881 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1095 = torch.constant.int 1 loc(#loc1) | |
| %6882 = torch.aten.sub.Tensor %6859, %6881, %int1_1095 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6883 = torch.aten.lift_fresh_copy %5802 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6884 = torch.aten.pow.Tensor_Tensor %6883, %6882 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6885 = torch.aten.neg %6884 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6886 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1096 = torch.constant.int 1 loc(#loc1) | |
| %6887 = torch.aten.sub.Tensor %6859, %6886, %int1_1096 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6888 = torch.aten.lift_fresh_copy %5801 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6889 = torch.aten.pow.Tensor_Tensor %6888, %6887 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6890 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1097 = torch.constant.int 1 loc(#loc1) | |
| %6891 = torch.aten.sub.Tensor %6889, %6890, %int1_1097 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6892 = torch.aten.gt.Tensor %6880, %6891 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6893 = torch.aten.where.self %6892, %6891, %6880 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6894 = torch.aten.lt.Tensor %6893, %6885 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6895 = torch.aten.where.self %6894, %6885, %6893 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6896 = torch.operator "aten.round"(%6895) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1098 = torch.constant.int 1 loc(#loc1) | |
| %6897 = torch.aten.sub.Tensor %6896, %6878, %int1_1098 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6898 = torch.aten.mul.Tensor %6897, %6876 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_1099 = torch.constant.int 320 loc(#loc1) | |
| %int20_1100 = torch.constant.int 20 loc(#loc1) | |
| %int16_1101 = torch.constant.int 16 loc(#loc1) | |
| %int3_1102 = torch.constant.int 3 loc(#loc1) | |
| %int3_1103 = torch.constant.int 3 loc(#loc1) | |
| %6899 = torch.prim.ListConstruct %int320_1099, %int20_1100, %int16_1101, %int3_1102, %int3_1103 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1104 = torch.constant.bool false loc(#loc1) | |
| %6900 = torch.aten.expand %5800, %6899, %false_1104 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int0_1105 = torch.constant.int 0 loc(#loc1) | |
| %6901 = torch.aten.clone %6900, %int0_1105 : !torch.tensor<[320,20,16,3,3],f16>, !torch.int -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int320_1106 = torch.constant.int 320 loc(#loc1) | |
| %int320_1107 = torch.constant.int 320 loc(#loc1) | |
| %int3_1108 = torch.constant.int 3 loc(#loc1) | |
| %int3_1109 = torch.constant.int 3 loc(#loc1) | |
| %6902 = torch.prim.ListConstruct %int320_1106, %int320_1107, %int3_1108, %int3_1109 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6903 = torch.aten._unsafe_view %6901, %6902 : !torch.tensor<[320,20,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %6904 = torch.aten.mul.Tensor %5799, %6903 : !torch.tensor, !torch.tensor<[320,320,3,3],f16> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %int6_1110 = torch.constant.int 6 loc(#loc1) | |
| %none_1111 = torch.constant.none loc(#loc0) | |
| %none_1112 = torch.constant.none loc(#loc0) | |
| %none_1113 = torch.constant.none loc(#loc0) | |
| %false_1114 = torch.constant.bool false loc(#loc1) | |
| %none_1115 = torch.constant.none loc(#loc0) | |
| %6905 = torch.aten._to_copy %6904, %int6_1110, %none_1111, %none_1112, %none_1113, %false_1114, %none_1115 : !torch.tensor<[320,320,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,3,3],f32> loc(#loc1) | |
| %int1_1116 = torch.constant.int 1 loc(#loc1) | |
| %int1_1117 = torch.constant.int 1 loc(#loc1) | |
| %6906 = torch.prim.ListConstruct %int1_1116, %int1_1117 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1118 = torch.constant.int 1 loc(#loc1) | |
| %int1_1119 = torch.constant.int 1 loc(#loc1) | |
| %6907 = torch.prim.ListConstruct %int1_1118, %int1_1119 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1120 = torch.constant.int 1 loc(#loc1) | |
| %int1_1121 = torch.constant.int 1 loc(#loc1) | |
| %6908 = torch.prim.ListConstruct %int1_1120, %int1_1121 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1122 = torch.constant.bool false loc(#loc1) | |
| %int0_1123 = torch.constant.int 0 loc(#loc1) | |
| %int0_1124 = torch.constant.int 0 loc(#loc1) | |
| %6909 = torch.prim.ListConstruct %int0_1123, %int0_1124 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1125 = torch.constant.int 1 loc(#loc1) | |
| %false_1126 = torch.constant.bool false loc(#loc1) | |
| %false_1127 = torch.constant.bool false loc(#loc1) | |
| %true_1128 = torch.constant.bool true loc(#loc1) | |
| %true_1129 = torch.constant.bool true loc(#loc1) | |
| %6910 = torch.aten._convolution %6898, %6905, %5798, %6906, %6907, %6908, %false_1122, %6909, %int1_1125, %false_1126, %false_1127, %true_1128, %true_1129 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6911 = torch.aten.silu %5951 : !torch.tensor<[2,1280],f32> -> !torch.tensor<[2,1280],f32> loc(#loc1) | |
| %6912 = torch.aten.t %5797 : !torch.tensor -> !torch.tensor<[1280,320],f32> loc(#loc1) | |
| %int1_1130 = torch.constant.int 1 loc(#loc1) | |
| %int1_1131 = torch.constant.int 1 loc(#loc1) | |
| %6913 = torch.aten.addmm %5796, %6911, %6912, %int1_1130, %int1_1131 : !torch.tensor, !torch.tensor<[2,1280],f32>, !torch.tensor<[1280,320],f32>, !torch.int, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int0_1132 = torch.constant.int 0 loc(#loc1) | |
| %int0_1133 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_1134 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_1135 = torch.constant.int 1 loc(#loc1) | |
| %6914 = torch.aten.slice.Tensor %6913, %int0_1132, %int0_1133, %int9223372036854775807_1134, %int1_1135 : !torch.tensor<[2,320],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int1_1136 = torch.constant.int 1 loc(#loc1) | |
| %int0_1137 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_1138 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_1139 = torch.constant.int 1 loc(#loc1) | |
| %6915 = torch.aten.slice.Tensor %6914, %int1_1136, %int0_1137, %int9223372036854775807_1138, %int1_1139 : !torch.tensor<[2,320],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,320],f32> loc(#loc1) | |
| %int2_1140 = torch.constant.int 2 loc(#loc1) | |
| %6916 = torch.aten.unsqueeze %6915, %int2_1140 : !torch.tensor<[2,320],f32>, !torch.int -> !torch.tensor<[2,320,1],f32> loc(#loc1) | |
| %int3_1141 = torch.constant.int 3 loc(#loc1) | |
| %6917 = torch.aten.unsqueeze %6916, %int3_1141 : !torch.tensor<[2,320,1],f32>, !torch.int -> !torch.tensor<[2,320,1,1],f32> loc(#loc1) | |
| %int1_1142 = torch.constant.int 1 loc(#loc1) | |
| %6918 = torch.aten.add.Tensor %6910, %6917, %int1_1142 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,1,1],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int2_1143 = torch.constant.int 2 loc(#loc1) | |
| %int32_1144 = torch.constant.int 32 loc(#loc1) | |
| %int10_1145 = torch.constant.int 10 loc(#loc1) | |
| %int4096_1146 = torch.constant.int 4096 loc(#loc1) | |
| %6919 = torch.prim.ListConstruct %int2_1143, %int32_1144, %int10_1145, %int4096_1146 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6920 = torch.aten.view %6918, %6919 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_1147 = torch.constant.int 2 loc(#loc1) | |
| %int3_1148 = torch.constant.int 3 loc(#loc1) | |
| %6921 = torch.prim.ListConstruct %int2_1147, %int3_1148 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_1149 = torch.constant.int 0 loc(#loc1) | |
| %true_1150 = torch.constant.bool true loc(#loc1) | |
| %6922 = torch.aten.var.correction %6920, %6921, %int0_1149, %true_1150 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_1151 = torch.constant.int 2 loc(#loc1) | |
| %int3_1152 = torch.constant.int 3 loc(#loc1) | |
| %6923 = torch.prim.ListConstruct %int2_1151, %int3_1152 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_1153 = torch.constant.bool true loc(#loc1) | |
| %none_1154 = torch.constant.none loc(#loc0) | |
| %6924 = torch.aten.mean.dim %6920, %6923, %true_1153, %none_1154 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %6925 = torch.tensor.literal(dense<1.000000e-05> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_1155 = torch.constant.int 1 loc(#loc1) | |
| %6926 = torch.aten.add.Tensor %6922, %6925, %int1_1155 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %6927 = torch.aten.rsqrt %6926 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_1156 = torch.constant.int 1 loc(#loc1) | |
| %6928 = torch.aten.sub.Tensor %6920, %6924, %int1_1156 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %6929 = torch.aten.mul.Tensor %6928, %6927 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_1157 = torch.constant.int 2 loc(#loc1) | |
| %int320_1158 = torch.constant.int 320 loc(#loc1) | |
| %int64_1159 = torch.constant.int 64 loc(#loc1) | |
| %int64_1160 = torch.constant.int 64 loc(#loc1) | |
| %6930 = torch.prim.ListConstruct %int2_1157, %int320_1158, %int64_1159, %int64_1160 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6931 = torch.aten.view %6929, %6930 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int-1_1161 = torch.constant.int -1 loc(#loc1) | |
| %6932 = torch.aten.unsqueeze %5795, %int-1_1161 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_1162 = torch.constant.int -1 loc(#loc1) | |
| %6933 = torch.aten.unsqueeze %6932, %int-1_1162 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %6934 = torch.aten.mul.Tensor %6931, %6933 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,1,1],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int-1_1163 = torch.constant.int -1 loc(#loc1) | |
| %6935 = torch.aten.unsqueeze %5794, %int-1_1163 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_1164 = torch.constant.int -1 loc(#loc1) | |
| %6936 = torch.aten.unsqueeze %6935, %int-1_1164 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %int1_1165 = torch.constant.int 1 loc(#loc1) | |
| %6937 = torch.aten.add.Tensor %6934, %6936, %int1_1165 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,1,1],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6938 = torch.aten.silu %6937 : !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6939 = torch.aten.detach %5793 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6940 = torch.aten.detach %6939 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1166 = torch.constant.int 2 loc(#loc1) | |
| %int20_1167 = torch.constant.int 20 loc(#loc1) | |
| %int16_1168 = torch.constant.int 16 loc(#loc1) | |
| %int64_1169 = torch.constant.int 64 loc(#loc1) | |
| %int64_1170 = torch.constant.int 64 loc(#loc1) | |
| %6941 = torch.prim.ListConstruct %int2_1166, %int20_1167, %int16_1168, %int64_1169, %int64_1170 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6942 = torch.aten.view %6938, %6941 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %6943 = torch.aten.abs %6942 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_1171 = torch.constant.int 2 loc(#loc1) | |
| %true_1172 = torch.constant.bool true loc(#loc1) | |
| %values_1173, %indices_1174 = torch.aten.max.dim %6943, %int2_1171, %true_1172 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_1175 = torch.constant.int 2 loc(#loc1) | |
| %int20_1176 = torch.constant.int 20 loc(#loc1) | |
| %int1_1177 = torch.constant.int 1 loc(#loc1) | |
| %int64_1178 = torch.constant.int 64 loc(#loc1) | |
| %int64_1179 = torch.constant.int 64 loc(#loc1) | |
| %6944 = torch.prim.ListConstruct %int2_1175, %int20_1176, %int1_1177, %int64_1178, %int64_1179 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6945 = torch.aten.view %values_1173, %6944 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_1180 = torch.constant.int 2 loc(#loc1) | |
| %int20_1181 = torch.constant.int 20 loc(#loc1) | |
| %int16_1182 = torch.constant.int 16 loc(#loc1) | |
| %int64_1183 = torch.constant.int 64 loc(#loc1) | |
| %int64_1184 = torch.constant.int 64 loc(#loc1) | |
| %6946 = torch.prim.ListConstruct %int2_1180, %int20_1181, %int16_1182, %int64_1183, %int64_1184 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1185 = torch.constant.bool false loc(#loc1) | |
| %6947 = torch.aten.expand %6945, %6946, %false_1185 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_1186 = torch.constant.int 0 loc(#loc1) | |
| %6948 = torch.aten.clone %6947, %int0_1186 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_1187 = torch.constant.int 2 loc(#loc1) | |
| %int320_1188 = torch.constant.int 320 loc(#loc1) | |
| %int64_1189 = torch.constant.int 64 loc(#loc1) | |
| %int64_1190 = torch.constant.int 64 loc(#loc1) | |
| %6949 = torch.prim.ListConstruct %int2_1187, %int320_1188, %int64_1189, %int64_1190 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6950 = torch.aten._unsafe_view %6948, %6949 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6951 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1191 = torch.constant.int 1 loc(#loc1) | |
| %6952 = torch.aten.sub.Tensor %6940, %6951, %int1_1191 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6953 = torch.aten.lift_fresh_copy %5792 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6954 = torch.aten.pow.Tensor_Tensor %6953, %6952 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6955 = torch.aten.neg %6954 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6956 = torch.aten.neg %6955 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6957 = torch.aten.div.Tensor %6950, %6956 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6958 = torch.aten.detach %5791 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6959 = torch.aten.detach %6958 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6960 = torch.aten.div.Tensor %6938, %6957 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1192 = torch.constant.int 1 loc(#loc1) | |
| %6961 = torch.aten.add.Tensor %6960, %6959, %int1_1192 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6962 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1193 = torch.constant.int 1 loc(#loc1) | |
| %6963 = torch.aten.sub.Tensor %6940, %6962, %int1_1193 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6964 = torch.aten.lift_fresh_copy %5790 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6965 = torch.aten.pow.Tensor_Tensor %6964, %6963 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6966 = torch.aten.neg %6965 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6967 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1194 = torch.constant.int 1 loc(#loc1) | |
| %6968 = torch.aten.sub.Tensor %6940, %6967, %int1_1194 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6969 = torch.aten.lift_fresh_copy %5789 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %6970 = torch.aten.pow.Tensor_Tensor %6969, %6968 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %6971 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1195 = torch.constant.int 1 loc(#loc1) | |
| %6972 = torch.aten.sub.Tensor %6970, %6971, %int1_1195 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %6973 = torch.aten.gt.Tensor %6961, %6972 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6974 = torch.aten.where.self %6973, %6972, %6961 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6975 = torch.aten.lt.Tensor %6974, %6966 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %6976 = torch.aten.where.self %6975, %6966, %6974 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6977 = torch.operator "aten.round"(%6976) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1196 = torch.constant.int 1 loc(#loc1) | |
| %6978 = torch.aten.sub.Tensor %6977, %6959, %int1_1196 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6979 = torch.aten.mul.Tensor %6978, %6957 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_1197 = torch.constant.int 320 loc(#loc1) | |
| %int20_1198 = torch.constant.int 20 loc(#loc1) | |
| %int16_1199 = torch.constant.int 16 loc(#loc1) | |
| %int3_1200 = torch.constant.int 3 loc(#loc1) | |
| %int3_1201 = torch.constant.int 3 loc(#loc1) | |
| %6980 = torch.prim.ListConstruct %int320_1197, %int20_1198, %int16_1199, %int3_1200, %int3_1201 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1202 = torch.constant.bool false loc(#loc1) | |
| %6981 = torch.aten.expand %5788, %6980, %false_1202 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int0_1203 = torch.constant.int 0 loc(#loc1) | |
| %6982 = torch.aten.clone %6981, %int0_1203 : !torch.tensor<[320,20,16,3,3],f16>, !torch.int -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int320_1204 = torch.constant.int 320 loc(#loc1) | |
| %int320_1205 = torch.constant.int 320 loc(#loc1) | |
| %int3_1206 = torch.constant.int 3 loc(#loc1) | |
| %int3_1207 = torch.constant.int 3 loc(#loc1) | |
| %6983 = torch.prim.ListConstruct %int320_1204, %int320_1205, %int3_1206, %int3_1207 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6984 = torch.aten._unsafe_view %6982, %6983 : !torch.tensor<[320,20,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %6985 = torch.aten.mul.Tensor %5787, %6984 : !torch.tensor, !torch.tensor<[320,320,3,3],f16> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %int6_1208 = torch.constant.int 6 loc(#loc1) | |
| %none_1209 = torch.constant.none loc(#loc0) | |
| %none_1210 = torch.constant.none loc(#loc0) | |
| %none_1211 = torch.constant.none loc(#loc0) | |
| %false_1212 = torch.constant.bool false loc(#loc1) | |
| %none_1213 = torch.constant.none loc(#loc0) | |
| %6986 = torch.aten._to_copy %6985, %int6_1208, %none_1209, %none_1210, %none_1211, %false_1212, %none_1213 : !torch.tensor<[320,320,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,3,3],f32> loc(#loc1) | |
| %int1_1214 = torch.constant.int 1 loc(#loc1) | |
| %int1_1215 = torch.constant.int 1 loc(#loc1) | |
| %6987 = torch.prim.ListConstruct %int1_1214, %int1_1215 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1216 = torch.constant.int 1 loc(#loc1) | |
| %int1_1217 = torch.constant.int 1 loc(#loc1) | |
| %6988 = torch.prim.ListConstruct %int1_1216, %int1_1217 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1218 = torch.constant.int 1 loc(#loc1) | |
| %int1_1219 = torch.constant.int 1 loc(#loc1) | |
| %6989 = torch.prim.ListConstruct %int1_1218, %int1_1219 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1220 = torch.constant.bool false loc(#loc1) | |
| %int0_1221 = torch.constant.int 0 loc(#loc1) | |
| %int0_1222 = torch.constant.int 0 loc(#loc1) | |
| %6990 = torch.prim.ListConstruct %int0_1221, %int0_1222 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1223 = torch.constant.int 1 loc(#loc1) | |
| %false_1224 = torch.constant.bool false loc(#loc1) | |
| %false_1225 = torch.constant.bool false loc(#loc1) | |
| %true_1226 = torch.constant.bool true loc(#loc1) | |
| %true_1227 = torch.constant.bool true loc(#loc1) | |
| %6991 = torch.aten._convolution %6979, %6986, %5786, %6987, %6988, %6989, %false_1220, %6990, %int1_1223, %false_1224, %false_1225, %true_1226, %true_1227 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1228 = torch.constant.int 1 loc(#loc1) | |
| %6992 = torch.aten.add.Tensor %6837, %6991, %int1_1228 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %6993 = torch.tensor.literal(dense<1.000000e+00> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %6994 = torch.aten.div.Tensor %6992, %6993 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f64> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int2_1229 = torch.constant.int 2 loc(#loc1) | |
| %int32_1230 = torch.constant.int 32 loc(#loc1) | |
| %int10_1231 = torch.constant.int 10 loc(#loc1) | |
| %int4096_1232 = torch.constant.int 4096 loc(#loc1) | |
| %6995 = torch.prim.ListConstruct %int2_1229, %int32_1230, %int10_1231, %int4096_1232 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %6996 = torch.aten.view %6994, %6995 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_1233 = torch.constant.int 2 loc(#loc1) | |
| %int3_1234 = torch.constant.int 3 loc(#loc1) | |
| %6997 = torch.prim.ListConstruct %int2_1233, %int3_1234 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_1235 = torch.constant.int 0 loc(#loc1) | |
| %true_1236 = torch.constant.bool true loc(#loc1) | |
| %6998 = torch.aten.var.correction %6996, %6997, %int0_1235, %true_1236 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_1237 = torch.constant.int 2 loc(#loc1) | |
| %int3_1238 = torch.constant.int 3 loc(#loc1) | |
| %6999 = torch.prim.ListConstruct %int2_1237, %int3_1238 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_1239 = torch.constant.bool true loc(#loc1) | |
| %none_1240 = torch.constant.none loc(#loc0) | |
| %7000 = torch.aten.mean.dim %6996, %6999, %true_1239, %none_1240 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %7001 = torch.tensor.literal(dense<9.9999999999999995E-7> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_1241 = torch.constant.int 1 loc(#loc1) | |
| %7002 = torch.aten.add.Tensor %6998, %7001, %int1_1241 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %7003 = torch.aten.rsqrt %7002 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_1242 = torch.constant.int 1 loc(#loc1) | |
| %7004 = torch.aten.sub.Tensor %6996, %7000, %int1_1242 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %7005 = torch.aten.mul.Tensor %7004, %7003 : !torch.tensor<[2,32,10,4096],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,10,4096],f32> loc(#loc1) | |
| %int2_1243 = torch.constant.int 2 loc(#loc1) | |
| %int320_1244 = torch.constant.int 320 loc(#loc1) | |
| %int64_1245 = torch.constant.int 64 loc(#loc1) | |
| %int64_1246 = torch.constant.int 64 loc(#loc1) | |
| %7006 = torch.prim.ListConstruct %int2_1243, %int320_1244, %int64_1245, %int64_1246 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7007 = torch.aten.view %7005, %7006 : !torch.tensor<[2,32,10,4096],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7008 = torch.aten.detach %5785 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7009 = torch.aten.detach %7008 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1247 = torch.constant.int 2 loc(#loc1) | |
| %int20_1248 = torch.constant.int 20 loc(#loc1) | |
| %int16_1249 = torch.constant.int 16 loc(#loc1) | |
| %int64_1250 = torch.constant.int 64 loc(#loc1) | |
| %int64_1251 = torch.constant.int 64 loc(#loc1) | |
| %7010 = torch.prim.ListConstruct %int2_1247, %int20_1248, %int16_1249, %int64_1250, %int64_1251 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7011 = torch.aten.view %7007, %7010 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %7012 = torch.aten.abs %7011 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_1252 = torch.constant.int 2 loc(#loc1) | |
| %true_1253 = torch.constant.bool true loc(#loc1) | |
| %values_1254, %indices_1255 = torch.aten.max.dim %7012, %int2_1252, %true_1253 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_1256 = torch.constant.int 2 loc(#loc1) | |
| %int20_1257 = torch.constant.int 20 loc(#loc1) | |
| %int1_1258 = torch.constant.int 1 loc(#loc1) | |
| %int64_1259 = torch.constant.int 64 loc(#loc1) | |
| %int64_1260 = torch.constant.int 64 loc(#loc1) | |
| %7013 = torch.prim.ListConstruct %int2_1256, %int20_1257, %int1_1258, %int64_1259, %int64_1260 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7014 = torch.aten.view %values_1254, %7013 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_1261 = torch.constant.int 2 loc(#loc1) | |
| %int20_1262 = torch.constant.int 20 loc(#loc1) | |
| %int16_1263 = torch.constant.int 16 loc(#loc1) | |
| %int64_1264 = torch.constant.int 64 loc(#loc1) | |
| %int64_1265 = torch.constant.int 64 loc(#loc1) | |
| %7015 = torch.prim.ListConstruct %int2_1261, %int20_1262, %int16_1263, %int64_1264, %int64_1265 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1266 = torch.constant.bool false loc(#loc1) | |
| %7016 = torch.aten.expand %7014, %7015, %false_1266 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_1267 = torch.constant.int 0 loc(#loc1) | |
| %7017 = torch.aten.clone %7016, %int0_1267 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_1268 = torch.constant.int 2 loc(#loc1) | |
| %int320_1269 = torch.constant.int 320 loc(#loc1) | |
| %int64_1270 = torch.constant.int 64 loc(#loc1) | |
| %int64_1271 = torch.constant.int 64 loc(#loc1) | |
| %7018 = torch.prim.ListConstruct %int2_1268, %int320_1269, %int64_1270, %int64_1271 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7019 = torch.aten._unsafe_view %7017, %7018 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7020 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1272 = torch.constant.int 1 loc(#loc1) | |
| %7021 = torch.aten.sub.Tensor %7009, %7020, %int1_1272 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7022 = torch.aten.lift_fresh_copy %5784 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7023 = torch.aten.pow.Tensor_Tensor %7022, %7021 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7024 = torch.aten.neg %7023 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7025 = torch.aten.neg %7024 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7026 = torch.aten.div.Tensor %7019, %7025 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7027 = torch.aten.detach %5783 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7028 = torch.aten.detach %7027 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7029 = torch.aten.div.Tensor %7007, %7026 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1273 = torch.constant.int 1 loc(#loc1) | |
| %7030 = torch.aten.add.Tensor %7029, %7028, %int1_1273 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7031 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1274 = torch.constant.int 1 loc(#loc1) | |
| %7032 = torch.aten.sub.Tensor %7009, %7031, %int1_1274 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7033 = torch.aten.lift_fresh_copy %5782 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7034 = torch.aten.pow.Tensor_Tensor %7033, %7032 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7035 = torch.aten.neg %7034 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7036 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1275 = torch.constant.int 1 loc(#loc1) | |
| %7037 = torch.aten.sub.Tensor %7009, %7036, %int1_1275 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7038 = torch.aten.lift_fresh_copy %5781 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7039 = torch.aten.pow.Tensor_Tensor %7038, %7037 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7040 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1276 = torch.constant.int 1 loc(#loc1) | |
| %7041 = torch.aten.sub.Tensor %7039, %7040, %int1_1276 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7042 = torch.aten.gt.Tensor %7030, %7041 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %7043 = torch.aten.where.self %7042, %7041, %7030 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7044 = torch.aten.lt.Tensor %7043, %7035 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %7045 = torch.aten.where.self %7044, %7035, %7043 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7046 = torch.operator "aten.round"(%7045) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_1277 = torch.constant.int 1 loc(#loc1) | |
| %7047 = torch.aten.sub.Tensor %7046, %7028, %int1_1277 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7048 = torch.aten.mul.Tensor %7047, %7026 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_1278 = torch.constant.int 320 loc(#loc1) | |
| %int20_1279 = torch.constant.int 20 loc(#loc1) | |
| %int16_1280 = torch.constant.int 16 loc(#loc1) | |
| %int1_1281 = torch.constant.int 1 loc(#loc1) | |
| %int1_1282 = torch.constant.int 1 loc(#loc1) | |
| %7049 = torch.prim.ListConstruct %int320_1278, %int20_1279, %int16_1280, %int1_1281, %int1_1282 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1283 = torch.constant.bool false loc(#loc1) | |
| %7050 = torch.aten.expand %5780, %7049, %false_1283 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,1,1],f16> loc(#loc1) | |
| %int0_1284 = torch.constant.int 0 loc(#loc1) | |
| %7051 = torch.aten.clone %7050, %int0_1284 : !torch.tensor<[320,20,16,1,1],f16>, !torch.int -> !torch.tensor<[320,20,16,1,1],f16> loc(#loc1) | |
| %int320_1285 = torch.constant.int 320 loc(#loc1) | |
| %int320_1286 = torch.constant.int 320 loc(#loc1) | |
| %int1_1287 = torch.constant.int 1 loc(#loc1) | |
| %int1_1288 = torch.constant.int 1 loc(#loc1) | |
| %7052 = torch.prim.ListConstruct %int320_1285, %int320_1286, %int1_1287, %int1_1288 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7053 = torch.aten._unsafe_view %7051, %7052 : !torch.tensor<[320,20,16,1,1],f16>, !torch.list<int> -> !torch.tensor<[320,320,1,1],f16> loc(#loc1) | |
| %7054 = torch.aten.mul.Tensor %5779, %7053 : !torch.tensor, !torch.tensor<[320,320,1,1],f16> -> !torch.tensor<[320,320,1,1],f16> loc(#loc1) | |
| %int6_1289 = torch.constant.int 6 loc(#loc1) | |
| %none_1290 = torch.constant.none loc(#loc0) | |
| %none_1291 = torch.constant.none loc(#loc0) | |
| %none_1292 = torch.constant.none loc(#loc0) | |
| %false_1293 = torch.constant.bool false loc(#loc1) | |
| %none_1294 = torch.constant.none loc(#loc0) | |
| %7055 = torch.aten._to_copy %7054, %int6_1289, %none_1290, %none_1291, %none_1292, %false_1293, %none_1294 : !torch.tensor<[320,320,1,1],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,1,1],f32> loc(#loc1) | |
| %int1_1295 = torch.constant.int 1 loc(#loc1) | |
| %int1_1296 = torch.constant.int 1 loc(#loc1) | |
| %7056 = torch.prim.ListConstruct %int1_1295, %int1_1296 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_1297 = torch.constant.int 0 loc(#loc1) | |
| %int0_1298 = torch.constant.int 0 loc(#loc1) | |
| %7057 = torch.prim.ListConstruct %int0_1297, %int0_1298 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1299 = torch.constant.int 1 loc(#loc1) | |
| %int1_1300 = torch.constant.int 1 loc(#loc1) | |
| %7058 = torch.prim.ListConstruct %int1_1299, %int1_1300 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1301 = torch.constant.bool false loc(#loc1) | |
| %int0_1302 = torch.constant.int 0 loc(#loc1) | |
| %int0_1303 = torch.constant.int 0 loc(#loc1) | |
| %7059 = torch.prim.ListConstruct %int0_1302, %int0_1303 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_1304 = torch.constant.int 1 loc(#loc1) | |
| %false_1305 = torch.constant.bool false loc(#loc1) | |
| %false_1306 = torch.constant.bool false loc(#loc1) | |
| %true_1307 = torch.constant.bool true loc(#loc1) | |
| %true_1308 = torch.constant.bool true loc(#loc1) | |
| %7060 = torch.aten._convolution %7048, %7055, %5778, %7056, %7057, %7058, %false_1301, %7059, %int1_1304, %false_1305, %false_1306, %true_1307, %true_1308 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,1,1],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int0_1309 = torch.constant.int 0 loc(#loc1) | |
| %int2_1310 = torch.constant.int 2 loc(#loc1) | |
| %int3_1311 = torch.constant.int 3 loc(#loc1) | |
| %int1_1312 = torch.constant.int 1 loc(#loc1) | |
| %7061 = torch.prim.ListConstruct %int0_1309, %int2_1310, %int3_1311, %int1_1312 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7062 = torch.aten.permute %7060, %7061 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,64,64,320],f32> loc(#loc1) | |
| %int2_1313 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1314 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1315 = torch.constant.int 320 loc(#loc1) | |
| %7063 = torch.prim.ListConstruct %int2_1313, %int4096_1314, %int320_1315 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_1316 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_1317 = torch.constant.int 320 loc(#loc1) | |
| %int1_1318 = torch.constant.int 1 loc(#loc1) | |
| %7064 = torch.prim.ListConstruct %int1310720_1316, %int320_1317, %int1_1318 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7065 = torch.aten._reshape_alias %7062, %7063, %7064 : !torch.tensor<[2,64,64,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1319 = torch.constant.int 320 loc(#loc1) | |
| %7066 = torch.prim.ListConstruct %int320_1319 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_1320 = torch.constant.none loc(#loc0) | |
| %none_1321 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05_1322 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0_1323, %result1_1324, %result2_1325 = torch.aten.native_layer_norm %7065, %7066, %none_1320, %none_1321, %float1.000000e-05_1322 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,1],f32>, !torch.tensor<[2,4096,1],f32> loc(#loc1) | |
| %7067 = torch.aten.detach %5777 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7068 = torch.aten.detach %7067 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1326 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1327 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1328 = torch.constant.int 20 loc(#loc1) | |
| %int16_1329 = torch.constant.int 16 loc(#loc1) | |
| %7069 = torch.prim.ListConstruct %int2_1326, %int4096_1327, %int20_1328, %int16_1329 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7070 = torch.aten.view %result0_1323, %7069 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %7071 = torch.aten.abs %7070 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_1330 = torch.constant.int 3 loc(#loc1) | |
| %true_1331 = torch.constant.bool true loc(#loc1) | |
| %values_1332, %indices_1333 = torch.aten.max.dim %7071, %int3_1330, %true_1331 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_1334 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1335 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1336 = torch.constant.int 20 loc(#loc1) | |
| %int1_1337 = torch.constant.int 1 loc(#loc1) | |
| %7072 = torch.prim.ListConstruct %int2_1334, %int4096_1335, %int20_1336, %int1_1337 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7073 = torch.aten.view %values_1332, %7072 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_1338 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1339 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1340 = torch.constant.int 20 loc(#loc1) | |
| %int16_1341 = torch.constant.int 16 loc(#loc1) | |
| %7074 = torch.prim.ListConstruct %int2_1338, %int4096_1339, %int20_1340, %int16_1341 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1342 = torch.constant.bool false loc(#loc1) | |
| %7075 = torch.aten.expand %7073, %7074, %false_1342 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_1343 = torch.constant.int 0 loc(#loc1) | |
| %7076 = torch.aten.clone %7075, %int0_1343 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_1344 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1345 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1346 = torch.constant.int 320 loc(#loc1) | |
| %7077 = torch.prim.ListConstruct %int2_1344, %int4096_1345, %int320_1346 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7078 = torch.aten._unsafe_view %7076, %7077 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7079 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1347 = torch.constant.int 1 loc(#loc1) | |
| %7080 = torch.aten.sub.Tensor %7068, %7079, %int1_1347 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7081 = torch.aten.lift_fresh_copy %5776 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7082 = torch.aten.pow.Tensor_Tensor %7081, %7080 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7083 = torch.aten.neg %7082 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7084 = torch.aten.neg %7083 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7085 = torch.aten.div.Tensor %7078, %7084 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7086 = torch.aten.detach %5775 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7087 = torch.aten.detach %7086 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7088 = torch.aten.div.Tensor %result0_1323, %7085 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1348 = torch.constant.int 1 loc(#loc1) | |
| %7089 = torch.aten.add.Tensor %7088, %7087, %int1_1348 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7090 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1349 = torch.constant.int 1 loc(#loc1) | |
| %7091 = torch.aten.sub.Tensor %7068, %7090, %int1_1349 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7092 = torch.aten.lift_fresh_copy %5774 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7093 = torch.aten.pow.Tensor_Tensor %7092, %7091 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7094 = torch.aten.neg %7093 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7095 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1350 = torch.constant.int 1 loc(#loc1) | |
| %7096 = torch.aten.sub.Tensor %7068, %7095, %int1_1350 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7097 = torch.aten.lift_fresh_copy %5773 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7098 = torch.aten.pow.Tensor_Tensor %7097, %7096 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7099 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1351 = torch.constant.int 1 loc(#loc1) | |
| %7100 = torch.aten.sub.Tensor %7098, %7099, %int1_1351 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7101 = torch.aten.gt.Tensor %7089, %7100 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7102 = torch.aten.where.self %7101, %7100, %7089 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7103 = torch.aten.lt.Tensor %7102, %7094 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7104 = torch.aten.where.self %7103, %7094, %7102 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7105 = torch.operator "aten.round"(%7104) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1352 = torch.constant.int 1 loc(#loc1) | |
| %7106 = torch.aten.sub.Tensor %7105, %7087, %int1_1352 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7107 = torch.aten.mul.Tensor %7106, %7085 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1353 = torch.constant.int 320 loc(#loc1) | |
| %int20_1354 = torch.constant.int 20 loc(#loc1) | |
| %int16_1355 = torch.constant.int 16 loc(#loc1) | |
| %7108 = torch.prim.ListConstruct %int320_1353, %int20_1354, %int16_1355 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1356 = torch.constant.bool false loc(#loc1) | |
| %7109 = torch.aten.expand %5772, %7108, %false_1356 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_1357 = torch.constant.int 0 loc(#loc1) | |
| %7110 = torch.aten.clone %7109, %int0_1357 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_1358 = torch.constant.int 320 loc(#loc1) | |
| %int320_1359 = torch.constant.int 320 loc(#loc1) | |
| %7111 = torch.prim.ListConstruct %int320_1358, %int320_1359 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7112 = torch.aten._unsafe_view %7110, %7111 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %7113 = torch.aten.mul.Tensor %5771, %7112 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_1360 = torch.constant.int 6 loc(#loc1) | |
| %none_1361 = torch.constant.none loc(#loc0) | |
| %none_1362 = torch.constant.none loc(#loc0) | |
| %none_1363 = torch.constant.none loc(#loc0) | |
| %false_1364 = torch.constant.bool false loc(#loc1) | |
| %none_1365 = torch.constant.none loc(#loc0) | |
| %7114 = torch.aten._to_copy %7113, %int6_1360, %none_1361, %none_1362, %none_1363, %false_1364, %none_1365 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %7115 = torch.aten.t %7114 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_1366 = torch.constant.int 8192 loc(#loc1) | |
| %int320_1367 = torch.constant.int 320 loc(#loc1) | |
| %7116 = torch.prim.ListConstruct %int8192_1366, %int320_1367 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7117 = torch.aten.view %7107, %7116 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_1368 = torch.constant.int 1 loc(#loc1) | |
| %int1_1369 = torch.constant.int 1 loc(#loc1) | |
| %7118 = torch.aten.addmm %5770, %7117, %7115, %int1_1368, %int1_1369 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_1370 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1371 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1372 = torch.constant.int 320 loc(#loc1) | |
| %7119 = torch.prim.ListConstruct %int2_1370, %int4096_1371, %int320_1372 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7120 = torch.aten.view %7118, %7119 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7121 = torch.aten.detach %5769 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7122 = torch.aten.detach %7121 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1373 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1374 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1375 = torch.constant.int 20 loc(#loc1) | |
| %int16_1376 = torch.constant.int 16 loc(#loc1) | |
| %7123 = torch.prim.ListConstruct %int2_1373, %int4096_1374, %int20_1375, %int16_1376 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7124 = torch.aten.view %result0_1323, %7123 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %7125 = torch.aten.abs %7124 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_1377 = torch.constant.int 3 loc(#loc1) | |
| %true_1378 = torch.constant.bool true loc(#loc1) | |
| %values_1379, %indices_1380 = torch.aten.max.dim %7125, %int3_1377, %true_1378 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_1381 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1382 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1383 = torch.constant.int 20 loc(#loc1) | |
| %int1_1384 = torch.constant.int 1 loc(#loc1) | |
| %7126 = torch.prim.ListConstruct %int2_1381, %int4096_1382, %int20_1383, %int1_1384 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7127 = torch.aten.view %values_1379, %7126 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_1385 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1386 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1387 = torch.constant.int 20 loc(#loc1) | |
| %int16_1388 = torch.constant.int 16 loc(#loc1) | |
| %7128 = torch.prim.ListConstruct %int2_1385, %int4096_1386, %int20_1387, %int16_1388 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1389 = torch.constant.bool false loc(#loc1) | |
| %7129 = torch.aten.expand %7127, %7128, %false_1389 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_1390 = torch.constant.int 0 loc(#loc1) | |
| %7130 = torch.aten.clone %7129, %int0_1390 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_1391 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1392 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1393 = torch.constant.int 320 loc(#loc1) | |
| %7131 = torch.prim.ListConstruct %int2_1391, %int4096_1392, %int320_1393 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7132 = torch.aten._unsafe_view %7130, %7131 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7133 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1394 = torch.constant.int 1 loc(#loc1) | |
| %7134 = torch.aten.sub.Tensor %7122, %7133, %int1_1394 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7135 = torch.aten.lift_fresh_copy %5768 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7136 = torch.aten.pow.Tensor_Tensor %7135, %7134 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7137 = torch.aten.neg %7136 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7138 = torch.aten.neg %7137 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7139 = torch.aten.div.Tensor %7132, %7138 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7140 = torch.aten.detach %5767 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7141 = torch.aten.detach %7140 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7142 = torch.aten.div.Tensor %result0_1323, %7139 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1395 = torch.constant.int 1 loc(#loc1) | |
| %7143 = torch.aten.add.Tensor %7142, %7141, %int1_1395 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7144 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1396 = torch.constant.int 1 loc(#loc1) | |
| %7145 = torch.aten.sub.Tensor %7122, %7144, %int1_1396 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7146 = torch.aten.lift_fresh_copy %5766 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7147 = torch.aten.pow.Tensor_Tensor %7146, %7145 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7148 = torch.aten.neg %7147 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7149 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1397 = torch.constant.int 1 loc(#loc1) | |
| %7150 = torch.aten.sub.Tensor %7122, %7149, %int1_1397 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7151 = torch.aten.lift_fresh_copy %5765 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7152 = torch.aten.pow.Tensor_Tensor %7151, %7150 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7153 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1398 = torch.constant.int 1 loc(#loc1) | |
| %7154 = torch.aten.sub.Tensor %7152, %7153, %int1_1398 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7155 = torch.aten.gt.Tensor %7143, %7154 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7156 = torch.aten.where.self %7155, %7154, %7143 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7157 = torch.aten.lt.Tensor %7156, %7148 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7158 = torch.aten.where.self %7157, %7148, %7156 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7159 = torch.operator "aten.round"(%7158) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1399 = torch.constant.int 1 loc(#loc1) | |
| %7160 = torch.aten.sub.Tensor %7159, %7141, %int1_1399 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7161 = torch.aten.mul.Tensor %7160, %7139 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1400 = torch.constant.int 320 loc(#loc1) | |
| %int20_1401 = torch.constant.int 20 loc(#loc1) | |
| %int16_1402 = torch.constant.int 16 loc(#loc1) | |
| %7162 = torch.prim.ListConstruct %int320_1400, %int20_1401, %int16_1402 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1403 = torch.constant.bool false loc(#loc1) | |
| %7163 = torch.aten.expand %5764, %7162, %false_1403 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_1404 = torch.constant.int 0 loc(#loc1) | |
| %7164 = torch.aten.clone %7163, %int0_1404 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_1405 = torch.constant.int 320 loc(#loc1) | |
| %int320_1406 = torch.constant.int 320 loc(#loc1) | |
| %7165 = torch.prim.ListConstruct %int320_1405, %int320_1406 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7166 = torch.aten._unsafe_view %7164, %7165 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %7167 = torch.aten.mul.Tensor %5763, %7166 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_1407 = torch.constant.int 6 loc(#loc1) | |
| %none_1408 = torch.constant.none loc(#loc0) | |
| %none_1409 = torch.constant.none loc(#loc0) | |
| %none_1410 = torch.constant.none loc(#loc0) | |
| %false_1411 = torch.constant.bool false loc(#loc1) | |
| %none_1412 = torch.constant.none loc(#loc0) | |
| %7168 = torch.aten._to_copy %7167, %int6_1407, %none_1408, %none_1409, %none_1410, %false_1411, %none_1412 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %7169 = torch.aten.t %7168 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_1413 = torch.constant.int 8192 loc(#loc1) | |
| %int320_1414 = torch.constant.int 320 loc(#loc1) | |
| %7170 = torch.prim.ListConstruct %int8192_1413, %int320_1414 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7171 = torch.aten.view %7161, %7170 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_1415 = torch.constant.int 1 loc(#loc1) | |
| %int1_1416 = torch.constant.int 1 loc(#loc1) | |
| %7172 = torch.aten.addmm %5762, %7171, %7169, %int1_1415, %int1_1416 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_1417 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1418 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1419 = torch.constant.int 320 loc(#loc1) | |
| %7173 = torch.prim.ListConstruct %int2_1417, %int4096_1418, %int320_1419 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7174 = torch.aten.view %7172, %7173 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7175 = torch.aten.detach %5761 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7176 = torch.aten.detach %7175 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1420 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1421 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1422 = torch.constant.int 20 loc(#loc1) | |
| %int16_1423 = torch.constant.int 16 loc(#loc1) | |
| %7177 = torch.prim.ListConstruct %int2_1420, %int4096_1421, %int20_1422, %int16_1423 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7178 = torch.aten.view %result0_1323, %7177 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %7179 = torch.aten.abs %7178 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_1424 = torch.constant.int 3 loc(#loc1) | |
| %true_1425 = torch.constant.bool true loc(#loc1) | |
| %values_1426, %indices_1427 = torch.aten.max.dim %7179, %int3_1424, %true_1425 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_1428 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1429 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1430 = torch.constant.int 20 loc(#loc1) | |
| %int1_1431 = torch.constant.int 1 loc(#loc1) | |
| %7180 = torch.prim.ListConstruct %int2_1428, %int4096_1429, %int20_1430, %int1_1431 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7181 = torch.aten.view %values_1426, %7180 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_1432 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1433 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1434 = torch.constant.int 20 loc(#loc1) | |
| %int16_1435 = torch.constant.int 16 loc(#loc1) | |
| %7182 = torch.prim.ListConstruct %int2_1432, %int4096_1433, %int20_1434, %int16_1435 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1436 = torch.constant.bool false loc(#loc1) | |
| %7183 = torch.aten.expand %7181, %7182, %false_1436 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_1437 = torch.constant.int 0 loc(#loc1) | |
| %7184 = torch.aten.clone %7183, %int0_1437 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_1438 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1439 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1440 = torch.constant.int 320 loc(#loc1) | |
| %7185 = torch.prim.ListConstruct %int2_1438, %int4096_1439, %int320_1440 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7186 = torch.aten._unsafe_view %7184, %7185 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7187 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1441 = torch.constant.int 1 loc(#loc1) | |
| %7188 = torch.aten.sub.Tensor %7176, %7187, %int1_1441 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7189 = torch.aten.lift_fresh_copy %5760 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7190 = torch.aten.pow.Tensor_Tensor %7189, %7188 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7191 = torch.aten.neg %7190 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7192 = torch.aten.neg %7191 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7193 = torch.aten.div.Tensor %7186, %7192 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7194 = torch.aten.detach %5759 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7195 = torch.aten.detach %7194 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7196 = torch.aten.div.Tensor %result0_1323, %7193 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1442 = torch.constant.int 1 loc(#loc1) | |
| %7197 = torch.aten.add.Tensor %7196, %7195, %int1_1442 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7198 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1443 = torch.constant.int 1 loc(#loc1) | |
| %7199 = torch.aten.sub.Tensor %7176, %7198, %int1_1443 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7200 = torch.aten.lift_fresh_copy %5758 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7201 = torch.aten.pow.Tensor_Tensor %7200, %7199 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7202 = torch.aten.neg %7201 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7203 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1444 = torch.constant.int 1 loc(#loc1) | |
| %7204 = torch.aten.sub.Tensor %7176, %7203, %int1_1444 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7205 = torch.aten.lift_fresh_copy %5757 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7206 = torch.aten.pow.Tensor_Tensor %7205, %7204 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7207 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1445 = torch.constant.int 1 loc(#loc1) | |
| %7208 = torch.aten.sub.Tensor %7206, %7207, %int1_1445 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7209 = torch.aten.gt.Tensor %7197, %7208 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7210 = torch.aten.where.self %7209, %7208, %7197 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7211 = torch.aten.lt.Tensor %7210, %7202 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7212 = torch.aten.where.self %7211, %7202, %7210 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7213 = torch.operator "aten.round"(%7212) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1446 = torch.constant.int 1 loc(#loc1) | |
| %7214 = torch.aten.sub.Tensor %7213, %7195, %int1_1446 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7215 = torch.aten.mul.Tensor %7214, %7193 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1447 = torch.constant.int 320 loc(#loc1) | |
| %int20_1448 = torch.constant.int 20 loc(#loc1) | |
| %int16_1449 = torch.constant.int 16 loc(#loc1) | |
| %7216 = torch.prim.ListConstruct %int320_1447, %int20_1448, %int16_1449 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1450 = torch.constant.bool false loc(#loc1) | |
| %7217 = torch.aten.expand %5756, %7216, %false_1450 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_1451 = torch.constant.int 0 loc(#loc1) | |
| %7218 = torch.aten.clone %7217, %int0_1451 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_1452 = torch.constant.int 320 loc(#loc1) | |
| %int320_1453 = torch.constant.int 320 loc(#loc1) | |
| %7219 = torch.prim.ListConstruct %int320_1452, %int320_1453 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7220 = torch.aten._unsafe_view %7218, %7219 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %7221 = torch.aten.mul.Tensor %5755, %7220 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_1454 = torch.constant.int 6 loc(#loc1) | |
| %none_1455 = torch.constant.none loc(#loc0) | |
| %none_1456 = torch.constant.none loc(#loc0) | |
| %none_1457 = torch.constant.none loc(#loc0) | |
| %false_1458 = torch.constant.bool false loc(#loc1) | |
| %none_1459 = torch.constant.none loc(#loc0) | |
| %7222 = torch.aten._to_copy %7221, %int6_1454, %none_1455, %none_1456, %none_1457, %false_1458, %none_1459 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %7223 = torch.aten.t %7222 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_1460 = torch.constant.int 8192 loc(#loc1) | |
| %int320_1461 = torch.constant.int 320 loc(#loc1) | |
| %7224 = torch.prim.ListConstruct %int8192_1460, %int320_1461 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7225 = torch.aten.view %7215, %7224 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_1462 = torch.constant.int 1 loc(#loc1) | |
| %int1_1463 = torch.constant.int 1 loc(#loc1) | |
| %7226 = torch.aten.addmm %5754, %7225, %7223, %int1_1462, %int1_1463 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_1464 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1465 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1466 = torch.constant.int 320 loc(#loc1) | |
| %7227 = torch.prim.ListConstruct %int2_1464, %int4096_1465, %int320_1466 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7228 = torch.aten.view %7226, %7227 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int2_1467 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1468 = torch.constant.int 4096 loc(#loc1) | |
| %int8_1469 = torch.constant.int 8 loc(#loc1) | |
| %int40_1470 = torch.constant.int 40 loc(#loc1) | |
| %7229 = torch.prim.ListConstruct %int2_1467, %int4096_1468, %int8_1469, %int40_1470 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_1471 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_1472 = torch.constant.int 320 loc(#loc1) | |
| %int40_1473 = torch.constant.int 40 loc(#loc1) | |
| %int1_1474 = torch.constant.int 1 loc(#loc1) | |
| %7230 = torch.prim.ListConstruct %int1310720_1471, %int320_1472, %int40_1473, %int1_1474 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7231 = torch.aten._reshape_alias %7120, %7229, %7230 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_1475 = torch.constant.int 0 loc(#loc1) | |
| %int2_1476 = torch.constant.int 2 loc(#loc1) | |
| %int1_1477 = torch.constant.int 1 loc(#loc1) | |
| %int3_1478 = torch.constant.int 3 loc(#loc1) | |
| %7232 = torch.prim.ListConstruct %int0_1475, %int2_1476, %int1_1477, %int3_1478 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7233 = torch.aten.permute %7231, %7232 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_1479 = torch.constant.int 0 loc(#loc1) | |
| %7234 = torch.aten.clone %7233, %int0_1479 : !torch.tensor<[2,8,4096,40],f32>, !torch.int -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int16_1480 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1481 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1482 = torch.constant.int 40 loc(#loc1) | |
| %7235 = torch.prim.ListConstruct %int16_1480, %int4096_1481, %int40_1482 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7236 = torch.aten._unsafe_view %7234, %7235 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_1483 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1484 = torch.constant.int 4096 loc(#loc1) | |
| %int8_1485 = torch.constant.int 8 loc(#loc1) | |
| %int40_1486 = torch.constant.int 40 loc(#loc1) | |
| %7237 = torch.prim.ListConstruct %int2_1483, %int4096_1484, %int8_1485, %int40_1486 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_1487 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_1488 = torch.constant.int 320 loc(#loc1) | |
| %int40_1489 = torch.constant.int 40 loc(#loc1) | |
| %int1_1490 = torch.constant.int 1 loc(#loc1) | |
| %7238 = torch.prim.ListConstruct %int1310720_1487, %int320_1488, %int40_1489, %int1_1490 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7239 = torch.aten._reshape_alias %7174, %7237, %7238 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_1491 = torch.constant.int 0 loc(#loc1) | |
| %int2_1492 = torch.constant.int 2 loc(#loc1) | |
| %int1_1493 = torch.constant.int 1 loc(#loc1) | |
| %int3_1494 = torch.constant.int 3 loc(#loc1) | |
| %7240 = torch.prim.ListConstruct %int0_1491, %int2_1492, %int1_1493, %int3_1494 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7241 = torch.aten.permute %7239, %7240 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_1495 = torch.constant.int 0 loc(#loc1) | |
| %7242 = torch.aten.clone %7241, %int0_1495 : !torch.tensor<[2,8,4096,40],f32>, !torch.int -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int16_1496 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1497 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1498 = torch.constant.int 40 loc(#loc1) | |
| %7243 = torch.prim.ListConstruct %int16_1496, %int4096_1497, %int40_1498 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7244 = torch.aten._unsafe_view %7242, %7243 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_1499 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1500 = torch.constant.int 4096 loc(#loc1) | |
| %int8_1501 = torch.constant.int 8 loc(#loc1) | |
| %int40_1502 = torch.constant.int 40 loc(#loc1) | |
| %7245 = torch.prim.ListConstruct %int2_1499, %int4096_1500, %int8_1501, %int40_1502 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_1503 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_1504 = torch.constant.int 320 loc(#loc1) | |
| %int40_1505 = torch.constant.int 40 loc(#loc1) | |
| %int1_1506 = torch.constant.int 1 loc(#loc1) | |
| %7246 = torch.prim.ListConstruct %int1310720_1503, %int320_1504, %int40_1505, %int1_1506 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7247 = torch.aten._reshape_alias %7228, %7245, %7246 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_1507 = torch.constant.int 0 loc(#loc1) | |
| %int2_1508 = torch.constant.int 2 loc(#loc1) | |
| %int1_1509 = torch.constant.int 1 loc(#loc1) | |
| %int3_1510 = torch.constant.int 3 loc(#loc1) | |
| %7248 = torch.prim.ListConstruct %int0_1507, %int2_1508, %int1_1509, %int3_1510 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7249 = torch.aten.permute %7247, %7248 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_1511 = torch.constant.int 0 loc(#loc1) | |
| %7250 = torch.aten.clone %7249, %int0_1511 : !torch.tensor<[2,8,4096,40],f32>, !torch.int -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int16_1512 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1513 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1514 = torch.constant.int 40 loc(#loc1) | |
| %7251 = torch.prim.ListConstruct %int16_1512, %int4096_1513, %int40_1514 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7252 = torch.aten._unsafe_view %7250, %7251 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int-1_1515 = torch.constant.int -1 loc(#loc1) | |
| %int-2_1516 = torch.constant.int -2 loc(#loc1) | |
| %7253 = torch.aten.transpose.int %7244, %int-1_1515, %int-2_1516 : !torch.tensor<[16,4096,40],f32>, !torch.int, !torch.int -> !torch.tensor<[16,40,4096],f32> loc(#loc1) | |
| %int16_1517 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1518 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1519 = torch.constant.int 40 loc(#loc1) | |
| %7254 = torch.prim.ListConstruct %int16_1517, %int4096_1518, %int40_1519 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1520 = torch.constant.bool false loc(#loc1) | |
| %7255 = torch.aten.expand %7236, %7254, %false_1520 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_1521 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1522 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1523 = torch.constant.int 40 loc(#loc1) | |
| %7256 = torch.prim.ListConstruct %int16_1521, %int4096_1522, %int40_1523 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int163840_1524 = torch.constant.int 163840 loc(#loc1) | |
| %int40_1525 = torch.constant.int 40 loc(#loc1) | |
| %int1_1526 = torch.constant.int 1 loc(#loc1) | |
| %7257 = torch.prim.ListConstruct %int163840_1524, %int40_1525, %int1_1526 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7258 = torch.aten._reshape_alias %7255, %7256, %7257 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_1527 = torch.constant.int 16 loc(#loc1) | |
| %int40_1528 = torch.constant.int 40 loc(#loc1) | |
| %int4096_1529 = torch.constant.int 4096 loc(#loc1) | |
| %7259 = torch.prim.ListConstruct %int16_1527, %int40_1528, %int4096_1529 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1530 = torch.constant.bool false loc(#loc1) | |
| %7260 = torch.aten.expand %7253, %7259, %false_1530 : !torch.tensor<[16,40,4096],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,40,4096],f32> loc(#loc1) | |
| %int16_1531 = torch.constant.int 16 loc(#loc1) | |
| %int40_1532 = torch.constant.int 40 loc(#loc1) | |
| %int4096_1533 = torch.constant.int 4096 loc(#loc1) | |
| %7261 = torch.prim.ListConstruct %int16_1531, %int40_1532, %int4096_1533 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int163840_1534 = torch.constant.int 163840 loc(#loc1) | |
| %int1_1535 = torch.constant.int 1 loc(#loc1) | |
| %int40_1536 = torch.constant.int 40 loc(#loc1) | |
| %7262 = torch.prim.ListConstruct %int163840_1534, %int1_1535, %int40_1536 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7263 = torch.aten._reshape_alias %7260, %7261, %7262 : !torch.tensor<[16,40,4096],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,40,4096],f32> loc(#loc1) | |
| %7264 = torch.aten.bmm %7258, %7263 : !torch.tensor<[16,4096,40],f32>, !torch.tensor<[16,40,4096],f32> -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int16_1537 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1538 = torch.constant.int 4096 loc(#loc1) | |
| %int4096_1539 = torch.constant.int 4096 loc(#loc1) | |
| %7265 = torch.prim.ListConstruct %int16_1537, %int4096_1538, %int4096_1539 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7266 = torch.aten._unsafe_view %7264, %7265 : !torch.tensor<[16,4096,4096],f32>, !torch.list<int> -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %7267 = torch.tensor.literal(dense<0.15811388300841897> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %7268 = torch.aten.mul.Tensor %7266, %7267 : !torch.tensor<[16,4096,4096],f32>, !torch.tensor<[],f64> -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int-1_1540 = torch.constant.int -1 loc(#loc1) | |
| %false_1541 = torch.constant.bool false loc(#loc1) | |
| %7269 = torch.aten._softmax %7268, %int-1_1540, %false_1541 : !torch.tensor<[16,4096,4096],f32>, !torch.int, !torch.bool -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int16_1542 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1543 = torch.constant.int 4096 loc(#loc1) | |
| %int4096_1544 = torch.constant.int 4096 loc(#loc1) | |
| %7270 = torch.prim.ListConstruct %int16_1542, %int4096_1543, %int4096_1544 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1545 = torch.constant.bool false loc(#loc1) | |
| %7271 = torch.aten.expand %7269, %7270, %false_1545 : !torch.tensor<[16,4096,4096],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int16_1546 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1547 = torch.constant.int 4096 loc(#loc1) | |
| %int4096_1548 = torch.constant.int 4096 loc(#loc1) | |
| %7272 = torch.prim.ListConstruct %int16_1546, %int4096_1547, %int4096_1548 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int16777216_1549 = torch.constant.int 16777216 loc(#loc1) | |
| %int4096_1550 = torch.constant.int 4096 loc(#loc1) | |
| %int1_1551 = torch.constant.int 1 loc(#loc1) | |
| %7273 = torch.prim.ListConstruct %int16777216_1549, %int4096_1550, %int1_1551 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7274 = torch.aten._reshape_alias %7271, %7272, %7273 : !torch.tensor<[16,4096,4096],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,4096],f32> loc(#loc1) | |
| %int16_1552 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1553 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1554 = torch.constant.int 40 loc(#loc1) | |
| %7275 = torch.prim.ListConstruct %int16_1552, %int4096_1553, %int40_1554 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1555 = torch.constant.bool false loc(#loc1) | |
| %7276 = torch.aten.expand %7252, %7275, %false_1555 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_1556 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1557 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1558 = torch.constant.int 40 loc(#loc1) | |
| %7277 = torch.prim.ListConstruct %int16_1556, %int4096_1557, %int40_1558 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int163840_1559 = torch.constant.int 163840 loc(#loc1) | |
| %int40_1560 = torch.constant.int 40 loc(#loc1) | |
| %int1_1561 = torch.constant.int 1 loc(#loc1) | |
| %7278 = torch.prim.ListConstruct %int163840_1559, %int40_1560, %int1_1561 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7279 = torch.aten._reshape_alias %7276, %7277, %7278 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %7280 = torch.aten.bmm %7274, %7279 : !torch.tensor<[16,4096,4096],f32>, !torch.tensor<[16,4096,40],f32> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_1562 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1563 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1564 = torch.constant.int 40 loc(#loc1) | |
| %7281 = torch.prim.ListConstruct %int16_1562, %int4096_1563, %int40_1564 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7282 = torch.aten._unsafe_view %7280, %7281 : !torch.tensor<[16,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_1565 = torch.constant.int 2 loc(#loc1) | |
| %int8_1566 = torch.constant.int 8 loc(#loc1) | |
| %int4096_1567 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1568 = torch.constant.int 40 loc(#loc1) | |
| %7283 = torch.prim.ListConstruct %int2_1565, %int8_1566, %int4096_1567, %int40_1568 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_1569 = torch.constant.int 1310720 loc(#loc1) | |
| %int163840_1570 = torch.constant.int 163840 loc(#loc1) | |
| %int40_1571 = torch.constant.int 40 loc(#loc1) | |
| %int1_1572 = torch.constant.int 1 loc(#loc1) | |
| %7284 = torch.prim.ListConstruct %int1310720_1569, %int163840_1570, %int40_1571, %int1_1572 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7285 = torch.aten._reshape_alias %7282, %7283, %7284 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_1573 = torch.constant.int 0 loc(#loc1) | |
| %int2_1574 = torch.constant.int 2 loc(#loc1) | |
| %int1_1575 = torch.constant.int 1 loc(#loc1) | |
| %int3_1576 = torch.constant.int 3 loc(#loc1) | |
| %7286 = torch.prim.ListConstruct %int0_1573, %int2_1574, %int1_1575, %int3_1576 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7287 = torch.aten.permute %7285, %7286 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_1577 = torch.constant.int 0 loc(#loc1) | |
| %7288 = torch.aten.clone %7287, %int0_1577 : !torch.tensor<[2,4096,8,40],f32>, !torch.int -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int2_1578 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1579 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1580 = torch.constant.int 320 loc(#loc1) | |
| %7289 = torch.prim.ListConstruct %int2_1578, %int4096_1579, %int320_1580 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7290 = torch.aten._unsafe_view %7288, %7289 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7291 = torch.aten.detach %5753 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7292 = torch.aten.detach %7291 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1581 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1582 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1583 = torch.constant.int 20 loc(#loc1) | |
| %int16_1584 = torch.constant.int 16 loc(#loc1) | |
| %7293 = torch.prim.ListConstruct %int2_1581, %int4096_1582, %int20_1583, %int16_1584 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7294 = torch.aten.view %7290, %7293 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %7295 = torch.aten.abs %7294 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_1585 = torch.constant.int 3 loc(#loc1) | |
| %true_1586 = torch.constant.bool true loc(#loc1) | |
| %values_1587, %indices_1588 = torch.aten.max.dim %7295, %int3_1585, %true_1586 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_1589 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1590 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1591 = torch.constant.int 20 loc(#loc1) | |
| %int1_1592 = torch.constant.int 1 loc(#loc1) | |
| %7296 = torch.prim.ListConstruct %int2_1589, %int4096_1590, %int20_1591, %int1_1592 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7297 = torch.aten.view %values_1587, %7296 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_1593 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1594 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1595 = torch.constant.int 20 loc(#loc1) | |
| %int16_1596 = torch.constant.int 16 loc(#loc1) | |
| %7298 = torch.prim.ListConstruct %int2_1593, %int4096_1594, %int20_1595, %int16_1596 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1597 = torch.constant.bool false loc(#loc1) | |
| %7299 = torch.aten.expand %7297, %7298, %false_1597 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_1598 = torch.constant.int 0 loc(#loc1) | |
| %7300 = torch.aten.clone %7299, %int0_1598 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_1599 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1600 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1601 = torch.constant.int 320 loc(#loc1) | |
| %7301 = torch.prim.ListConstruct %int2_1599, %int4096_1600, %int320_1601 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7302 = torch.aten._unsafe_view %7300, %7301 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7303 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1602 = torch.constant.int 1 loc(#loc1) | |
| %7304 = torch.aten.sub.Tensor %7292, %7303, %int1_1602 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7305 = torch.aten.lift_fresh_copy %5752 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7306 = torch.aten.pow.Tensor_Tensor %7305, %7304 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7307 = torch.aten.neg %7306 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7308 = torch.aten.neg %7307 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7309 = torch.aten.div.Tensor %7302, %7308 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7310 = torch.aten.detach %5751 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7311 = torch.aten.detach %7310 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7312 = torch.aten.div.Tensor %7290, %7309 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1603 = torch.constant.int 1 loc(#loc1) | |
| %7313 = torch.aten.add.Tensor %7312, %7311, %int1_1603 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7314 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1604 = torch.constant.int 1 loc(#loc1) | |
| %7315 = torch.aten.sub.Tensor %7292, %7314, %int1_1604 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7316 = torch.aten.lift_fresh_copy %5750 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7317 = torch.aten.pow.Tensor_Tensor %7316, %7315 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7318 = torch.aten.neg %7317 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7319 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1605 = torch.constant.int 1 loc(#loc1) | |
| %7320 = torch.aten.sub.Tensor %7292, %7319, %int1_1605 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7321 = torch.aten.lift_fresh_copy %5749 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7322 = torch.aten.pow.Tensor_Tensor %7321, %7320 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7323 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1606 = torch.constant.int 1 loc(#loc1) | |
| %7324 = torch.aten.sub.Tensor %7322, %7323, %int1_1606 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7325 = torch.aten.gt.Tensor %7313, %7324 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7326 = torch.aten.where.self %7325, %7324, %7313 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7327 = torch.aten.lt.Tensor %7326, %7318 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7328 = torch.aten.where.self %7327, %7318, %7326 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7329 = torch.operator "aten.round"(%7328) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1607 = torch.constant.int 1 loc(#loc1) | |
| %7330 = torch.aten.sub.Tensor %7329, %7311, %int1_1607 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7331 = torch.aten.mul.Tensor %7330, %7309 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1608 = torch.constant.int 320 loc(#loc1) | |
| %int20_1609 = torch.constant.int 20 loc(#loc1) | |
| %int16_1610 = torch.constant.int 16 loc(#loc1) | |
| %7332 = torch.prim.ListConstruct %int320_1608, %int20_1609, %int16_1610 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1611 = torch.constant.bool false loc(#loc1) | |
| %7333 = torch.aten.expand %5748, %7332, %false_1611 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_1612 = torch.constant.int 0 loc(#loc1) | |
| %7334 = torch.aten.clone %7333, %int0_1612 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_1613 = torch.constant.int 320 loc(#loc1) | |
| %int320_1614 = torch.constant.int 320 loc(#loc1) | |
| %7335 = torch.prim.ListConstruct %int320_1613, %int320_1614 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7336 = torch.aten._unsafe_view %7334, %7335 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %7337 = torch.aten.mul.Tensor %5747, %7336 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_1615 = torch.constant.int 6 loc(#loc1) | |
| %none_1616 = torch.constant.none loc(#loc0) | |
| %none_1617 = torch.constant.none loc(#loc0) | |
| %none_1618 = torch.constant.none loc(#loc0) | |
| %false_1619 = torch.constant.bool false loc(#loc1) | |
| %none_1620 = torch.constant.none loc(#loc0) | |
| %7338 = torch.aten._to_copy %7337, %int6_1615, %none_1616, %none_1617, %none_1618, %false_1619, %none_1620 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %7339 = torch.aten.t %7338 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_1621 = torch.constant.int 8192 loc(#loc1) | |
| %int320_1622 = torch.constant.int 320 loc(#loc1) | |
| %7340 = torch.prim.ListConstruct %int8192_1621, %int320_1622 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7341 = torch.aten.view %7331, %7340 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_1623 = torch.constant.int 1 loc(#loc1) | |
| %int1_1624 = torch.constant.int 1 loc(#loc1) | |
| %7342 = torch.aten.addmm %5746, %7341, %7339, %int1_1623, %int1_1624 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_1625 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1626 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1627 = torch.constant.int 320 loc(#loc1) | |
| %7343 = torch.prim.ListConstruct %int2_1625, %int4096_1626, %int320_1627 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7344 = torch.aten.view %7342, %7343 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1628 = torch.constant.int 1 loc(#loc1) | |
| %7345 = torch.aten.add.Tensor %7344, %7065, %int1_1628 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1629 = torch.constant.int 320 loc(#loc1) | |
| %7346 = torch.prim.ListConstruct %int320_1629 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_1630 = torch.constant.none loc(#loc0) | |
| %none_1631 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05_1632 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0_1633, %result1_1634, %result2_1635 = torch.aten.native_layer_norm %7345, %7346, %none_1630, %none_1631, %float1.000000e-05_1632 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,1],f32>, !torch.tensor<[2,4096,1],f32> loc(#loc1) | |
| %7347 = torch.aten.detach %5745 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7348 = torch.aten.detach %7347 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1636 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1637 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1638 = torch.constant.int 20 loc(#loc1) | |
| %int16_1639 = torch.constant.int 16 loc(#loc1) | |
| %7349 = torch.prim.ListConstruct %int2_1636, %int4096_1637, %int20_1638, %int16_1639 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7350 = torch.aten.view %result0_1633, %7349 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %7351 = torch.aten.abs %7350 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_1640 = torch.constant.int 3 loc(#loc1) | |
| %true_1641 = torch.constant.bool true loc(#loc1) | |
| %values_1642, %indices_1643 = torch.aten.max.dim %7351, %int3_1640, %true_1641 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_1644 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1645 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1646 = torch.constant.int 20 loc(#loc1) | |
| %int1_1647 = torch.constant.int 1 loc(#loc1) | |
| %7352 = torch.prim.ListConstruct %int2_1644, %int4096_1645, %int20_1646, %int1_1647 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7353 = torch.aten.view %values_1642, %7352 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_1648 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1649 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1650 = torch.constant.int 20 loc(#loc1) | |
| %int16_1651 = torch.constant.int 16 loc(#loc1) | |
| %7354 = torch.prim.ListConstruct %int2_1648, %int4096_1649, %int20_1650, %int16_1651 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1652 = torch.constant.bool false loc(#loc1) | |
| %7355 = torch.aten.expand %7353, %7354, %false_1652 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_1653 = torch.constant.int 0 loc(#loc1) | |
| %7356 = torch.aten.clone %7355, %int0_1653 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_1654 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1655 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1656 = torch.constant.int 320 loc(#loc1) | |
| %7357 = torch.prim.ListConstruct %int2_1654, %int4096_1655, %int320_1656 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7358 = torch.aten._unsafe_view %7356, %7357 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7359 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1657 = torch.constant.int 1 loc(#loc1) | |
| %7360 = torch.aten.sub.Tensor %7348, %7359, %int1_1657 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7361 = torch.aten.lift_fresh_copy %5744 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7362 = torch.aten.pow.Tensor_Tensor %7361, %7360 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7363 = torch.aten.neg %7362 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7364 = torch.aten.neg %7363 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7365 = torch.aten.div.Tensor %7358, %7364 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7366 = torch.aten.detach %5743 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7367 = torch.aten.detach %7366 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7368 = torch.aten.div.Tensor %result0_1633, %7365 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1658 = torch.constant.int 1 loc(#loc1) | |
| %7369 = torch.aten.add.Tensor %7368, %7367, %int1_1658 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7370 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1659 = torch.constant.int 1 loc(#loc1) | |
| %7371 = torch.aten.sub.Tensor %7348, %7370, %int1_1659 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7372 = torch.aten.lift_fresh_copy %5742 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7373 = torch.aten.pow.Tensor_Tensor %7372, %7371 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7374 = torch.aten.neg %7373 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7375 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1660 = torch.constant.int 1 loc(#loc1) | |
| %7376 = torch.aten.sub.Tensor %7348, %7375, %int1_1660 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7377 = torch.aten.lift_fresh_copy %5741 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7378 = torch.aten.pow.Tensor_Tensor %7377, %7376 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7379 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1661 = torch.constant.int 1 loc(#loc1) | |
| %7380 = torch.aten.sub.Tensor %7378, %7379, %int1_1661 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7381 = torch.aten.gt.Tensor %7369, %7380 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7382 = torch.aten.where.self %7381, %7380, %7369 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7383 = torch.aten.lt.Tensor %7382, %7374 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7384 = torch.aten.where.self %7383, %7374, %7382 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7385 = torch.operator "aten.round"(%7384) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1662 = torch.constant.int 1 loc(#loc1) | |
| %7386 = torch.aten.sub.Tensor %7385, %7367, %int1_1662 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7387 = torch.aten.mul.Tensor %7386, %7365 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1663 = torch.constant.int 320 loc(#loc1) | |
| %int20_1664 = torch.constant.int 20 loc(#loc1) | |
| %int16_1665 = torch.constant.int 16 loc(#loc1) | |
| %7388 = torch.prim.ListConstruct %int320_1663, %int20_1664, %int16_1665 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1666 = torch.constant.bool false loc(#loc1) | |
| %7389 = torch.aten.expand %5740, %7388, %false_1666 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_1667 = torch.constant.int 0 loc(#loc1) | |
| %7390 = torch.aten.clone %7389, %int0_1667 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_1668 = torch.constant.int 320 loc(#loc1) | |
| %int320_1669 = torch.constant.int 320 loc(#loc1) | |
| %7391 = torch.prim.ListConstruct %int320_1668, %int320_1669 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7392 = torch.aten._unsafe_view %7390, %7391 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %7393 = torch.aten.mul.Tensor %5739, %7392 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_1670 = torch.constant.int 6 loc(#loc1) | |
| %none_1671 = torch.constant.none loc(#loc0) | |
| %none_1672 = torch.constant.none loc(#loc0) | |
| %none_1673 = torch.constant.none loc(#loc0) | |
| %false_1674 = torch.constant.bool false loc(#loc1) | |
| %none_1675 = torch.constant.none loc(#loc0) | |
| %7394 = torch.aten._to_copy %7393, %int6_1670, %none_1671, %none_1672, %none_1673, %false_1674, %none_1675 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %7395 = torch.aten.t %7394 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_1676 = torch.constant.int 8192 loc(#loc1) | |
| %int320_1677 = torch.constant.int 320 loc(#loc1) | |
| %7396 = torch.prim.ListConstruct %int8192_1676, %int320_1677 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7397 = torch.aten.view %7387, %7396 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_1678 = torch.constant.int 1 loc(#loc1) | |
| %int1_1679 = torch.constant.int 1 loc(#loc1) | |
| %7398 = torch.aten.addmm %5738, %7397, %7395, %int1_1678, %int1_1679 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_1680 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1681 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1682 = torch.constant.int 320 loc(#loc1) | |
| %7399 = torch.prim.ListConstruct %int2_1680, %int4096_1681, %int320_1682 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7400 = torch.aten.view %7398, %7399 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7401 = torch.aten.detach %5737 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7402 = torch.aten.detach %7401 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1683 = torch.constant.int 2 loc(#loc1) | |
| %int77_1684 = torch.constant.int 77 loc(#loc1) | |
| %int48_1685 = torch.constant.int 48 loc(#loc1) | |
| %int16_1686 = torch.constant.int 16 loc(#loc1) | |
| %7403 = torch.prim.ListConstruct %int2_1683, %int77_1684, %int48_1685, %int16_1686 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7404 = torch.aten.view %3921, %7403 : !torch.tensor<[2,77,768],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %7405 = torch.aten.abs %7404 : !torch.tensor<[2,77,48,16],f32> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int3_1687 = torch.constant.int 3 loc(#loc1) | |
| %true_1688 = torch.constant.bool true loc(#loc1) | |
| %values_1689, %indices_1690 = torch.aten.max.dim %7405, %int3_1687, %true_1688 : !torch.tensor<[2,77,48,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,77,48,1],f32>, !torch.tensor<[2,77,48,1],si64> loc(#loc1) | |
| %int2_1691 = torch.constant.int 2 loc(#loc1) | |
| %int77_1692 = torch.constant.int 77 loc(#loc1) | |
| %int48_1693 = torch.constant.int 48 loc(#loc1) | |
| %int1_1694 = torch.constant.int 1 loc(#loc1) | |
| %7406 = torch.prim.ListConstruct %int2_1691, %int77_1692, %int48_1693, %int1_1694 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7407 = torch.aten.view %values_1689, %7406 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,1],f32> loc(#loc1) | |
| %int2_1695 = torch.constant.int 2 loc(#loc1) | |
| %int77_1696 = torch.constant.int 77 loc(#loc1) | |
| %int48_1697 = torch.constant.int 48 loc(#loc1) | |
| %int16_1698 = torch.constant.int 16 loc(#loc1) | |
| %7408 = torch.prim.ListConstruct %int2_1695, %int77_1696, %int48_1697, %int16_1698 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1699 = torch.constant.bool false loc(#loc1) | |
| %7409 = torch.aten.expand %7407, %7408, %false_1699 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int0_1700 = torch.constant.int 0 loc(#loc1) | |
| %7410 = torch.aten.clone %7409, %int0_1700 : !torch.tensor<[2,77,48,16],f32>, !torch.int -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int2_1701 = torch.constant.int 2 loc(#loc1) | |
| %int77_1702 = torch.constant.int 77 loc(#loc1) | |
| %int768_1703 = torch.constant.int 768 loc(#loc1) | |
| %7411 = torch.prim.ListConstruct %int2_1701, %int77_1702, %int768_1703 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7412 = torch.aten._unsafe_view %7410, %7411 : !torch.tensor<[2,77,48,16],f32>, !torch.list<int> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7413 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1704 = torch.constant.int 1 loc(#loc1) | |
| %7414 = torch.aten.sub.Tensor %7402, %7413, %int1_1704 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7415 = torch.aten.lift_fresh_copy %5736 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7416 = torch.aten.pow.Tensor_Tensor %7415, %7414 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7417 = torch.aten.neg %7416 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7418 = torch.aten.neg %7417 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7419 = torch.aten.div.Tensor %7412, %7418 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7420 = torch.aten.detach %5735 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7421 = torch.aten.detach %7420 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7422 = torch.aten.div.Tensor %3921, %7419 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_1705 = torch.constant.int 1 loc(#loc1) | |
| %7423 = torch.aten.add.Tensor %7422, %7421, %int1_1705 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7424 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1706 = torch.constant.int 1 loc(#loc1) | |
| %7425 = torch.aten.sub.Tensor %7402, %7424, %int1_1706 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7426 = torch.aten.lift_fresh_copy %5734 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7427 = torch.aten.pow.Tensor_Tensor %7426, %7425 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7428 = torch.aten.neg %7427 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7429 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1707 = torch.constant.int 1 loc(#loc1) | |
| %7430 = torch.aten.sub.Tensor %7402, %7429, %int1_1707 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7431 = torch.aten.lift_fresh_copy %5733 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7432 = torch.aten.pow.Tensor_Tensor %7431, %7430 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7433 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1708 = torch.constant.int 1 loc(#loc1) | |
| %7434 = torch.aten.sub.Tensor %7432, %7433, %int1_1708 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7435 = torch.aten.gt.Tensor %7423, %7434 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %7436 = torch.aten.where.self %7435, %7434, %7423 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7437 = torch.aten.lt.Tensor %7436, %7428 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %7438 = torch.aten.where.self %7437, %7428, %7436 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7439 = torch.operator "aten.round"(%7438) : (!torch.tensor<[2,77,768],f32>) -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_1709 = torch.constant.int 1 loc(#loc1) | |
| %7440 = torch.aten.sub.Tensor %7439, %7421, %int1_1709 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7441 = torch.aten.mul.Tensor %7440, %7419 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int320_1710 = torch.constant.int 320 loc(#loc1) | |
| %int48_1711 = torch.constant.int 48 loc(#loc1) | |
| %int16_1712 = torch.constant.int 16 loc(#loc1) | |
| %7442 = torch.prim.ListConstruct %int320_1710, %int48_1711, %int16_1712 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1713 = torch.constant.bool false loc(#loc1) | |
| %7443 = torch.aten.expand %5732, %7442, %false_1713 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,48,16],f16> loc(#loc1) | |
| %int0_1714 = torch.constant.int 0 loc(#loc1) | |
| %7444 = torch.aten.clone %7443, %int0_1714 : !torch.tensor<[320,48,16],f16>, !torch.int -> !torch.tensor<[320,48,16],f16> loc(#loc1) | |
| %int320_1715 = torch.constant.int 320 loc(#loc1) | |
| %int768_1716 = torch.constant.int 768 loc(#loc1) | |
| %7445 = torch.prim.ListConstruct %int320_1715, %int768_1716 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7446 = torch.aten._unsafe_view %7444, %7445 : !torch.tensor<[320,48,16],f16>, !torch.list<int> -> !torch.tensor<[320,768],f16> loc(#loc1) | |
| %7447 = torch.aten.mul.Tensor %5731, %7446 : !torch.tensor, !torch.tensor<[320,768],f16> -> !torch.tensor<[320,768],f16> loc(#loc1) | |
| %int6_1717 = torch.constant.int 6 loc(#loc1) | |
| %none_1718 = torch.constant.none loc(#loc0) | |
| %none_1719 = torch.constant.none loc(#loc0) | |
| %none_1720 = torch.constant.none loc(#loc0) | |
| %false_1721 = torch.constant.bool false loc(#loc1) | |
| %none_1722 = torch.constant.none loc(#loc0) | |
| %7448 = torch.aten._to_copy %7447, %int6_1717, %none_1718, %none_1719, %none_1720, %false_1721, %none_1722 : !torch.tensor<[320,768],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,768],f32> loc(#loc1) | |
| %7449 = torch.aten.t %7448 : !torch.tensor<[320,768],f32> -> !torch.tensor<[768,320],f32> loc(#loc1) | |
| %int154_1723 = torch.constant.int 154 loc(#loc1) | |
| %int768_1724 = torch.constant.int 768 loc(#loc1) | |
| %7450 = torch.prim.ListConstruct %int154_1723, %int768_1724 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int768_1725 = torch.constant.int 768 loc(#loc1) | |
| %int1_1726 = torch.constant.int 1 loc(#loc1) | |
| %7451 = torch.prim.ListConstruct %int768_1725, %int1_1726 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7452 = torch.aten._reshape_alias %7441, %7450, %7451 : !torch.tensor<[2,77,768],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[154,768],f32> loc(#loc1) | |
| %7453 = torch.aten.mm %7452, %7449 : !torch.tensor<[154,768],f32>, !torch.tensor<[768,320],f32> -> !torch.tensor<[154,320],f32> loc(#loc1) | |
| %int2_1727 = torch.constant.int 2 loc(#loc1) | |
| %int77_1728 = torch.constant.int 77 loc(#loc1) | |
| %int320_1729 = torch.constant.int 320 loc(#loc1) | |
| %7454 = torch.prim.ListConstruct %int2_1727, %int77_1728, %int320_1729 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7455 = torch.aten._unsafe_view %7453, %7454 : !torch.tensor<[154,320],f32>, !torch.list<int> -> !torch.tensor<[2,77,320],f32> loc(#loc1) | |
| %7456 = torch.aten.detach %5730 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7457 = torch.aten.detach %7456 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1730 = torch.constant.int 2 loc(#loc1) | |
| %int77_1731 = torch.constant.int 77 loc(#loc1) | |
| %int48_1732 = torch.constant.int 48 loc(#loc1) | |
| %int16_1733 = torch.constant.int 16 loc(#loc1) | |
| %7458 = torch.prim.ListConstruct %int2_1730, %int77_1731, %int48_1732, %int16_1733 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7459 = torch.aten.view %3921, %7458 : !torch.tensor<[2,77,768],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %7460 = torch.aten.abs %7459 : !torch.tensor<[2,77,48,16],f32> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int3_1734 = torch.constant.int 3 loc(#loc1) | |
| %true_1735 = torch.constant.bool true loc(#loc1) | |
| %values_1736, %indices_1737 = torch.aten.max.dim %7460, %int3_1734, %true_1735 : !torch.tensor<[2,77,48,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,77,48,1],f32>, !torch.tensor<[2,77,48,1],si64> loc(#loc1) | |
| %int2_1738 = torch.constant.int 2 loc(#loc1) | |
| %int77_1739 = torch.constant.int 77 loc(#loc1) | |
| %int48_1740 = torch.constant.int 48 loc(#loc1) | |
| %int1_1741 = torch.constant.int 1 loc(#loc1) | |
| %7461 = torch.prim.ListConstruct %int2_1738, %int77_1739, %int48_1740, %int1_1741 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7462 = torch.aten.view %values_1736, %7461 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,1],f32> loc(#loc1) | |
| %int2_1742 = torch.constant.int 2 loc(#loc1) | |
| %int77_1743 = torch.constant.int 77 loc(#loc1) | |
| %int48_1744 = torch.constant.int 48 loc(#loc1) | |
| %int16_1745 = torch.constant.int 16 loc(#loc1) | |
| %7463 = torch.prim.ListConstruct %int2_1742, %int77_1743, %int48_1744, %int16_1745 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1746 = torch.constant.bool false loc(#loc1) | |
| %7464 = torch.aten.expand %7462, %7463, %false_1746 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int0_1747 = torch.constant.int 0 loc(#loc1) | |
| %7465 = torch.aten.clone %7464, %int0_1747 : !torch.tensor<[2,77,48,16],f32>, !torch.int -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int2_1748 = torch.constant.int 2 loc(#loc1) | |
| %int77_1749 = torch.constant.int 77 loc(#loc1) | |
| %int768_1750 = torch.constant.int 768 loc(#loc1) | |
| %7466 = torch.prim.ListConstruct %int2_1748, %int77_1749, %int768_1750 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7467 = torch.aten._unsafe_view %7465, %7466 : !torch.tensor<[2,77,48,16],f32>, !torch.list<int> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7468 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1751 = torch.constant.int 1 loc(#loc1) | |
| %7469 = torch.aten.sub.Tensor %7457, %7468, %int1_1751 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7470 = torch.aten.lift_fresh_copy %5729 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7471 = torch.aten.pow.Tensor_Tensor %7470, %7469 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7472 = torch.aten.neg %7471 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7473 = torch.aten.neg %7472 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7474 = torch.aten.div.Tensor %7467, %7473 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7475 = torch.aten.detach %5728 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7476 = torch.aten.detach %7475 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7477 = torch.aten.div.Tensor %3921, %7474 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_1752 = torch.constant.int 1 loc(#loc1) | |
| %7478 = torch.aten.add.Tensor %7477, %7476, %int1_1752 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7479 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1753 = torch.constant.int 1 loc(#loc1) | |
| %7480 = torch.aten.sub.Tensor %7457, %7479, %int1_1753 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7481 = torch.aten.lift_fresh_copy %5727 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7482 = torch.aten.pow.Tensor_Tensor %7481, %7480 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7483 = torch.aten.neg %7482 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7484 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1754 = torch.constant.int 1 loc(#loc1) | |
| %7485 = torch.aten.sub.Tensor %7457, %7484, %int1_1754 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7486 = torch.aten.lift_fresh_copy %5726 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7487 = torch.aten.pow.Tensor_Tensor %7486, %7485 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7488 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1755 = torch.constant.int 1 loc(#loc1) | |
| %7489 = torch.aten.sub.Tensor %7487, %7488, %int1_1755 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7490 = torch.aten.gt.Tensor %7478, %7489 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %7491 = torch.aten.where.self %7490, %7489, %7478 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7492 = torch.aten.lt.Tensor %7491, %7483 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %7493 = torch.aten.where.self %7492, %7483, %7491 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7494 = torch.operator "aten.round"(%7493) : (!torch.tensor<[2,77,768],f32>) -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_1756 = torch.constant.int 1 loc(#loc1) | |
| %7495 = torch.aten.sub.Tensor %7494, %7476, %int1_1756 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %7496 = torch.aten.mul.Tensor %7495, %7474 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int320_1757 = torch.constant.int 320 loc(#loc1) | |
| %int48_1758 = torch.constant.int 48 loc(#loc1) | |
| %int16_1759 = torch.constant.int 16 loc(#loc1) | |
| %7497 = torch.prim.ListConstruct %int320_1757, %int48_1758, %int16_1759 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1760 = torch.constant.bool false loc(#loc1) | |
| %7498 = torch.aten.expand %5725, %7497, %false_1760 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,48,16],f16> loc(#loc1) | |
| %int0_1761 = torch.constant.int 0 loc(#loc1) | |
| %7499 = torch.aten.clone %7498, %int0_1761 : !torch.tensor<[320,48,16],f16>, !torch.int -> !torch.tensor<[320,48,16],f16> loc(#loc1) | |
| %int320_1762 = torch.constant.int 320 loc(#loc1) | |
| %int768_1763 = torch.constant.int 768 loc(#loc1) | |
| %7500 = torch.prim.ListConstruct %int320_1762, %int768_1763 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7501 = torch.aten._unsafe_view %7499, %7500 : !torch.tensor<[320,48,16],f16>, !torch.list<int> -> !torch.tensor<[320,768],f16> loc(#loc1) | |
| %7502 = torch.aten.mul.Tensor %5724, %7501 : !torch.tensor, !torch.tensor<[320,768],f16> -> !torch.tensor<[320,768],f16> loc(#loc1) | |
| %int6_1764 = torch.constant.int 6 loc(#loc1) | |
| %none_1765 = torch.constant.none loc(#loc0) | |
| %none_1766 = torch.constant.none loc(#loc0) | |
| %none_1767 = torch.constant.none loc(#loc0) | |
| %false_1768 = torch.constant.bool false loc(#loc1) | |
| %none_1769 = torch.constant.none loc(#loc0) | |
| %7503 = torch.aten._to_copy %7502, %int6_1764, %none_1765, %none_1766, %none_1767, %false_1768, %none_1769 : !torch.tensor<[320,768],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,768],f32> loc(#loc1) | |
| %7504 = torch.aten.t %7503 : !torch.tensor<[320,768],f32> -> !torch.tensor<[768,320],f32> loc(#loc1) | |
| %int154_1770 = torch.constant.int 154 loc(#loc1) | |
| %int768_1771 = torch.constant.int 768 loc(#loc1) | |
| %7505 = torch.prim.ListConstruct %int154_1770, %int768_1771 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int768_1772 = torch.constant.int 768 loc(#loc1) | |
| %int1_1773 = torch.constant.int 1 loc(#loc1) | |
| %7506 = torch.prim.ListConstruct %int768_1772, %int1_1773 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7507 = torch.aten._reshape_alias %7496, %7505, %7506 : !torch.tensor<[2,77,768],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[154,768],f32> loc(#loc1) | |
| %7508 = torch.aten.mm %7507, %7504 : !torch.tensor<[154,768],f32>, !torch.tensor<[768,320],f32> -> !torch.tensor<[154,320],f32> loc(#loc1) | |
| %int2_1774 = torch.constant.int 2 loc(#loc1) | |
| %int77_1775 = torch.constant.int 77 loc(#loc1) | |
| %int320_1776 = torch.constant.int 320 loc(#loc1) | |
| %7509 = torch.prim.ListConstruct %int2_1774, %int77_1775, %int320_1776 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7510 = torch.aten._unsafe_view %7508, %7509 : !torch.tensor<[154,320],f32>, !torch.list<int> -> !torch.tensor<[2,77,320],f32> loc(#loc1) | |
| %int2_1777 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1778 = torch.constant.int 4096 loc(#loc1) | |
| %int8_1779 = torch.constant.int 8 loc(#loc1) | |
| %int40_1780 = torch.constant.int 40 loc(#loc1) | |
| %7511 = torch.prim.ListConstruct %int2_1777, %int4096_1778, %int8_1779, %int40_1780 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_1781 = torch.constant.int 1310720 loc(#loc1) | |
| %int320_1782 = torch.constant.int 320 loc(#loc1) | |
| %int40_1783 = torch.constant.int 40 loc(#loc1) | |
| %int1_1784 = torch.constant.int 1 loc(#loc1) | |
| %7512 = torch.prim.ListConstruct %int1310720_1781, %int320_1782, %int40_1783, %int1_1784 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7513 = torch.aten._reshape_alias %7400, %7511, %7512 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_1785 = torch.constant.int 0 loc(#loc1) | |
| %int2_1786 = torch.constant.int 2 loc(#loc1) | |
| %int1_1787 = torch.constant.int 1 loc(#loc1) | |
| %int3_1788 = torch.constant.int 3 loc(#loc1) | |
| %7514 = torch.prim.ListConstruct %int0_1785, %int2_1786, %int1_1787, %int3_1788 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7515 = torch.aten.permute %7513, %7514 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_1789 = torch.constant.int 0 loc(#loc1) | |
| %7516 = torch.aten.clone %7515, %int0_1789 : !torch.tensor<[2,8,4096,40],f32>, !torch.int -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int16_1790 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1791 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1792 = torch.constant.int 40 loc(#loc1) | |
| %7517 = torch.prim.ListConstruct %int16_1790, %int4096_1791, %int40_1792 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7518 = torch.aten._unsafe_view %7516, %7517 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_1793 = torch.constant.int 2 loc(#loc1) | |
| %int77_1794 = torch.constant.int 77 loc(#loc1) | |
| %int8_1795 = torch.constant.int 8 loc(#loc1) | |
| %int40_1796 = torch.constant.int 40 loc(#loc1) | |
| %7519 = torch.prim.ListConstruct %int2_1793, %int77_1794, %int8_1795, %int40_1796 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int24640_1797 = torch.constant.int 24640 loc(#loc1) | |
| %int320_1798 = torch.constant.int 320 loc(#loc1) | |
| %int40_1799 = torch.constant.int 40 loc(#loc1) | |
| %int1_1800 = torch.constant.int 1 loc(#loc1) | |
| %7520 = torch.prim.ListConstruct %int24640_1797, %int320_1798, %int40_1799, %int1_1800 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7521 = torch.aten._reshape_alias %7455, %7519, %7520 : !torch.tensor<[2,77,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,77,8,40],f32> loc(#loc1) | |
| %int0_1801 = torch.constant.int 0 loc(#loc1) | |
| %int2_1802 = torch.constant.int 2 loc(#loc1) | |
| %int1_1803 = torch.constant.int 1 loc(#loc1) | |
| %int3_1804 = torch.constant.int 3 loc(#loc1) | |
| %7522 = torch.prim.ListConstruct %int0_1801, %int2_1802, %int1_1803, %int3_1804 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7523 = torch.aten.permute %7521, %7522 : !torch.tensor<[2,77,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,77,40],f32> loc(#loc1) | |
| %int0_1805 = torch.constant.int 0 loc(#loc1) | |
| %7524 = torch.aten.clone %7523, %int0_1805 : !torch.tensor<[2,8,77,40],f32>, !torch.int -> !torch.tensor<[2,8,77,40],f32> loc(#loc1) | |
| %int16_1806 = torch.constant.int 16 loc(#loc1) | |
| %int77_1807 = torch.constant.int 77 loc(#loc1) | |
| %int40_1808 = torch.constant.int 40 loc(#loc1) | |
| %7525 = torch.prim.ListConstruct %int16_1806, %int77_1807, %int40_1808 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7526 = torch.aten._unsafe_view %7524, %7525 : !torch.tensor<[2,8,77,40],f32>, !torch.list<int> -> !torch.tensor<[16,77,40],f32> loc(#loc1) | |
| %int2_1809 = torch.constant.int 2 loc(#loc1) | |
| %int77_1810 = torch.constant.int 77 loc(#loc1) | |
| %int8_1811 = torch.constant.int 8 loc(#loc1) | |
| %int40_1812 = torch.constant.int 40 loc(#loc1) | |
| %7527 = torch.prim.ListConstruct %int2_1809, %int77_1810, %int8_1811, %int40_1812 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int24640_1813 = torch.constant.int 24640 loc(#loc1) | |
| %int320_1814 = torch.constant.int 320 loc(#loc1) | |
| %int40_1815 = torch.constant.int 40 loc(#loc1) | |
| %int1_1816 = torch.constant.int 1 loc(#loc1) | |
| %7528 = torch.prim.ListConstruct %int24640_1813, %int320_1814, %int40_1815, %int1_1816 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7529 = torch.aten._reshape_alias %7510, %7527, %7528 : !torch.tensor<[2,77,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,77,8,40],f32> loc(#loc1) | |
| %int0_1817 = torch.constant.int 0 loc(#loc1) | |
| %int2_1818 = torch.constant.int 2 loc(#loc1) | |
| %int1_1819 = torch.constant.int 1 loc(#loc1) | |
| %int3_1820 = torch.constant.int 3 loc(#loc1) | |
| %7530 = torch.prim.ListConstruct %int0_1817, %int2_1818, %int1_1819, %int3_1820 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7531 = torch.aten.permute %7529, %7530 : !torch.tensor<[2,77,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,8,77,40],f32> loc(#loc1) | |
| %int0_1821 = torch.constant.int 0 loc(#loc1) | |
| %7532 = torch.aten.clone %7531, %int0_1821 : !torch.tensor<[2,8,77,40],f32>, !torch.int -> !torch.tensor<[2,8,77,40],f32> loc(#loc1) | |
| %int16_1822 = torch.constant.int 16 loc(#loc1) | |
| %int77_1823 = torch.constant.int 77 loc(#loc1) | |
| %int40_1824 = torch.constant.int 40 loc(#loc1) | |
| %7533 = torch.prim.ListConstruct %int16_1822, %int77_1823, %int40_1824 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7534 = torch.aten._unsafe_view %7532, %7533 : !torch.tensor<[2,8,77,40],f32>, !torch.list<int> -> !torch.tensor<[16,77,40],f32> loc(#loc1) | |
| %int-1_1825 = torch.constant.int -1 loc(#loc1) | |
| %int-2_1826 = torch.constant.int -2 loc(#loc1) | |
| %7535 = torch.aten.transpose.int %7526, %int-1_1825, %int-2_1826 : !torch.tensor<[16,77,40],f32>, !torch.int, !torch.int -> !torch.tensor<[16,40,77],f32> loc(#loc1) | |
| %int16_1827 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1828 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1829 = torch.constant.int 40 loc(#loc1) | |
| %7536 = torch.prim.ListConstruct %int16_1827, %int4096_1828, %int40_1829 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1830 = torch.constant.bool false loc(#loc1) | |
| %7537 = torch.aten.expand %7518, %7536, %false_1830 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_1831 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1832 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1833 = torch.constant.int 40 loc(#loc1) | |
| %7538 = torch.prim.ListConstruct %int16_1831, %int4096_1832, %int40_1833 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int163840_1834 = torch.constant.int 163840 loc(#loc1) | |
| %int40_1835 = torch.constant.int 40 loc(#loc1) | |
| %int1_1836 = torch.constant.int 1 loc(#loc1) | |
| %7539 = torch.prim.ListConstruct %int163840_1834, %int40_1835, %int1_1836 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7540 = torch.aten._reshape_alias %7537, %7538, %7539 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_1837 = torch.constant.int 16 loc(#loc1) | |
| %int40_1838 = torch.constant.int 40 loc(#loc1) | |
| %int77_1839 = torch.constant.int 77 loc(#loc1) | |
| %7541 = torch.prim.ListConstruct %int16_1837, %int40_1838, %int77_1839 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1840 = torch.constant.bool false loc(#loc1) | |
| %7542 = torch.aten.expand %7535, %7541, %false_1840 : !torch.tensor<[16,40,77],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,40,77],f32> loc(#loc1) | |
| %int16_1841 = torch.constant.int 16 loc(#loc1) | |
| %int40_1842 = torch.constant.int 40 loc(#loc1) | |
| %int77_1843 = torch.constant.int 77 loc(#loc1) | |
| %7543 = torch.prim.ListConstruct %int16_1841, %int40_1842, %int77_1843 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int3080_1844 = torch.constant.int 3080 loc(#loc1) | |
| %int1_1845 = torch.constant.int 1 loc(#loc1) | |
| %int40_1846 = torch.constant.int 40 loc(#loc1) | |
| %7544 = torch.prim.ListConstruct %int3080_1844, %int1_1845, %int40_1846 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7545 = torch.aten._reshape_alias %7542, %7543, %7544 : !torch.tensor<[16,40,77],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,40,77],f32> loc(#loc1) | |
| %7546 = torch.aten.bmm %7540, %7545 : !torch.tensor<[16,4096,40],f32>, !torch.tensor<[16,40,77],f32> -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int16_1847 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1848 = torch.constant.int 4096 loc(#loc1) | |
| %int77_1849 = torch.constant.int 77 loc(#loc1) | |
| %7547 = torch.prim.ListConstruct %int16_1847, %int4096_1848, %int77_1849 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7548 = torch.aten._unsafe_view %7546, %7547 : !torch.tensor<[16,4096,77],f32>, !torch.list<int> -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %7549 = torch.tensor.literal(dense<0.15811388300841897> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %7550 = torch.aten.mul.Tensor %7548, %7549 : !torch.tensor<[16,4096,77],f32>, !torch.tensor<[],f64> -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int-1_1850 = torch.constant.int -1 loc(#loc1) | |
| %false_1851 = torch.constant.bool false loc(#loc1) | |
| %7551 = torch.aten._softmax %7550, %int-1_1850, %false_1851 : !torch.tensor<[16,4096,77],f32>, !torch.int, !torch.bool -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int16_1852 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1853 = torch.constant.int 4096 loc(#loc1) | |
| %int77_1854 = torch.constant.int 77 loc(#loc1) | |
| %7552 = torch.prim.ListConstruct %int16_1852, %int4096_1853, %int77_1854 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1855 = torch.constant.bool false loc(#loc1) | |
| %7553 = torch.aten.expand %7551, %7552, %false_1855 : !torch.tensor<[16,4096,77],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int16_1856 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1857 = torch.constant.int 4096 loc(#loc1) | |
| %int77_1858 = torch.constant.int 77 loc(#loc1) | |
| %7554 = torch.prim.ListConstruct %int16_1856, %int4096_1857, %int77_1858 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int315392_1859 = torch.constant.int 315392 loc(#loc1) | |
| %int77_1860 = torch.constant.int 77 loc(#loc1) | |
| %int1_1861 = torch.constant.int 1 loc(#loc1) | |
| %7555 = torch.prim.ListConstruct %int315392_1859, %int77_1860, %int1_1861 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7556 = torch.aten._reshape_alias %7553, %7554, %7555 : !torch.tensor<[16,4096,77],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,4096,77],f32> loc(#loc1) | |
| %int16_1862 = torch.constant.int 16 loc(#loc1) | |
| %int77_1863 = torch.constant.int 77 loc(#loc1) | |
| %int40_1864 = torch.constant.int 40 loc(#loc1) | |
| %7557 = torch.prim.ListConstruct %int16_1862, %int77_1863, %int40_1864 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1865 = torch.constant.bool false loc(#loc1) | |
| %7558 = torch.aten.expand %7534, %7557, %false_1865 : !torch.tensor<[16,77,40],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,77,40],f32> loc(#loc1) | |
| %int16_1866 = torch.constant.int 16 loc(#loc1) | |
| %int77_1867 = torch.constant.int 77 loc(#loc1) | |
| %int40_1868 = torch.constant.int 40 loc(#loc1) | |
| %7559 = torch.prim.ListConstruct %int16_1866, %int77_1867, %int40_1868 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int3080_1869 = torch.constant.int 3080 loc(#loc1) | |
| %int40_1870 = torch.constant.int 40 loc(#loc1) | |
| %int1_1871 = torch.constant.int 1 loc(#loc1) | |
| %7560 = torch.prim.ListConstruct %int3080_1869, %int40_1870, %int1_1871 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7561 = torch.aten._reshape_alias %7558, %7559, %7560 : !torch.tensor<[16,77,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,77,40],f32> loc(#loc1) | |
| %7562 = torch.aten.bmm %7556, %7561 : !torch.tensor<[16,4096,77],f32>, !torch.tensor<[16,77,40],f32> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int16_1872 = torch.constant.int 16 loc(#loc1) | |
| %int4096_1873 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1874 = torch.constant.int 40 loc(#loc1) | |
| %7563 = torch.prim.ListConstruct %int16_1872, %int4096_1873, %int40_1874 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7564 = torch.aten._unsafe_view %7562, %7563 : !torch.tensor<[16,4096,40],f32>, !torch.list<int> -> !torch.tensor<[16,4096,40],f32> loc(#loc1) | |
| %int2_1875 = torch.constant.int 2 loc(#loc1) | |
| %int8_1876 = torch.constant.int 8 loc(#loc1) | |
| %int4096_1877 = torch.constant.int 4096 loc(#loc1) | |
| %int40_1878 = torch.constant.int 40 loc(#loc1) | |
| %7565 = torch.prim.ListConstruct %int2_1875, %int8_1876, %int4096_1877, %int40_1878 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_1879 = torch.constant.int 1310720 loc(#loc1) | |
| %int163840_1880 = torch.constant.int 163840 loc(#loc1) | |
| %int40_1881 = torch.constant.int 40 loc(#loc1) | |
| %int1_1882 = torch.constant.int 1 loc(#loc1) | |
| %7566 = torch.prim.ListConstruct %int1310720_1879, %int163840_1880, %int40_1881, %int1_1882 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7567 = torch.aten._reshape_alias %7564, %7565, %7566 : !torch.tensor<[16,4096,40],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,8,4096,40],f32> loc(#loc1) | |
| %int0_1883 = torch.constant.int 0 loc(#loc1) | |
| %int2_1884 = torch.constant.int 2 loc(#loc1) | |
| %int1_1885 = torch.constant.int 1 loc(#loc1) | |
| %int3_1886 = torch.constant.int 3 loc(#loc1) | |
| %7568 = torch.prim.ListConstruct %int0_1883, %int2_1884, %int1_1885, %int3_1886 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7569 = torch.aten.permute %7567, %7568 : !torch.tensor<[2,8,4096,40],f32>, !torch.list<int> -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int0_1887 = torch.constant.int 0 loc(#loc1) | |
| %7570 = torch.aten.clone %7569, %int0_1887 : !torch.tensor<[2,4096,8,40],f32>, !torch.int -> !torch.tensor<[2,4096,8,40],f32> loc(#loc1) | |
| %int2_1888 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1889 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1890 = torch.constant.int 320 loc(#loc1) | |
| %7571 = torch.prim.ListConstruct %int2_1888, %int4096_1889, %int320_1890 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7572 = torch.aten._unsafe_view %7570, %7571 : !torch.tensor<[2,4096,8,40],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7573 = torch.aten.detach %5723 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7574 = torch.aten.detach %7573 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_1891 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1892 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1893 = torch.constant.int 20 loc(#loc1) | |
| %int16_1894 = torch.constant.int 16 loc(#loc1) | |
| %7575 = torch.prim.ListConstruct %int2_1891, %int4096_1892, %int20_1893, %int16_1894 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7576 = torch.aten.view %7572, %7575 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %7577 = torch.aten.abs %7576 : !torch.tensor<[2,4096,20,16],f32> -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int3_1895 = torch.constant.int 3 loc(#loc1) | |
| %true_1896 = torch.constant.bool true loc(#loc1) | |
| %values_1897, %indices_1898 = torch.aten.max.dim %7577, %int3_1895, %true_1896 : !torch.tensor<[2,4096,20,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,4096,20,1],f32>, !torch.tensor<[2,4096,20,1],si64> loc(#loc1) | |
| %int2_1899 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1900 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1901 = torch.constant.int 20 loc(#loc1) | |
| %int1_1902 = torch.constant.int 1 loc(#loc1) | |
| %7578 = torch.prim.ListConstruct %int2_1899, %int4096_1900, %int20_1901, %int1_1902 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7579 = torch.aten.view %values_1897, %7578 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int> -> !torch.tensor<[2,4096,20,1],f32> loc(#loc1) | |
| %int2_1903 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1904 = torch.constant.int 4096 loc(#loc1) | |
| %int20_1905 = torch.constant.int 20 loc(#loc1) | |
| %int16_1906 = torch.constant.int 16 loc(#loc1) | |
| %7580 = torch.prim.ListConstruct %int2_1903, %int4096_1904, %int20_1905, %int16_1906 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1907 = torch.constant.bool false loc(#loc1) | |
| %7581 = torch.aten.expand %7579, %7580, %false_1907 : !torch.tensor<[2,4096,20,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int0_1908 = torch.constant.int 0 loc(#loc1) | |
| %7582 = torch.aten.clone %7581, %int0_1908 : !torch.tensor<[2,4096,20,16],f32>, !torch.int -> !torch.tensor<[2,4096,20,16],f32> loc(#loc1) | |
| %int2_1909 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1910 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1911 = torch.constant.int 320 loc(#loc1) | |
| %7583 = torch.prim.ListConstruct %int2_1909, %int4096_1910, %int320_1911 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7584 = torch.aten._unsafe_view %7582, %7583 : !torch.tensor<[2,4096,20,16],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7585 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1912 = torch.constant.int 1 loc(#loc1) | |
| %7586 = torch.aten.sub.Tensor %7574, %7585, %int1_1912 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7587 = torch.aten.lift_fresh_copy %5722 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7588 = torch.aten.pow.Tensor_Tensor %7587, %7586 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7589 = torch.aten.neg %7588 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7590 = torch.aten.neg %7589 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7591 = torch.aten.div.Tensor %7584, %7590 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7592 = torch.aten.detach %5721 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7593 = torch.aten.detach %7592 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7594 = torch.aten.div.Tensor %7572, %7591 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1913 = torch.constant.int 1 loc(#loc1) | |
| %7595 = torch.aten.add.Tensor %7594, %7593, %int1_1913 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7596 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1914 = torch.constant.int 1 loc(#loc1) | |
| %7597 = torch.aten.sub.Tensor %7574, %7596, %int1_1914 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7598 = torch.aten.lift_fresh_copy %5720 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7599 = torch.aten.pow.Tensor_Tensor %7598, %7597 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7600 = torch.aten.neg %7599 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7601 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1915 = torch.constant.int 1 loc(#loc1) | |
| %7602 = torch.aten.sub.Tensor %7574, %7601, %int1_1915 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7603 = torch.aten.lift_fresh_copy %5719 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7604 = torch.aten.pow.Tensor_Tensor %7603, %7602 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7605 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_1916 = torch.constant.int 1 loc(#loc1) | |
| %7606 = torch.aten.sub.Tensor %7604, %7605, %int1_1916 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7607 = torch.aten.gt.Tensor %7595, %7606 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7608 = torch.aten.where.self %7607, %7606, %7595 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7609 = torch.aten.lt.Tensor %7608, %7600 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,4096,320],i1> loc(#loc1) | |
| %7610 = torch.aten.where.self %7609, %7600, %7608 : !torch.tensor<[2,4096,320],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7611 = torch.operator "aten.round"(%7610) : (!torch.tensor<[2,4096,320],f32>) -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1917 = torch.constant.int 1 loc(#loc1) | |
| %7612 = torch.aten.sub.Tensor %7611, %7593, %int1_1917 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %7613 = torch.aten.mul.Tensor %7612, %7591 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1918 = torch.constant.int 320 loc(#loc1) | |
| %int20_1919 = torch.constant.int 20 loc(#loc1) | |
| %int16_1920 = torch.constant.int 16 loc(#loc1) | |
| %7614 = torch.prim.ListConstruct %int320_1918, %int20_1919, %int16_1920 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1921 = torch.constant.bool false loc(#loc1) | |
| %7615 = torch.aten.expand %5718, %7614, %false_1921 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int0_1922 = torch.constant.int 0 loc(#loc1) | |
| %7616 = torch.aten.clone %7615, %int0_1922 : !torch.tensor<[320,20,16],f16>, !torch.int -> !torch.tensor<[320,20,16],f16> loc(#loc1) | |
| %int320_1923 = torch.constant.int 320 loc(#loc1) | |
| %int320_1924 = torch.constant.int 320 loc(#loc1) | |
| %7617 = torch.prim.ListConstruct %int320_1923, %int320_1924 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7618 = torch.aten._unsafe_view %7616, %7617 : !torch.tensor<[320,20,16],f16>, !torch.list<int> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %7619 = torch.aten.mul.Tensor %5717, %7618 : !torch.tensor, !torch.tensor<[320,320],f16> -> !torch.tensor<[320,320],f16> loc(#loc1) | |
| %int6_1925 = torch.constant.int 6 loc(#loc1) | |
| %none_1926 = torch.constant.none loc(#loc0) | |
| %none_1927 = torch.constant.none loc(#loc0) | |
| %none_1928 = torch.constant.none loc(#loc0) | |
| %false_1929 = torch.constant.bool false loc(#loc1) | |
| %none_1930 = torch.constant.none loc(#loc0) | |
| %7620 = torch.aten._to_copy %7619, %int6_1925, %none_1926, %none_1927, %none_1928, %false_1929, %none_1930 : !torch.tensor<[320,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %7621 = torch.aten.t %7620 : !torch.tensor<[320,320],f32> -> !torch.tensor<[320,320],f32> loc(#loc1) | |
| %int8192_1931 = torch.constant.int 8192 loc(#loc1) | |
| %int320_1932 = torch.constant.int 320 loc(#loc1) | |
| %7622 = torch.prim.ListConstruct %int8192_1931, %int320_1932 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7623 = torch.aten.view %7613, %7622 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_1933 = torch.constant.int 1 loc(#loc1) | |
| %int1_1934 = torch.constant.int 1 loc(#loc1) | |
| %7624 = torch.aten.addmm %5716, %7623, %7621, %int1_1933, %int1_1934 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_1935 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1936 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1937 = torch.constant.int 320 loc(#loc1) | |
| %7625 = torch.prim.ListConstruct %int2_1935, %int4096_1936, %int320_1937 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7626 = torch.aten.view %7624, %7625 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1938 = torch.constant.int 1 loc(#loc1) | |
| %7627 = torch.aten.add.Tensor %7626, %7345, %int1_1938 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int320_1939 = torch.constant.int 320 loc(#loc1) | |
| %7628 = torch.prim.ListConstruct %int320_1939 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_1940 = torch.constant.none loc(#loc0) | |
| %none_1941 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05_1942 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0_1943, %result1_1944, %result2_1945 = torch.aten.native_layer_norm %7627, %7628, %none_1940, %none_1941, %float1.000000e-05_1942 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,1],f32>, !torch.tensor<[2,4096,1],f32> loc(#loc1) | |
| %int2560_1946 = torch.constant.int 2560 loc(#loc1) | |
| %int20_1947 = torch.constant.int 20 loc(#loc1) | |
| %int16_1948 = torch.constant.int 16 loc(#loc1) | |
| %7629 = torch.prim.ListConstruct %int2560_1946, %int20_1947, %int16_1948 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1949 = torch.constant.bool false loc(#loc1) | |
| %7630 = torch.aten.expand %5715, %7629, %false_1949 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[2560,20,16],f16> loc(#loc1) | |
| %int0_1950 = torch.constant.int 0 loc(#loc1) | |
| %7631 = torch.aten.clone %7630, %int0_1950 : !torch.tensor<[2560,20,16],f16>, !torch.int -> !torch.tensor<[2560,20,16],f16> loc(#loc1) | |
| %int2560_1951 = torch.constant.int 2560 loc(#loc1) | |
| %int320_1952 = torch.constant.int 320 loc(#loc1) | |
| %7632 = torch.prim.ListConstruct %int2560_1951, %int320_1952 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7633 = torch.aten._unsafe_view %7631, %7632 : !torch.tensor<[2560,20,16],f16>, !torch.list<int> -> !torch.tensor<[2560,320],f16> loc(#loc1) | |
| %7634 = torch.aten.mul.Tensor %5714, %7633 : !torch.tensor, !torch.tensor<[2560,320],f16> -> !torch.tensor<[2560,320],f16> loc(#loc1) | |
| %int6_1953 = torch.constant.int 6 loc(#loc1) | |
| %none_1954 = torch.constant.none loc(#loc0) | |
| %none_1955 = torch.constant.none loc(#loc0) | |
| %none_1956 = torch.constant.none loc(#loc0) | |
| %false_1957 = torch.constant.bool false loc(#loc1) | |
| %none_1958 = torch.constant.none loc(#loc0) | |
| %7635 = torch.aten._to_copy %7634, %int6_1953, %none_1954, %none_1955, %none_1956, %false_1957, %none_1958 : !torch.tensor<[2560,320],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[2560,320],f32> loc(#loc1) | |
| %7636 = torch.aten.t %7635 : !torch.tensor<[2560,320],f32> -> !torch.tensor<[320,2560],f32> loc(#loc1) | |
| %int8192_1959 = torch.constant.int 8192 loc(#loc1) | |
| %int320_1960 = torch.constant.int 320 loc(#loc1) | |
| %7637 = torch.prim.ListConstruct %int8192_1959, %int320_1960 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7638 = torch.aten.view %result0_1943, %7637 : !torch.tensor<[2,4096,320],f32>, !torch.list<int> -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int1_1961 = torch.constant.int 1 loc(#loc1) | |
| %int1_1962 = torch.constant.int 1 loc(#loc1) | |
| %7639 = torch.aten.addmm %5713, %7638, %7636, %int1_1961, %int1_1962 : !torch.tensor, !torch.tensor<[8192,320],f32>, !torch.tensor<[320,2560],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,2560],f32> loc(#loc1) | |
| %int2_1963 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1964 = torch.constant.int 4096 loc(#loc1) | |
| %int2560_1965 = torch.constant.int 2560 loc(#loc1) | |
| %7640 = torch.prim.ListConstruct %int2_1963, %int4096_1964, %int2560_1965 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7641 = torch.aten.view %7639, %7640 : !torch.tensor<[8192,2560],f32>, !torch.list<int> -> !torch.tensor<[2,4096,2560],f32> loc(#loc1) | |
| %int-1_1966 = torch.constant.int -1 loc(#loc1) | |
| %int0_1967 = torch.constant.int 0 loc(#loc1) | |
| %int1280_1968 = torch.constant.int 1280 loc(#loc1) | |
| %int1_1969 = torch.constant.int 1 loc(#loc1) | |
| %7642 = torch.aten.slice.Tensor %7641, %int-1_1966, %int0_1967, %int1280_1968, %int1_1969 : !torch.tensor<[2,4096,2560],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,4096,1280],f32> loc(#loc1) | |
| %int-1_1970 = torch.constant.int -1 loc(#loc1) | |
| %int1280_1971 = torch.constant.int 1280 loc(#loc1) | |
| %int2560_1972 = torch.constant.int 2560 loc(#loc1) | |
| %int1_1973 = torch.constant.int 1 loc(#loc1) | |
| %7643 = torch.aten.slice.Tensor %7641, %int-1_1970, %int1280_1971, %int2560_1972, %int1_1973 : !torch.tensor<[2,4096,2560],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,4096,1280],f32> loc(#loc1) | |
| %str_1974 = torch.constant.str "none" loc(#loc1) | |
| %7644 = torch.aten.gelu %7643, %str_1974 : !torch.tensor<[2,4096,1280],f32>, !torch.str -> !torch.tensor<[2,4096,1280],f32> loc(#loc1) | |
| %7645 = torch.aten.mul.Tensor %7642, %7644 : !torch.tensor<[2,4096,1280],f32>, !torch.tensor<[2,4096,1280],f32> -> !torch.tensor<[2,4096,1280],f32> loc(#loc1) | |
| %int320_1975 = torch.constant.int 320 loc(#loc1) | |
| %int80_1976 = torch.constant.int 80 loc(#loc1) | |
| %int16_1977 = torch.constant.int 16 loc(#loc1) | |
| %7646 = torch.prim.ListConstruct %int320_1975, %int80_1976, %int16_1977 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_1978 = torch.constant.bool false loc(#loc1) | |
| %7647 = torch.aten.expand %5712, %7646, %false_1978 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,80,16],f16> loc(#loc1) | |
| %int0_1979 = torch.constant.int 0 loc(#loc1) | |
| %7648 = torch.aten.clone %7647, %int0_1979 : !torch.tensor<[320,80,16],f16>, !torch.int -> !torch.tensor<[320,80,16],f16> loc(#loc1) | |
| %int320_1980 = torch.constant.int 320 loc(#loc1) | |
| %int1280_1981 = torch.constant.int 1280 loc(#loc1) | |
| %7649 = torch.prim.ListConstruct %int320_1980, %int1280_1981 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7650 = torch.aten._unsafe_view %7648, %7649 : !torch.tensor<[320,80,16],f16>, !torch.list<int> -> !torch.tensor<[320,1280],f16> loc(#loc1) | |
| %7651 = torch.aten.mul.Tensor %5711, %7650 : !torch.tensor, !torch.tensor<[320,1280],f16> -> !torch.tensor<[320,1280],f16> loc(#loc1) | |
| %int6_1982 = torch.constant.int 6 loc(#loc1) | |
| %none_1983 = torch.constant.none loc(#loc0) | |
| %none_1984 = torch.constant.none loc(#loc0) | |
| %none_1985 = torch.constant.none loc(#loc0) | |
| %false_1986 = torch.constant.bool false loc(#loc1) | |
| %none_1987 = torch.constant.none loc(#loc0) | |
| %7652 = torch.aten._to_copy %7651, %int6_1982, %none_1983, %none_1984, %none_1985, %false_1986, %none_1987 : !torch.tensor<[320,1280],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,1280],f32> loc(#loc1) | |
| %7653 = torch.aten.t %7652 : !torch.tensor<[320,1280],f32> -> !torch.tensor<[1280,320],f32> loc(#loc1) | |
| %int8192_1988 = torch.constant.int 8192 loc(#loc1) | |
| %int1280_1989 = torch.constant.int 1280 loc(#loc1) | |
| %7654 = torch.prim.ListConstruct %int8192_1988, %int1280_1989 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7655 = torch.aten.view %7645, %7654 : !torch.tensor<[2,4096,1280],f32>, !torch.list<int> -> !torch.tensor<[8192,1280],f32> loc(#loc1) | |
| %int1_1990 = torch.constant.int 1 loc(#loc1) | |
| %int1_1991 = torch.constant.int 1 loc(#loc1) | |
| %7656 = torch.aten.addmm %5710, %7655, %7653, %int1_1990, %int1_1991 : !torch.tensor, !torch.tensor<[8192,1280],f32>, !torch.tensor<[1280,320],f32>, !torch.int, !torch.int -> !torch.tensor<[8192,320],f32> loc(#loc1) | |
| %int2_1992 = torch.constant.int 2 loc(#loc1) | |
| %int4096_1993 = torch.constant.int 4096 loc(#loc1) | |
| %int320_1994 = torch.constant.int 320 loc(#loc1) | |
| %7657 = torch.prim.ListConstruct %int2_1992, %int4096_1993, %int320_1994 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7658 = torch.aten.view %7656, %7657 : !torch.tensor<[8192,320],f32>, !torch.list<int> -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int1_1995 = torch.constant.int 1 loc(#loc1) | |
| %7659 = torch.aten.add.Tensor %7658, %7627, %int1_1995 : !torch.tensor<[2,4096,320],f32>, !torch.tensor<[2,4096,320],f32>, !torch.int -> !torch.tensor<[2,4096,320],f32> loc(#loc1) | |
| %int2_1996 = torch.constant.int 2 loc(#loc1) | |
| %int64_1997 = torch.constant.int 64 loc(#loc1) | |
| %int64_1998 = torch.constant.int 64 loc(#loc1) | |
| %int320_1999 = torch.constant.int 320 loc(#loc1) | |
| %7660 = torch.prim.ListConstruct %int2_1996, %int64_1997, %int64_1998, %int320_1999 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1310720_2000 = torch.constant.int 1310720 loc(#loc1) | |
| %int20480_2001 = torch.constant.int 20480 loc(#loc1) | |
| %int320_2002 = torch.constant.int 320 loc(#loc1) | |
| %int1_2003 = torch.constant.int 1 loc(#loc1) | |
| %7661 = torch.prim.ListConstruct %int1310720_2000, %int20480_2001, %int320_2002, %int1_2003 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7662 = torch.aten._reshape_alias %7659, %7660, %7661 : !torch.tensor<[2,4096,320],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,64,64,320],f32> loc(#loc1) | |
| %int0_2004 = torch.constant.int 0 loc(#loc1) | |
| %int3_2005 = torch.constant.int 3 loc(#loc1) | |
| %int1_2006 = torch.constant.int 1 loc(#loc1) | |
| %int2_2007 = torch.constant.int 2 loc(#loc1) | |
| %7663 = torch.prim.ListConstruct %int0_2004, %int3_2005, %int1_2006, %int2_2007 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7664 = torch.aten.permute %7662, %7663 : !torch.tensor<[2,64,64,320],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7665 = torch.aten.detach %5709 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7666 = torch.aten.detach %7665 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2008 = torch.constant.int 2 loc(#loc1) | |
| %int20_2009 = torch.constant.int 20 loc(#loc1) | |
| %int16_2010 = torch.constant.int 16 loc(#loc1) | |
| %int64_2011 = torch.constant.int 64 loc(#loc1) | |
| %int64_2012 = torch.constant.int 64 loc(#loc1) | |
| %7667 = torch.prim.ListConstruct %int2_2008, %int20_2009, %int16_2010, %int64_2011, %int64_2012 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7668 = torch.aten.view %7664, %7667 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %7669 = torch.aten.abs %7668 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_2013 = torch.constant.int 2 loc(#loc1) | |
| %true_2014 = torch.constant.bool true loc(#loc1) | |
| %values_2015, %indices_2016 = torch.aten.max.dim %7669, %int2_2013, %true_2014 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_2017 = torch.constant.int 2 loc(#loc1) | |
| %int20_2018 = torch.constant.int 20 loc(#loc1) | |
| %int1_2019 = torch.constant.int 1 loc(#loc1) | |
| %int64_2020 = torch.constant.int 64 loc(#loc1) | |
| %int64_2021 = torch.constant.int 64 loc(#loc1) | |
| %7670 = torch.prim.ListConstruct %int2_2017, %int20_2018, %int1_2019, %int64_2020, %int64_2021 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7671 = torch.aten.view %values_2015, %7670 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_2022 = torch.constant.int 2 loc(#loc1) | |
| %int20_2023 = torch.constant.int 20 loc(#loc1) | |
| %int16_2024 = torch.constant.int 16 loc(#loc1) | |
| %int64_2025 = torch.constant.int 64 loc(#loc1) | |
| %int64_2026 = torch.constant.int 64 loc(#loc1) | |
| %7672 = torch.prim.ListConstruct %int2_2022, %int20_2023, %int16_2024, %int64_2025, %int64_2026 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2027 = torch.constant.bool false loc(#loc1) | |
| %7673 = torch.aten.expand %7671, %7672, %false_2027 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_2028 = torch.constant.int 0 loc(#loc1) | |
| %7674 = torch.aten.clone %7673, %int0_2028 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_2029 = torch.constant.int 2 loc(#loc1) | |
| %int320_2030 = torch.constant.int 320 loc(#loc1) | |
| %int64_2031 = torch.constant.int 64 loc(#loc1) | |
| %int64_2032 = torch.constant.int 64 loc(#loc1) | |
| %7675 = torch.prim.ListConstruct %int2_2029, %int320_2030, %int64_2031, %int64_2032 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7676 = torch.aten._unsafe_view %7674, %7675 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7677 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2033 = torch.constant.int 1 loc(#loc1) | |
| %7678 = torch.aten.sub.Tensor %7666, %7677, %int1_2033 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7679 = torch.aten.lift_fresh_copy %5708 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7680 = torch.aten.pow.Tensor_Tensor %7679, %7678 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7681 = torch.aten.neg %7680 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7682 = torch.aten.neg %7681 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7683 = torch.aten.div.Tensor %7676, %7682 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7684 = torch.aten.detach %5707 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7685 = torch.aten.detach %7684 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7686 = torch.aten.div.Tensor %7664, %7683 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_2034 = torch.constant.int 1 loc(#loc1) | |
| %7687 = torch.aten.add.Tensor %7686, %7685, %int1_2034 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7688 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2035 = torch.constant.int 1 loc(#loc1) | |
| %7689 = torch.aten.sub.Tensor %7666, %7688, %int1_2035 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7690 = torch.aten.lift_fresh_copy %5706 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7691 = torch.aten.pow.Tensor_Tensor %7690, %7689 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7692 = torch.aten.neg %7691 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7693 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2036 = torch.constant.int 1 loc(#loc1) | |
| %7694 = torch.aten.sub.Tensor %7666, %7693, %int1_2036 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7695 = torch.aten.lift_fresh_copy %5705 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7696 = torch.aten.pow.Tensor_Tensor %7695, %7694 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7697 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2037 = torch.constant.int 1 loc(#loc1) | |
| %7698 = torch.aten.sub.Tensor %7696, %7697, %int1_2037 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7699 = torch.aten.gt.Tensor %7687, %7698 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %7700 = torch.aten.where.self %7699, %7698, %7687 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7701 = torch.aten.lt.Tensor %7700, %7692 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %7702 = torch.aten.where.self %7701, %7692, %7700 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7703 = torch.operator "aten.round"(%7702) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_2038 = torch.constant.int 1 loc(#loc1) | |
| %7704 = torch.aten.sub.Tensor %7703, %7685, %int1_2038 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7705 = torch.aten.mul.Tensor %7704, %7683 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_2039 = torch.constant.int 320 loc(#loc1) | |
| %int20_2040 = torch.constant.int 20 loc(#loc1) | |
| %int16_2041 = torch.constant.int 16 loc(#loc1) | |
| %int1_2042 = torch.constant.int 1 loc(#loc1) | |
| %int1_2043 = torch.constant.int 1 loc(#loc1) | |
| %7706 = torch.prim.ListConstruct %int320_2039, %int20_2040, %int16_2041, %int1_2042, %int1_2043 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2044 = torch.constant.bool false loc(#loc1) | |
| %7707 = torch.aten.expand %5704, %7706, %false_2044 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,1,1],f16> loc(#loc1) | |
| %int0_2045 = torch.constant.int 0 loc(#loc1) | |
| %7708 = torch.aten.clone %7707, %int0_2045 : !torch.tensor<[320,20,16,1,1],f16>, !torch.int -> !torch.tensor<[320,20,16,1,1],f16> loc(#loc1) | |
| %int320_2046 = torch.constant.int 320 loc(#loc1) | |
| %int320_2047 = torch.constant.int 320 loc(#loc1) | |
| %int1_2048 = torch.constant.int 1 loc(#loc1) | |
| %int1_2049 = torch.constant.int 1 loc(#loc1) | |
| %7709 = torch.prim.ListConstruct %int320_2046, %int320_2047, %int1_2048, %int1_2049 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7710 = torch.aten._unsafe_view %7708, %7709 : !torch.tensor<[320,20,16,1,1],f16>, !torch.list<int> -> !torch.tensor<[320,320,1,1],f16> loc(#loc1) | |
| %7711 = torch.aten.mul.Tensor %5703, %7710 : !torch.tensor, !torch.tensor<[320,320,1,1],f16> -> !torch.tensor<[320,320,1,1],f16> loc(#loc1) | |
| %int6_2050 = torch.constant.int 6 loc(#loc1) | |
| %none_2051 = torch.constant.none loc(#loc0) | |
| %none_2052 = torch.constant.none loc(#loc0) | |
| %none_2053 = torch.constant.none loc(#loc0) | |
| %false_2054 = torch.constant.bool false loc(#loc1) | |
| %none_2055 = torch.constant.none loc(#loc0) | |
| %7712 = torch.aten._to_copy %7711, %int6_2050, %none_2051, %none_2052, %none_2053, %false_2054, %none_2055 : !torch.tensor<[320,320,1,1],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,1,1],f32> loc(#loc1) | |
| %int1_2056 = torch.constant.int 1 loc(#loc1) | |
| %int1_2057 = torch.constant.int 1 loc(#loc1) | |
| %7713 = torch.prim.ListConstruct %int1_2056, %int1_2057 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_2058 = torch.constant.int 0 loc(#loc1) | |
| %int0_2059 = torch.constant.int 0 loc(#loc1) | |
| %7714 = torch.prim.ListConstruct %int0_2058, %int0_2059 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2060 = torch.constant.int 1 loc(#loc1) | |
| %int1_2061 = torch.constant.int 1 loc(#loc1) | |
| %7715 = torch.prim.ListConstruct %int1_2060, %int1_2061 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2062 = torch.constant.bool false loc(#loc1) | |
| %int0_2063 = torch.constant.int 0 loc(#loc1) | |
| %int0_2064 = torch.constant.int 0 loc(#loc1) | |
| %7716 = torch.prim.ListConstruct %int0_2063, %int0_2064 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2065 = torch.constant.int 1 loc(#loc1) | |
| %false_2066 = torch.constant.bool false loc(#loc1) | |
| %false_2067 = torch.constant.bool false loc(#loc1) | |
| %true_2068 = torch.constant.bool true loc(#loc1) | |
| %true_2069 = torch.constant.bool true loc(#loc1) | |
| %7717 = torch.aten._convolution %7705, %7712, %5702, %7713, %7714, %7715, %false_2062, %7716, %int1_2065, %false_2066, %false_2067, %true_2068, %true_2069 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,1,1],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_2070 = torch.constant.int 1 loc(#loc1) | |
| %7718 = torch.aten.add.Tensor %7717, %6994, %int1_2070 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7719 = torch.aten.detach %5701 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7720 = torch.aten.detach %7719 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2071 = torch.constant.int 2 loc(#loc1) | |
| %int20_2072 = torch.constant.int 20 loc(#loc1) | |
| %int16_2073 = torch.constant.int 16 loc(#loc1) | |
| %int64_2074 = torch.constant.int 64 loc(#loc1) | |
| %int64_2075 = torch.constant.int 64 loc(#loc1) | |
| %7721 = torch.prim.ListConstruct %int2_2071, %int20_2072, %int16_2073, %int64_2074, %int64_2075 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7722 = torch.aten.view %7718, %7721 : !torch.tensor<[2,320,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %7723 = torch.aten.abs %7722 : !torch.tensor<[2,20,16,64,64],f32> -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_2076 = torch.constant.int 2 loc(#loc1) | |
| %true_2077 = torch.constant.bool true loc(#loc1) | |
| %values_2078, %indices_2079 = torch.aten.max.dim %7723, %int2_2076, %true_2077 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,64,64],f32>, !torch.tensor<[2,20,1,64,64],si64> loc(#loc1) | |
| %int2_2080 = torch.constant.int 2 loc(#loc1) | |
| %int20_2081 = torch.constant.int 20 loc(#loc1) | |
| %int1_2082 = torch.constant.int 1 loc(#loc1) | |
| %int64_2083 = torch.constant.int 64 loc(#loc1) | |
| %int64_2084 = torch.constant.int 64 loc(#loc1) | |
| %7724 = torch.prim.ListConstruct %int2_2080, %int20_2081, %int1_2082, %int64_2083, %int64_2084 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7725 = torch.aten.view %values_2078, %7724 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,64,64],f32> loc(#loc1) | |
| %int2_2085 = torch.constant.int 2 loc(#loc1) | |
| %int20_2086 = torch.constant.int 20 loc(#loc1) | |
| %int16_2087 = torch.constant.int 16 loc(#loc1) | |
| %int64_2088 = torch.constant.int 64 loc(#loc1) | |
| %int64_2089 = torch.constant.int 64 loc(#loc1) | |
| %7726 = torch.prim.ListConstruct %int2_2085, %int20_2086, %int16_2087, %int64_2088, %int64_2089 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2090 = torch.constant.bool false loc(#loc1) | |
| %7727 = torch.aten.expand %7725, %7726, %false_2090 : !torch.tensor<[2,20,1,64,64],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int0_2091 = torch.constant.int 0 loc(#loc1) | |
| %7728 = torch.aten.clone %7727, %int0_2091 : !torch.tensor<[2,20,16,64,64],f32>, !torch.int -> !torch.tensor<[2,20,16,64,64],f32> loc(#loc1) | |
| %int2_2092 = torch.constant.int 2 loc(#loc1) | |
| %int320_2093 = torch.constant.int 320 loc(#loc1) | |
| %int64_2094 = torch.constant.int 64 loc(#loc1) | |
| %int64_2095 = torch.constant.int 64 loc(#loc1) | |
| %7729 = torch.prim.ListConstruct %int2_2092, %int320_2093, %int64_2094, %int64_2095 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7730 = torch.aten._unsafe_view %7728, %7729 : !torch.tensor<[2,20,16,64,64],f32>, !torch.list<int> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7731 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2096 = torch.constant.int 1 loc(#loc1) | |
| %7732 = torch.aten.sub.Tensor %7720, %7731, %int1_2096 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7733 = torch.aten.lift_fresh_copy %5700 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7734 = torch.aten.pow.Tensor_Tensor %7733, %7732 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7735 = torch.aten.neg %7734 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7736 = torch.aten.neg %7735 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7737 = torch.aten.div.Tensor %7730, %7736 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7738 = torch.aten.detach %5699 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7739 = torch.aten.detach %7738 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7740 = torch.aten.div.Tensor %7718, %7737 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_2097 = torch.constant.int 1 loc(#loc1) | |
| %7741 = torch.aten.add.Tensor %7740, %7739, %int1_2097 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7742 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2098 = torch.constant.int 1 loc(#loc1) | |
| %7743 = torch.aten.sub.Tensor %7720, %7742, %int1_2098 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7744 = torch.aten.lift_fresh_copy %5698 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7745 = torch.aten.pow.Tensor_Tensor %7744, %7743 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7746 = torch.aten.neg %7745 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7747 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2099 = torch.constant.int 1 loc(#loc1) | |
| %7748 = torch.aten.sub.Tensor %7720, %7747, %int1_2099 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7749 = torch.aten.lift_fresh_copy %5697 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7750 = torch.aten.pow.Tensor_Tensor %7749, %7748 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7751 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2100 = torch.constant.int 1 loc(#loc1) | |
| %7752 = torch.aten.sub.Tensor %7750, %7751, %int1_2100 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7753 = torch.aten.gt.Tensor %7741, %7752 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %7754 = torch.aten.where.self %7753, %7752, %7741 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7755 = torch.aten.lt.Tensor %7754, %7746 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,64,64],i1> loc(#loc1) | |
| %7756 = torch.aten.where.self %7755, %7746, %7754 : !torch.tensor<[2,320,64,64],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7757 = torch.operator "aten.round"(%7756) : (!torch.tensor<[2,320,64,64],f32>) -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int1_2101 = torch.constant.int 1 loc(#loc1) | |
| %7758 = torch.aten.sub.Tensor %7757, %7739, %int1_2101 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %7759 = torch.aten.mul.Tensor %7758, %7737 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[2,320,64,64],f32> -> !torch.tensor<[2,320,64,64],f32> loc(#loc1) | |
| %int320_2102 = torch.constant.int 320 loc(#loc1) | |
| %int20_2103 = torch.constant.int 20 loc(#loc1) | |
| %int16_2104 = torch.constant.int 16 loc(#loc1) | |
| %int3_2105 = torch.constant.int 3 loc(#loc1) | |
| %int3_2106 = torch.constant.int 3 loc(#loc1) | |
| %7760 = torch.prim.ListConstruct %int320_2102, %int20_2103, %int16_2104, %int3_2105, %int3_2106 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2107 = torch.constant.bool false loc(#loc1) | |
| %7761 = torch.aten.expand %5696, %7760, %false_2107 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int0_2108 = torch.constant.int 0 loc(#loc1) | |
| %7762 = torch.aten.clone %7761, %int0_2108 : !torch.tensor<[320,20,16,3,3],f16>, !torch.int -> !torch.tensor<[320,20,16,3,3],f16> loc(#loc1) | |
| %int320_2109 = torch.constant.int 320 loc(#loc1) | |
| %int320_2110 = torch.constant.int 320 loc(#loc1) | |
| %int3_2111 = torch.constant.int 3 loc(#loc1) | |
| %int3_2112 = torch.constant.int 3 loc(#loc1) | |
| %7763 = torch.prim.ListConstruct %int320_2109, %int320_2110, %int3_2111, %int3_2112 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7764 = torch.aten._unsafe_view %7762, %7763 : !torch.tensor<[320,20,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %7765 = torch.aten.mul.Tensor %5695, %7764 : !torch.tensor, !torch.tensor<[320,320,3,3],f16> -> !torch.tensor<[320,320,3,3],f16> loc(#loc1) | |
| %int6_2113 = torch.constant.int 6 loc(#loc1) | |
| %none_2114 = torch.constant.none loc(#loc0) | |
| %none_2115 = torch.constant.none loc(#loc0) | |
| %none_2116 = torch.constant.none loc(#loc0) | |
| %false_2117 = torch.constant.bool false loc(#loc1) | |
| %none_2118 = torch.constant.none loc(#loc0) | |
| %7766 = torch.aten._to_copy %7765, %int6_2113, %none_2114, %none_2115, %none_2116, %false_2117, %none_2118 : !torch.tensor<[320,320,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[320,320,3,3],f32> loc(#loc1) | |
| %int2_2119 = torch.constant.int 2 loc(#loc1) | |
| %int2_2120 = torch.constant.int 2 loc(#loc1) | |
| %7767 = torch.prim.ListConstruct %int2_2119, %int2_2120 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2121 = torch.constant.int 1 loc(#loc1) | |
| %int1_2122 = torch.constant.int 1 loc(#loc1) | |
| %7768 = torch.prim.ListConstruct %int1_2121, %int1_2122 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2123 = torch.constant.int 1 loc(#loc1) | |
| %int1_2124 = torch.constant.int 1 loc(#loc1) | |
| %7769 = torch.prim.ListConstruct %int1_2123, %int1_2124 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2125 = torch.constant.bool false loc(#loc1) | |
| %int0_2126 = torch.constant.int 0 loc(#loc1) | |
| %int0_2127 = torch.constant.int 0 loc(#loc1) | |
| %7770 = torch.prim.ListConstruct %int0_2126, %int0_2127 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2128 = torch.constant.int 1 loc(#loc1) | |
| %false_2129 = torch.constant.bool false loc(#loc1) | |
| %false_2130 = torch.constant.bool false loc(#loc1) | |
| %true_2131 = torch.constant.bool true loc(#loc1) | |
| %true_2132 = torch.constant.bool true loc(#loc1) | |
| %7771 = torch.aten._convolution %7759, %7766, %5694, %7767, %7768, %7769, %false_2125, %7770, %int1_2128, %false_2129, %false_2130, %true_2131, %true_2132 : !torch.tensor<[2,320,64,64],f32>, !torch.tensor<[320,320,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int2_2133 = torch.constant.int 2 loc(#loc1) | |
| %int32_2134 = torch.constant.int 32 loc(#loc1) | |
| %int10_2135 = torch.constant.int 10 loc(#loc1) | |
| %int1024 = torch.constant.int 1024 loc(#loc1) | |
| %7772 = torch.prim.ListConstruct %int2_2133, %int32_2134, %int10_2135, %int1024 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7773 = torch.aten.view %7771, %7772 : !torch.tensor<[2,320,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,32,10,1024],f32> loc(#loc1) | |
| %int2_2136 = torch.constant.int 2 loc(#loc1) | |
| %int3_2137 = torch.constant.int 3 loc(#loc1) | |
| %7774 = torch.prim.ListConstruct %int2_2136, %int3_2137 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_2138 = torch.constant.int 0 loc(#loc1) | |
| %true_2139 = torch.constant.bool true loc(#loc1) | |
| %7775 = torch.aten.var.correction %7773, %7774, %int0_2138, %true_2139 : !torch.tensor<[2,32,10,1024],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_2140 = torch.constant.int 2 loc(#loc1) | |
| %int3_2141 = torch.constant.int 3 loc(#loc1) | |
| %7776 = torch.prim.ListConstruct %int2_2140, %int3_2141 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_2142 = torch.constant.bool true loc(#loc1) | |
| %none_2143 = torch.constant.none loc(#loc0) | |
| %7777 = torch.aten.mean.dim %7773, %7776, %true_2142, %none_2143 : !torch.tensor<[2,32,10,1024],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %7778 = torch.tensor.literal(dense<1.000000e-05> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_2144 = torch.constant.int 1 loc(#loc1) | |
| %7779 = torch.aten.add.Tensor %7775, %7778, %int1_2144 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %7780 = torch.aten.rsqrt %7779 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_2145 = torch.constant.int 1 loc(#loc1) | |
| %7781 = torch.aten.sub.Tensor %7773, %7777, %int1_2145 : !torch.tensor<[2,32,10,1024],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,10,1024],f32> loc(#loc1) | |
| %7782 = torch.aten.mul.Tensor %7781, %7780 : !torch.tensor<[2,32,10,1024],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,10,1024],f32> loc(#loc1) | |
| %int2_2146 = torch.constant.int 2 loc(#loc1) | |
| %int320_2147 = torch.constant.int 320 loc(#loc1) | |
| %int32_2148 = torch.constant.int 32 loc(#loc1) | |
| %int32_2149 = torch.constant.int 32 loc(#loc1) | |
| %7783 = torch.prim.ListConstruct %int2_2146, %int320_2147, %int32_2148, %int32_2149 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7784 = torch.aten.view %7782, %7783 : !torch.tensor<[2,32,10,1024],f32>, !torch.list<int> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int-1_2150 = torch.constant.int -1 loc(#loc1) | |
| %7785 = torch.aten.unsqueeze %5693, %int-1_2150 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_2151 = torch.constant.int -1 loc(#loc1) | |
| %7786 = torch.aten.unsqueeze %7785, %int-1_2151 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %7787 = torch.aten.mul.Tensor %7784, %7786 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[320,1,1],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int-1_2152 = torch.constant.int -1 loc(#loc1) | |
| %7788 = torch.aten.unsqueeze %5692, %int-1_2152 : !torch.tensor, !torch.int -> !torch.tensor<[320,1],f32> loc(#loc1) | |
| %int-1_2153 = torch.constant.int -1 loc(#loc1) | |
| %7789 = torch.aten.unsqueeze %7788, %int-1_2153 : !torch.tensor<[320,1],f32>, !torch.int -> !torch.tensor<[320,1,1],f32> loc(#loc1) | |
| %int1_2154 = torch.constant.int 1 loc(#loc1) | |
| %7790 = torch.aten.add.Tensor %7787, %7789, %int1_2154 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[320,1,1],f32>, !torch.int -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7791 = torch.aten.silu %7790 : !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7792 = torch.aten.detach %5691 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7793 = torch.aten.detach %7792 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2155 = torch.constant.int 2 loc(#loc1) | |
| %int20_2156 = torch.constant.int 20 loc(#loc1) | |
| %int16_2157 = torch.constant.int 16 loc(#loc1) | |
| %int32_2158 = torch.constant.int 32 loc(#loc1) | |
| %int32_2159 = torch.constant.int 32 loc(#loc1) | |
| %7794 = torch.prim.ListConstruct %int2_2155, %int20_2156, %int16_2157, %int32_2158, %int32_2159 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7795 = torch.aten.view %7791, %7794 : !torch.tensor<[2,320,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,32,32],f32> loc(#loc1) | |
| %7796 = torch.aten.abs %7795 : !torch.tensor<[2,20,16,32,32],f32> -> !torch.tensor<[2,20,16,32,32],f32> loc(#loc1) | |
| %int2_2160 = torch.constant.int 2 loc(#loc1) | |
| %true_2161 = torch.constant.bool true loc(#loc1) | |
| %values_2162, %indices_2163 = torch.aten.max.dim %7796, %int2_2160, %true_2161 : !torch.tensor<[2,20,16,32,32],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,32,32],f32>, !torch.tensor<[2,20,1,32,32],si64> loc(#loc1) | |
| %int2_2164 = torch.constant.int 2 loc(#loc1) | |
| %int20_2165 = torch.constant.int 20 loc(#loc1) | |
| %int1_2166 = torch.constant.int 1 loc(#loc1) | |
| %int32_2167 = torch.constant.int 32 loc(#loc1) | |
| %int32_2168 = torch.constant.int 32 loc(#loc1) | |
| %7797 = torch.prim.ListConstruct %int2_2164, %int20_2165, %int1_2166, %int32_2167, %int32_2168 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7798 = torch.aten.view %values_2162, %7797 : !torch.tensor<[2,20,1,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,32,32],f32> loc(#loc1) | |
| %int2_2169 = torch.constant.int 2 loc(#loc1) | |
| %int20_2170 = torch.constant.int 20 loc(#loc1) | |
| %int16_2171 = torch.constant.int 16 loc(#loc1) | |
| %int32_2172 = torch.constant.int 32 loc(#loc1) | |
| %int32_2173 = torch.constant.int 32 loc(#loc1) | |
| %7799 = torch.prim.ListConstruct %int2_2169, %int20_2170, %int16_2171, %int32_2172, %int32_2173 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2174 = torch.constant.bool false loc(#loc1) | |
| %7800 = torch.aten.expand %7798, %7799, %false_2174 : !torch.tensor<[2,20,1,32,32],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,32,32],f32> loc(#loc1) | |
| %int0_2175 = torch.constant.int 0 loc(#loc1) | |
| %7801 = torch.aten.clone %7800, %int0_2175 : !torch.tensor<[2,20,16,32,32],f32>, !torch.int -> !torch.tensor<[2,20,16,32,32],f32> loc(#loc1) | |
| %int2_2176 = torch.constant.int 2 loc(#loc1) | |
| %int320_2177 = torch.constant.int 320 loc(#loc1) | |
| %int32_2178 = torch.constant.int 32 loc(#loc1) | |
| %int32_2179 = torch.constant.int 32 loc(#loc1) | |
| %7802 = torch.prim.ListConstruct %int2_2176, %int320_2177, %int32_2178, %int32_2179 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7803 = torch.aten._unsafe_view %7801, %7802 : !torch.tensor<[2,20,16,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7804 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2180 = torch.constant.int 1 loc(#loc1) | |
| %7805 = torch.aten.sub.Tensor %7793, %7804, %int1_2180 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7806 = torch.aten.lift_fresh_copy %5690 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7807 = torch.aten.pow.Tensor_Tensor %7806, %7805 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7808 = torch.aten.neg %7807 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7809 = torch.aten.neg %7808 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7810 = torch.aten.div.Tensor %7803, %7809 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7811 = torch.aten.detach %5689 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7812 = torch.aten.detach %7811 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7813 = torch.aten.div.Tensor %7791, %7810 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int1_2181 = torch.constant.int 1 loc(#loc1) | |
| %7814 = torch.aten.add.Tensor %7813, %7812, %int1_2181 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7815 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2182 = torch.constant.int 1 loc(#loc1) | |
| %7816 = torch.aten.sub.Tensor %7793, %7815, %int1_2182 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7817 = torch.aten.lift_fresh_copy %5688 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7818 = torch.aten.pow.Tensor_Tensor %7817, %7816 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7819 = torch.aten.neg %7818 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7820 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2183 = torch.constant.int 1 loc(#loc1) | |
| %7821 = torch.aten.sub.Tensor %7793, %7820, %int1_2183 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7822 = torch.aten.lift_fresh_copy %5687 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7823 = torch.aten.pow.Tensor_Tensor %7822, %7821 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7824 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2184 = torch.constant.int 1 loc(#loc1) | |
| %7825 = torch.aten.sub.Tensor %7823, %7824, %int1_2184 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7826 = torch.aten.gt.Tensor %7814, %7825 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,32,32],i1> loc(#loc1) | |
| %7827 = torch.aten.where.self %7826, %7825, %7814 : !torch.tensor<[2,320,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7828 = torch.aten.lt.Tensor %7827, %7819 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,32,32],i1> loc(#loc1) | |
| %7829 = torch.aten.where.self %7828, %7819, %7827 : !torch.tensor<[2,320,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7830 = torch.operator "aten.round"(%7829) : (!torch.tensor<[2,320,32,32],f32>) -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int1_2185 = torch.constant.int 1 loc(#loc1) | |
| %7831 = torch.aten.sub.Tensor %7830, %7812, %int1_2185 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7832 = torch.aten.mul.Tensor %7831, %7810 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int640 = torch.constant.int 640 loc(#loc1) | |
| %int20_2186 = torch.constant.int 20 loc(#loc1) | |
| %int16_2187 = torch.constant.int 16 loc(#loc1) | |
| %int3_2188 = torch.constant.int 3 loc(#loc1) | |
| %int3_2189 = torch.constant.int 3 loc(#loc1) | |
| %7833 = torch.prim.ListConstruct %int640, %int20_2186, %int16_2187, %int3_2188, %int3_2189 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2190 = torch.constant.bool false loc(#loc1) | |
| %7834 = torch.aten.expand %5686, %7833, %false_2190 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,20,16,3,3],f16> loc(#loc1) | |
| %int0_2191 = torch.constant.int 0 loc(#loc1) | |
| %7835 = torch.aten.clone %7834, %int0_2191 : !torch.tensor<[640,20,16,3,3],f16>, !torch.int -> !torch.tensor<[640,20,16,3,3],f16> loc(#loc1) | |
| %int640_2192 = torch.constant.int 640 loc(#loc1) | |
| %int320_2193 = torch.constant.int 320 loc(#loc1) | |
| %int3_2194 = torch.constant.int 3 loc(#loc1) | |
| %int3_2195 = torch.constant.int 3 loc(#loc1) | |
| %7836 = torch.prim.ListConstruct %int640_2192, %int320_2193, %int3_2194, %int3_2195 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7837 = torch.aten._unsafe_view %7835, %7836 : !torch.tensor<[640,20,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[640,320,3,3],f16> loc(#loc1) | |
| %7838 = torch.aten.mul.Tensor %5685, %7837 : !torch.tensor, !torch.tensor<[640,320,3,3],f16> -> !torch.tensor<[640,320,3,3],f16> loc(#loc1) | |
| %int6_2196 = torch.constant.int 6 loc(#loc1) | |
| %none_2197 = torch.constant.none loc(#loc0) | |
| %none_2198 = torch.constant.none loc(#loc0) | |
| %none_2199 = torch.constant.none loc(#loc0) | |
| %false_2200 = torch.constant.bool false loc(#loc1) | |
| %none_2201 = torch.constant.none loc(#loc0) | |
| %7839 = torch.aten._to_copy %7838, %int6_2196, %none_2197, %none_2198, %none_2199, %false_2200, %none_2201 : !torch.tensor<[640,320,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,320,3,3],f32> loc(#loc1) | |
| %int1_2202 = torch.constant.int 1 loc(#loc1) | |
| %int1_2203 = torch.constant.int 1 loc(#loc1) | |
| %7840 = torch.prim.ListConstruct %int1_2202, %int1_2203 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2204 = torch.constant.int 1 loc(#loc1) | |
| %int1_2205 = torch.constant.int 1 loc(#loc1) | |
| %7841 = torch.prim.ListConstruct %int1_2204, %int1_2205 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2206 = torch.constant.int 1 loc(#loc1) | |
| %int1_2207 = torch.constant.int 1 loc(#loc1) | |
| %7842 = torch.prim.ListConstruct %int1_2206, %int1_2207 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2208 = torch.constant.bool false loc(#loc1) | |
| %int0_2209 = torch.constant.int 0 loc(#loc1) | |
| %int0_2210 = torch.constant.int 0 loc(#loc1) | |
| %7843 = torch.prim.ListConstruct %int0_2209, %int0_2210 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2211 = torch.constant.int 1 loc(#loc1) | |
| %false_2212 = torch.constant.bool false loc(#loc1) | |
| %false_2213 = torch.constant.bool false loc(#loc1) | |
| %true_2214 = torch.constant.bool true loc(#loc1) | |
| %true_2215 = torch.constant.bool true loc(#loc1) | |
| %7844 = torch.aten._convolution %7832, %7839, %5684, %7840, %7841, %7842, %false_2208, %7843, %int1_2211, %false_2212, %false_2213, %true_2214, %true_2215 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[640,320,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7845 = torch.aten.silu %5951 : !torch.tensor<[2,1280],f32> -> !torch.tensor<[2,1280],f32> loc(#loc1) | |
| %7846 = torch.aten.t %5683 : !torch.tensor -> !torch.tensor<[1280,640],f32> loc(#loc1) | |
| %int1_2216 = torch.constant.int 1 loc(#loc1) | |
| %int1_2217 = torch.constant.int 1 loc(#loc1) | |
| %7847 = torch.aten.addmm %5682, %7845, %7846, %int1_2216, %int1_2217 : !torch.tensor, !torch.tensor<[2,1280],f32>, !torch.tensor<[1280,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2,640],f32> loc(#loc1) | |
| %int0_2218 = torch.constant.int 0 loc(#loc1) | |
| %int0_2219 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_2220 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_2221 = torch.constant.int 1 loc(#loc1) | |
| %7848 = torch.aten.slice.Tensor %7847, %int0_2218, %int0_2219, %int9223372036854775807_2220, %int1_2221 : !torch.tensor<[2,640],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,640],f32> loc(#loc1) | |
| %int1_2222 = torch.constant.int 1 loc(#loc1) | |
| %int0_2223 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_2224 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_2225 = torch.constant.int 1 loc(#loc1) | |
| %7849 = torch.aten.slice.Tensor %7848, %int1_2222, %int0_2223, %int9223372036854775807_2224, %int1_2225 : !torch.tensor<[2,640],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,640],f32> loc(#loc1) | |
| %int2_2226 = torch.constant.int 2 loc(#loc1) | |
| %7850 = torch.aten.unsqueeze %7849, %int2_2226 : !torch.tensor<[2,640],f32>, !torch.int -> !torch.tensor<[2,640,1],f32> loc(#loc1) | |
| %int3_2227 = torch.constant.int 3 loc(#loc1) | |
| %7851 = torch.aten.unsqueeze %7850, %int3_2227 : !torch.tensor<[2,640,1],f32>, !torch.int -> !torch.tensor<[2,640,1,1],f32> loc(#loc1) | |
| %int1_2228 = torch.constant.int 1 loc(#loc1) | |
| %7852 = torch.aten.add.Tensor %7844, %7851, %int1_2228 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,1,1],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int2_2229 = torch.constant.int 2 loc(#loc1) | |
| %int32_2230 = torch.constant.int 32 loc(#loc1) | |
| %int20_2231 = torch.constant.int 20 loc(#loc1) | |
| %int1024_2232 = torch.constant.int 1024 loc(#loc1) | |
| %7853 = torch.prim.ListConstruct %int2_2229, %int32_2230, %int20_2231, %int1024_2232 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7854 = torch.aten.view %7852, %7853 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_2233 = torch.constant.int 2 loc(#loc1) | |
| %int3_2234 = torch.constant.int 3 loc(#loc1) | |
| %7855 = torch.prim.ListConstruct %int2_2233, %int3_2234 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_2235 = torch.constant.int 0 loc(#loc1) | |
| %true_2236 = torch.constant.bool true loc(#loc1) | |
| %7856 = torch.aten.var.correction %7854, %7855, %int0_2235, %true_2236 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_2237 = torch.constant.int 2 loc(#loc1) | |
| %int3_2238 = torch.constant.int 3 loc(#loc1) | |
| %7857 = torch.prim.ListConstruct %int2_2237, %int3_2238 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_2239 = torch.constant.bool true loc(#loc1) | |
| %none_2240 = torch.constant.none loc(#loc0) | |
| %7858 = torch.aten.mean.dim %7854, %7857, %true_2239, %none_2240 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %7859 = torch.tensor.literal(dense<1.000000e-05> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_2241 = torch.constant.int 1 loc(#loc1) | |
| %7860 = torch.aten.add.Tensor %7856, %7859, %int1_2241 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %7861 = torch.aten.rsqrt %7860 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_2242 = torch.constant.int 1 loc(#loc1) | |
| %7862 = torch.aten.sub.Tensor %7854, %7858, %int1_2242 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %7863 = torch.aten.mul.Tensor %7862, %7861 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_2243 = torch.constant.int 2 loc(#loc1) | |
| %int640_2244 = torch.constant.int 640 loc(#loc1) | |
| %int32_2245 = torch.constant.int 32 loc(#loc1) | |
| %int32_2246 = torch.constant.int 32 loc(#loc1) | |
| %7864 = torch.prim.ListConstruct %int2_2243, %int640_2244, %int32_2245, %int32_2246 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7865 = torch.aten.view %7863, %7864 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int-1_2247 = torch.constant.int -1 loc(#loc1) | |
| %7866 = torch.aten.unsqueeze %5681, %int-1_2247 : !torch.tensor, !torch.int -> !torch.tensor<[640,1],f32> loc(#loc1) | |
| %int-1_2248 = torch.constant.int -1 loc(#loc1) | |
| %7867 = torch.aten.unsqueeze %7866, %int-1_2248 : !torch.tensor<[640,1],f32>, !torch.int -> !torch.tensor<[640,1,1],f32> loc(#loc1) | |
| %7868 = torch.aten.mul.Tensor %7865, %7867 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,1,1],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int-1_2249 = torch.constant.int -1 loc(#loc1) | |
| %7869 = torch.aten.unsqueeze %5680, %int-1_2249 : !torch.tensor, !torch.int -> !torch.tensor<[640,1],f32> loc(#loc1) | |
| %int-1_2250 = torch.constant.int -1 loc(#loc1) | |
| %7870 = torch.aten.unsqueeze %7869, %int-1_2250 : !torch.tensor<[640,1],f32>, !torch.int -> !torch.tensor<[640,1,1],f32> loc(#loc1) | |
| %int1_2251 = torch.constant.int 1 loc(#loc1) | |
| %7871 = torch.aten.add.Tensor %7868, %7870, %int1_2251 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,1,1],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7872 = torch.aten.silu %7871 : !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7873 = torch.aten.detach %5679 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7874 = torch.aten.detach %7873 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2252 = torch.constant.int 2 loc(#loc1) | |
| %int40_2253 = torch.constant.int 40 loc(#loc1) | |
| %int16_2254 = torch.constant.int 16 loc(#loc1) | |
| %int32_2255 = torch.constant.int 32 loc(#loc1) | |
| %int32_2256 = torch.constant.int 32 loc(#loc1) | |
| %7875 = torch.prim.ListConstruct %int2_2252, %int40_2253, %int16_2254, %int32_2255, %int32_2256 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7876 = torch.aten.view %7872, %7875 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %7877 = torch.aten.abs %7876 : !torch.tensor<[2,40,16,32,32],f32> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_2257 = torch.constant.int 2 loc(#loc1) | |
| %true_2258 = torch.constant.bool true loc(#loc1) | |
| %values_2259, %indices_2260 = torch.aten.max.dim %7877, %int2_2257, %true_2258 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,40,1,32,32],f32>, !torch.tensor<[2,40,1,32,32],si64> loc(#loc1) | |
| %int2_2261 = torch.constant.int 2 loc(#loc1) | |
| %int40_2262 = torch.constant.int 40 loc(#loc1) | |
| %int1_2263 = torch.constant.int 1 loc(#loc1) | |
| %int32_2264 = torch.constant.int 32 loc(#loc1) | |
| %int32_2265 = torch.constant.int 32 loc(#loc1) | |
| %7878 = torch.prim.ListConstruct %int2_2261, %int40_2262, %int1_2263, %int32_2264, %int32_2265 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7879 = torch.aten.view %values_2259, %7878 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,1,32,32],f32> loc(#loc1) | |
| %int2_2266 = torch.constant.int 2 loc(#loc1) | |
| %int40_2267 = torch.constant.int 40 loc(#loc1) | |
| %int16_2268 = torch.constant.int 16 loc(#loc1) | |
| %int32_2269 = torch.constant.int 32 loc(#loc1) | |
| %int32_2270 = torch.constant.int 32 loc(#loc1) | |
| %7880 = torch.prim.ListConstruct %int2_2266, %int40_2267, %int16_2268, %int32_2269, %int32_2270 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2271 = torch.constant.bool false loc(#loc1) | |
| %7881 = torch.aten.expand %7879, %7880, %false_2271 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int0_2272 = torch.constant.int 0 loc(#loc1) | |
| %7882 = torch.aten.clone %7881, %int0_2272 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_2273 = torch.constant.int 2 loc(#loc1) | |
| %int640_2274 = torch.constant.int 640 loc(#loc1) | |
| %int32_2275 = torch.constant.int 32 loc(#loc1) | |
| %int32_2276 = torch.constant.int 32 loc(#loc1) | |
| %7883 = torch.prim.ListConstruct %int2_2273, %int640_2274, %int32_2275, %int32_2276 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7884 = torch.aten._unsafe_view %7882, %7883 : !torch.tensor<[2,40,16,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7885 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2277 = torch.constant.int 1 loc(#loc1) | |
| %7886 = torch.aten.sub.Tensor %7874, %7885, %int1_2277 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7887 = torch.aten.lift_fresh_copy %5678 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7888 = torch.aten.pow.Tensor_Tensor %7887, %7886 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7889 = torch.aten.neg %7888 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7890 = torch.aten.neg %7889 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7891 = torch.aten.div.Tensor %7884, %7890 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7892 = torch.aten.detach %5677 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7893 = torch.aten.detach %7892 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7894 = torch.aten.div.Tensor %7872, %7891 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_2278 = torch.constant.int 1 loc(#loc1) | |
| %7895 = torch.aten.add.Tensor %7894, %7893, %int1_2278 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7896 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2279 = torch.constant.int 1 loc(#loc1) | |
| %7897 = torch.aten.sub.Tensor %7874, %7896, %int1_2279 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7898 = torch.aten.lift_fresh_copy %5676 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7899 = torch.aten.pow.Tensor_Tensor %7898, %7897 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7900 = torch.aten.neg %7899 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7901 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2280 = torch.constant.int 1 loc(#loc1) | |
| %7902 = torch.aten.sub.Tensor %7874, %7901, %int1_2280 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7903 = torch.aten.lift_fresh_copy %5675 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7904 = torch.aten.pow.Tensor_Tensor %7903, %7902 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7905 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2281 = torch.constant.int 1 loc(#loc1) | |
| %7906 = torch.aten.sub.Tensor %7904, %7905, %int1_2281 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7907 = torch.aten.gt.Tensor %7895, %7906 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %7908 = torch.aten.where.self %7907, %7906, %7895 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7909 = torch.aten.lt.Tensor %7908, %7900 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %7910 = torch.aten.where.self %7909, %7900, %7908 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7911 = torch.operator "aten.round"(%7910) : (!torch.tensor<[2,640,32,32],f32>) -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_2282 = torch.constant.int 1 loc(#loc1) | |
| %7912 = torch.aten.sub.Tensor %7911, %7893, %int1_2282 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7913 = torch.aten.mul.Tensor %7912, %7891 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int640_2283 = torch.constant.int 640 loc(#loc1) | |
| %int40_2284 = torch.constant.int 40 loc(#loc1) | |
| %int16_2285 = torch.constant.int 16 loc(#loc1) | |
| %int3_2286 = torch.constant.int 3 loc(#loc1) | |
| %int3_2287 = torch.constant.int 3 loc(#loc1) | |
| %7914 = torch.prim.ListConstruct %int640_2283, %int40_2284, %int16_2285, %int3_2286, %int3_2287 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2288 = torch.constant.bool false loc(#loc1) | |
| %7915 = torch.aten.expand %5674, %7914, %false_2288 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16,3,3],f16> loc(#loc1) | |
| %int0_2289 = torch.constant.int 0 loc(#loc1) | |
| %7916 = torch.aten.clone %7915, %int0_2289 : !torch.tensor<[640,40,16,3,3],f16>, !torch.int -> !torch.tensor<[640,40,16,3,3],f16> loc(#loc1) | |
| %int640_2290 = torch.constant.int 640 loc(#loc1) | |
| %int640_2291 = torch.constant.int 640 loc(#loc1) | |
| %int3_2292 = torch.constant.int 3 loc(#loc1) | |
| %int3_2293 = torch.constant.int 3 loc(#loc1) | |
| %7917 = torch.prim.ListConstruct %int640_2290, %int640_2291, %int3_2292, %int3_2293 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7918 = torch.aten._unsafe_view %7916, %7917 : !torch.tensor<[640,40,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[640,640,3,3],f16> loc(#loc1) | |
| %7919 = torch.aten.mul.Tensor %5673, %7918 : !torch.tensor, !torch.tensor<[640,640,3,3],f16> -> !torch.tensor<[640,640,3,3],f16> loc(#loc1) | |
| %int6_2294 = torch.constant.int 6 loc(#loc1) | |
| %none_2295 = torch.constant.none loc(#loc0) | |
| %none_2296 = torch.constant.none loc(#loc0) | |
| %none_2297 = torch.constant.none loc(#loc0) | |
| %false_2298 = torch.constant.bool false loc(#loc1) | |
| %none_2299 = torch.constant.none loc(#loc0) | |
| %7920 = torch.aten._to_copy %7919, %int6_2294, %none_2295, %none_2296, %none_2297, %false_2298, %none_2299 : !torch.tensor<[640,640,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640,3,3],f32> loc(#loc1) | |
| %int1_2300 = torch.constant.int 1 loc(#loc1) | |
| %int1_2301 = torch.constant.int 1 loc(#loc1) | |
| %7921 = torch.prim.ListConstruct %int1_2300, %int1_2301 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2302 = torch.constant.int 1 loc(#loc1) | |
| %int1_2303 = torch.constant.int 1 loc(#loc1) | |
| %7922 = torch.prim.ListConstruct %int1_2302, %int1_2303 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2304 = torch.constant.int 1 loc(#loc1) | |
| %int1_2305 = torch.constant.int 1 loc(#loc1) | |
| %7923 = torch.prim.ListConstruct %int1_2304, %int1_2305 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2306 = torch.constant.bool false loc(#loc1) | |
| %int0_2307 = torch.constant.int 0 loc(#loc1) | |
| %int0_2308 = torch.constant.int 0 loc(#loc1) | |
| %7924 = torch.prim.ListConstruct %int0_2307, %int0_2308 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2309 = torch.constant.int 1 loc(#loc1) | |
| %false_2310 = torch.constant.bool false loc(#loc1) | |
| %false_2311 = torch.constant.bool false loc(#loc1) | |
| %true_2312 = torch.constant.bool true loc(#loc1) | |
| %true_2313 = torch.constant.bool true loc(#loc1) | |
| %7925 = torch.aten._convolution %7913, %7920, %5672, %7921, %7922, %7923, %false_2306, %7924, %int1_2309, %false_2310, %false_2311, %true_2312, %true_2313 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,640,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7926 = torch.aten.detach %5671 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7927 = torch.aten.detach %7926 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2314 = torch.constant.int 2 loc(#loc1) | |
| %int20_2315 = torch.constant.int 20 loc(#loc1) | |
| %int16_2316 = torch.constant.int 16 loc(#loc1) | |
| %int32_2317 = torch.constant.int 32 loc(#loc1) | |
| %int32_2318 = torch.constant.int 32 loc(#loc1) | |
| %7928 = torch.prim.ListConstruct %int2_2314, %int20_2315, %int16_2316, %int32_2317, %int32_2318 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7929 = torch.aten.view %7771, %7928 : !torch.tensor<[2,320,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,20,16,32,32],f32> loc(#loc1) | |
| %7930 = torch.aten.abs %7929 : !torch.tensor<[2,20,16,32,32],f32> -> !torch.tensor<[2,20,16,32,32],f32> loc(#loc1) | |
| %int2_2319 = torch.constant.int 2 loc(#loc1) | |
| %true_2320 = torch.constant.bool true loc(#loc1) | |
| %values_2321, %indices_2322 = torch.aten.max.dim %7930, %int2_2319, %true_2320 : !torch.tensor<[2,20,16,32,32],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,20,1,32,32],f32>, !torch.tensor<[2,20,1,32,32],si64> loc(#loc1) | |
| %int2_2323 = torch.constant.int 2 loc(#loc1) | |
| %int20_2324 = torch.constant.int 20 loc(#loc1) | |
| %int1_2325 = torch.constant.int 1 loc(#loc1) | |
| %int32_2326 = torch.constant.int 32 loc(#loc1) | |
| %int32_2327 = torch.constant.int 32 loc(#loc1) | |
| %7931 = torch.prim.ListConstruct %int2_2323, %int20_2324, %int1_2325, %int32_2326, %int32_2327 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7932 = torch.aten.view %values_2321, %7931 : !torch.tensor<[2,20,1,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,20,1,32,32],f32> loc(#loc1) | |
| %int2_2328 = torch.constant.int 2 loc(#loc1) | |
| %int20_2329 = torch.constant.int 20 loc(#loc1) | |
| %int16_2330 = torch.constant.int 16 loc(#loc1) | |
| %int32_2331 = torch.constant.int 32 loc(#loc1) | |
| %int32_2332 = torch.constant.int 32 loc(#loc1) | |
| %7933 = torch.prim.ListConstruct %int2_2328, %int20_2329, %int16_2330, %int32_2331, %int32_2332 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2333 = torch.constant.bool false loc(#loc1) | |
| %7934 = torch.aten.expand %7932, %7933, %false_2333 : !torch.tensor<[2,20,1,32,32],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,20,16,32,32],f32> loc(#loc1) | |
| %int0_2334 = torch.constant.int 0 loc(#loc1) | |
| %7935 = torch.aten.clone %7934, %int0_2334 : !torch.tensor<[2,20,16,32,32],f32>, !torch.int -> !torch.tensor<[2,20,16,32,32],f32> loc(#loc1) | |
| %int2_2335 = torch.constant.int 2 loc(#loc1) | |
| %int320_2336 = torch.constant.int 320 loc(#loc1) | |
| %int32_2337 = torch.constant.int 32 loc(#loc1) | |
| %int32_2338 = torch.constant.int 32 loc(#loc1) | |
| %7936 = torch.prim.ListConstruct %int2_2335, %int320_2336, %int32_2337, %int32_2338 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7937 = torch.aten._unsafe_view %7935, %7936 : !torch.tensor<[2,20,16,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7938 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2339 = torch.constant.int 1 loc(#loc1) | |
| %7939 = torch.aten.sub.Tensor %7927, %7938, %int1_2339 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7940 = torch.aten.lift_fresh_copy %5670 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7941 = torch.aten.pow.Tensor_Tensor %7940, %7939 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7942 = torch.aten.neg %7941 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7943 = torch.aten.neg %7942 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7944 = torch.aten.div.Tensor %7937, %7943 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7945 = torch.aten.detach %5669 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7946 = torch.aten.detach %7945 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7947 = torch.aten.div.Tensor %7771, %7944 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int1_2340 = torch.constant.int 1 loc(#loc1) | |
| %7948 = torch.aten.add.Tensor %7947, %7946, %int1_2340 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7949 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2341 = torch.constant.int 1 loc(#loc1) | |
| %7950 = torch.aten.sub.Tensor %7927, %7949, %int1_2341 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7951 = torch.aten.lift_fresh_copy %5668 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7952 = torch.aten.pow.Tensor_Tensor %7951, %7950 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7953 = torch.aten.neg %7952 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7954 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2342 = torch.constant.int 1 loc(#loc1) | |
| %7955 = torch.aten.sub.Tensor %7927, %7954, %int1_2342 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7956 = torch.aten.lift_fresh_copy %5667 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7957 = torch.aten.pow.Tensor_Tensor %7956, %7955 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %7958 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2343 = torch.constant.int 1 loc(#loc1) | |
| %7959 = torch.aten.sub.Tensor %7957, %7958, %int1_2343 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %7960 = torch.aten.gt.Tensor %7948, %7959 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,32,32],i1> loc(#loc1) | |
| %7961 = torch.aten.where.self %7960, %7959, %7948 : !torch.tensor<[2,320,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7962 = torch.aten.lt.Tensor %7961, %7953 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,320,32,32],i1> loc(#loc1) | |
| %7963 = torch.aten.where.self %7962, %7953, %7961 : !torch.tensor<[2,320,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7964 = torch.operator "aten.round"(%7963) : (!torch.tensor<[2,320,32,32],f32>) -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int1_2344 = torch.constant.int 1 loc(#loc1) | |
| %7965 = torch.aten.sub.Tensor %7964, %7946, %int1_2344 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %7966 = torch.aten.mul.Tensor %7965, %7944 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[2,320,32,32],f32> -> !torch.tensor<[2,320,32,32],f32> loc(#loc1) | |
| %int640_2345 = torch.constant.int 640 loc(#loc1) | |
| %int20_2346 = torch.constant.int 20 loc(#loc1) | |
| %int16_2347 = torch.constant.int 16 loc(#loc1) | |
| %int1_2348 = torch.constant.int 1 loc(#loc1) | |
| %int1_2349 = torch.constant.int 1 loc(#loc1) | |
| %7967 = torch.prim.ListConstruct %int640_2345, %int20_2346, %int16_2347, %int1_2348, %int1_2349 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2350 = torch.constant.bool false loc(#loc1) | |
| %7968 = torch.aten.expand %5666, %7967, %false_2350 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,20,16,1,1],f16> loc(#loc1) | |
| %int0_2351 = torch.constant.int 0 loc(#loc1) | |
| %7969 = torch.aten.clone %7968, %int0_2351 : !torch.tensor<[640,20,16,1,1],f16>, !torch.int -> !torch.tensor<[640,20,16,1,1],f16> loc(#loc1) | |
| %int640_2352 = torch.constant.int 640 loc(#loc1) | |
| %int320_2353 = torch.constant.int 320 loc(#loc1) | |
| %int1_2354 = torch.constant.int 1 loc(#loc1) | |
| %int1_2355 = torch.constant.int 1 loc(#loc1) | |
| %7970 = torch.prim.ListConstruct %int640_2352, %int320_2353, %int1_2354, %int1_2355 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7971 = torch.aten._unsafe_view %7969, %7970 : !torch.tensor<[640,20,16,1,1],f16>, !torch.list<int> -> !torch.tensor<[640,320,1,1],f16> loc(#loc1) | |
| %7972 = torch.aten.mul.Tensor %5665, %7971 : !torch.tensor, !torch.tensor<[640,320,1,1],f16> -> !torch.tensor<[640,320,1,1],f16> loc(#loc1) | |
| %int6_2356 = torch.constant.int 6 loc(#loc1) | |
| %none_2357 = torch.constant.none loc(#loc0) | |
| %none_2358 = torch.constant.none loc(#loc0) | |
| %none_2359 = torch.constant.none loc(#loc0) | |
| %false_2360 = torch.constant.bool false loc(#loc1) | |
| %none_2361 = torch.constant.none loc(#loc0) | |
| %7973 = torch.aten._to_copy %7972, %int6_2356, %none_2357, %none_2358, %none_2359, %false_2360, %none_2361 : !torch.tensor<[640,320,1,1],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,320,1,1],f32> loc(#loc1) | |
| %int1_2362 = torch.constant.int 1 loc(#loc1) | |
| %int1_2363 = torch.constant.int 1 loc(#loc1) | |
| %7974 = torch.prim.ListConstruct %int1_2362, %int1_2363 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_2364 = torch.constant.int 0 loc(#loc1) | |
| %int0_2365 = torch.constant.int 0 loc(#loc1) | |
| %7975 = torch.prim.ListConstruct %int0_2364, %int0_2365 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2366 = torch.constant.int 1 loc(#loc1) | |
| %int1_2367 = torch.constant.int 1 loc(#loc1) | |
| %7976 = torch.prim.ListConstruct %int1_2366, %int1_2367 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2368 = torch.constant.bool false loc(#loc1) | |
| %int0_2369 = torch.constant.int 0 loc(#loc1) | |
| %int0_2370 = torch.constant.int 0 loc(#loc1) | |
| %7977 = torch.prim.ListConstruct %int0_2369, %int0_2370 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2371 = torch.constant.int 1 loc(#loc1) | |
| %false_2372 = torch.constant.bool false loc(#loc1) | |
| %false_2373 = torch.constant.bool false loc(#loc1) | |
| %true_2374 = torch.constant.bool true loc(#loc1) | |
| %true_2375 = torch.constant.bool true loc(#loc1) | |
| %7978 = torch.aten._convolution %7966, %7973, %5664, %7974, %7975, %7976, %false_2368, %7977, %int1_2371, %false_2372, %false_2373, %true_2374, %true_2375 : !torch.tensor<[2,320,32,32],f32>, !torch.tensor<[640,320,1,1],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_2376 = torch.constant.int 1 loc(#loc1) | |
| %7979 = torch.aten.add.Tensor %7978, %7925, %int1_2376 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7980 = torch.tensor.literal(dense<1.000000e+00> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %7981 = torch.aten.div.Tensor %7979, %7980 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f64> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int2_2377 = torch.constant.int 2 loc(#loc1) | |
| %int32_2378 = torch.constant.int 32 loc(#loc1) | |
| %int20_2379 = torch.constant.int 20 loc(#loc1) | |
| %int1024_2380 = torch.constant.int 1024 loc(#loc1) | |
| %7982 = torch.prim.ListConstruct %int2_2377, %int32_2378, %int20_2379, %int1024_2380 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7983 = torch.aten.view %7981, %7982 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_2381 = torch.constant.int 2 loc(#loc1) | |
| %int3_2382 = torch.constant.int 3 loc(#loc1) | |
| %7984 = torch.prim.ListConstruct %int2_2381, %int3_2382 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_2383 = torch.constant.int 0 loc(#loc1) | |
| %true_2384 = torch.constant.bool true loc(#loc1) | |
| %7985 = torch.aten.var.correction %7983, %7984, %int0_2383, %true_2384 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_2385 = torch.constant.int 2 loc(#loc1) | |
| %int3_2386 = torch.constant.int 3 loc(#loc1) | |
| %7986 = torch.prim.ListConstruct %int2_2385, %int3_2386 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_2387 = torch.constant.bool true loc(#loc1) | |
| %none_2388 = torch.constant.none loc(#loc0) | |
| %7987 = torch.aten.mean.dim %7983, %7986, %true_2387, %none_2388 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %7988 = torch.tensor.literal(dense<9.9999999999999995E-7> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_2389 = torch.constant.int 1 loc(#loc1) | |
| %7989 = torch.aten.add.Tensor %7985, %7988, %int1_2389 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %7990 = torch.aten.rsqrt %7989 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_2390 = torch.constant.int 1 loc(#loc1) | |
| %7991 = torch.aten.sub.Tensor %7983, %7987, %int1_2390 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %7992 = torch.aten.mul.Tensor %7991, %7990 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_2391 = torch.constant.int 2 loc(#loc1) | |
| %int640_2392 = torch.constant.int 640 loc(#loc1) | |
| %int32_2393 = torch.constant.int 32 loc(#loc1) | |
| %int32_2394 = torch.constant.int 32 loc(#loc1) | |
| %7993 = torch.prim.ListConstruct %int2_2391, %int640_2392, %int32_2393, %int32_2394 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7994 = torch.aten.view %7992, %7993 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %7995 = torch.aten.detach %5663 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %7996 = torch.aten.detach %7995 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2395 = torch.constant.int 2 loc(#loc1) | |
| %int40_2396 = torch.constant.int 40 loc(#loc1) | |
| %int16_2397 = torch.constant.int 16 loc(#loc1) | |
| %int32_2398 = torch.constant.int 32 loc(#loc1) | |
| %int32_2399 = torch.constant.int 32 loc(#loc1) | |
| %7997 = torch.prim.ListConstruct %int2_2395, %int40_2396, %int16_2397, %int32_2398, %int32_2399 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %7998 = torch.aten.view %7994, %7997 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %7999 = torch.aten.abs %7998 : !torch.tensor<[2,40,16,32,32],f32> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_2400 = torch.constant.int 2 loc(#loc1) | |
| %true_2401 = torch.constant.bool true loc(#loc1) | |
| %values_2402, %indices_2403 = torch.aten.max.dim %7999, %int2_2400, %true_2401 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,40,1,32,32],f32>, !torch.tensor<[2,40,1,32,32],si64> loc(#loc1) | |
| %int2_2404 = torch.constant.int 2 loc(#loc1) | |
| %int40_2405 = torch.constant.int 40 loc(#loc1) | |
| %int1_2406 = torch.constant.int 1 loc(#loc1) | |
| %int32_2407 = torch.constant.int 32 loc(#loc1) | |
| %int32_2408 = torch.constant.int 32 loc(#loc1) | |
| %8000 = torch.prim.ListConstruct %int2_2404, %int40_2405, %int1_2406, %int32_2407, %int32_2408 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8001 = torch.aten.view %values_2402, %8000 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,1,32,32],f32> loc(#loc1) | |
| %int2_2409 = torch.constant.int 2 loc(#loc1) | |
| %int40_2410 = torch.constant.int 40 loc(#loc1) | |
| %int16_2411 = torch.constant.int 16 loc(#loc1) | |
| %int32_2412 = torch.constant.int 32 loc(#loc1) | |
| %int32_2413 = torch.constant.int 32 loc(#loc1) | |
| %8002 = torch.prim.ListConstruct %int2_2409, %int40_2410, %int16_2411, %int32_2412, %int32_2413 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2414 = torch.constant.bool false loc(#loc1) | |
| %8003 = torch.aten.expand %8001, %8002, %false_2414 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int0_2415 = torch.constant.int 0 loc(#loc1) | |
| %8004 = torch.aten.clone %8003, %int0_2415 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_2416 = torch.constant.int 2 loc(#loc1) | |
| %int640_2417 = torch.constant.int 640 loc(#loc1) | |
| %int32_2418 = torch.constant.int 32 loc(#loc1) | |
| %int32_2419 = torch.constant.int 32 loc(#loc1) | |
| %8005 = torch.prim.ListConstruct %int2_2416, %int640_2417, %int32_2418, %int32_2419 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8006 = torch.aten._unsafe_view %8004, %8005 : !torch.tensor<[2,40,16,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8007 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2420 = torch.constant.int 1 loc(#loc1) | |
| %8008 = torch.aten.sub.Tensor %7996, %8007, %int1_2420 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8009 = torch.aten.lift_fresh_copy %5662 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8010 = torch.aten.pow.Tensor_Tensor %8009, %8008 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8011 = torch.aten.neg %8010 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8012 = torch.aten.neg %8011 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8013 = torch.aten.div.Tensor %8006, %8012 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8014 = torch.aten.detach %5661 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8015 = torch.aten.detach %8014 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8016 = torch.aten.div.Tensor %7994, %8013 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_2421 = torch.constant.int 1 loc(#loc1) | |
| %8017 = torch.aten.add.Tensor %8016, %8015, %int1_2421 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8018 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2422 = torch.constant.int 1 loc(#loc1) | |
| %8019 = torch.aten.sub.Tensor %7996, %8018, %int1_2422 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8020 = torch.aten.lift_fresh_copy %5660 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8021 = torch.aten.pow.Tensor_Tensor %8020, %8019 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8022 = torch.aten.neg %8021 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8023 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2423 = torch.constant.int 1 loc(#loc1) | |
| %8024 = torch.aten.sub.Tensor %7996, %8023, %int1_2423 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8025 = torch.aten.lift_fresh_copy %5659 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8026 = torch.aten.pow.Tensor_Tensor %8025, %8024 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8027 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2424 = torch.constant.int 1 loc(#loc1) | |
| %8028 = torch.aten.sub.Tensor %8026, %8027, %int1_2424 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8029 = torch.aten.gt.Tensor %8017, %8028 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8030 = torch.aten.where.self %8029, %8028, %8017 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8031 = torch.aten.lt.Tensor %8030, %8022 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8032 = torch.aten.where.self %8031, %8022, %8030 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8033 = torch.operator "aten.round"(%8032) : (!torch.tensor<[2,640,32,32],f32>) -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_2425 = torch.constant.int 1 loc(#loc1) | |
| %8034 = torch.aten.sub.Tensor %8033, %8015, %int1_2425 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8035 = torch.aten.mul.Tensor %8034, %8013 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int640_2426 = torch.constant.int 640 loc(#loc1) | |
| %int40_2427 = torch.constant.int 40 loc(#loc1) | |
| %int16_2428 = torch.constant.int 16 loc(#loc1) | |
| %int1_2429 = torch.constant.int 1 loc(#loc1) | |
| %int1_2430 = torch.constant.int 1 loc(#loc1) | |
| %8036 = torch.prim.ListConstruct %int640_2426, %int40_2427, %int16_2428, %int1_2429, %int1_2430 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2431 = torch.constant.bool false loc(#loc1) | |
| %8037 = torch.aten.expand %5658, %8036, %false_2431 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16,1,1],f16> loc(#loc1) | |
| %int0_2432 = torch.constant.int 0 loc(#loc1) | |
| %8038 = torch.aten.clone %8037, %int0_2432 : !torch.tensor<[640,40,16,1,1],f16>, !torch.int -> !torch.tensor<[640,40,16,1,1],f16> loc(#loc1) | |
| %int640_2433 = torch.constant.int 640 loc(#loc1) | |
| %int640_2434 = torch.constant.int 640 loc(#loc1) | |
| %int1_2435 = torch.constant.int 1 loc(#loc1) | |
| %int1_2436 = torch.constant.int 1 loc(#loc1) | |
| %8039 = torch.prim.ListConstruct %int640_2433, %int640_2434, %int1_2435, %int1_2436 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8040 = torch.aten._unsafe_view %8038, %8039 : !torch.tensor<[640,40,16,1,1],f16>, !torch.list<int> -> !torch.tensor<[640,640,1,1],f16> loc(#loc1) | |
| %8041 = torch.aten.mul.Tensor %5657, %8040 : !torch.tensor, !torch.tensor<[640,640,1,1],f16> -> !torch.tensor<[640,640,1,1],f16> loc(#loc1) | |
| %int6_2437 = torch.constant.int 6 loc(#loc1) | |
| %none_2438 = torch.constant.none loc(#loc0) | |
| %none_2439 = torch.constant.none loc(#loc0) | |
| %none_2440 = torch.constant.none loc(#loc0) | |
| %false_2441 = torch.constant.bool false loc(#loc1) | |
| %none_2442 = torch.constant.none loc(#loc0) | |
| %8042 = torch.aten._to_copy %8041, %int6_2437, %none_2438, %none_2439, %none_2440, %false_2441, %none_2442 : !torch.tensor<[640,640,1,1],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640,1,1],f32> loc(#loc1) | |
| %int1_2443 = torch.constant.int 1 loc(#loc1) | |
| %int1_2444 = torch.constant.int 1 loc(#loc1) | |
| %8043 = torch.prim.ListConstruct %int1_2443, %int1_2444 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_2445 = torch.constant.int 0 loc(#loc1) | |
| %int0_2446 = torch.constant.int 0 loc(#loc1) | |
| %8044 = torch.prim.ListConstruct %int0_2445, %int0_2446 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2447 = torch.constant.int 1 loc(#loc1) | |
| %int1_2448 = torch.constant.int 1 loc(#loc1) | |
| %8045 = torch.prim.ListConstruct %int1_2447, %int1_2448 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2449 = torch.constant.bool false loc(#loc1) | |
| %int0_2450 = torch.constant.int 0 loc(#loc1) | |
| %int0_2451 = torch.constant.int 0 loc(#loc1) | |
| %8046 = torch.prim.ListConstruct %int0_2450, %int0_2451 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_2452 = torch.constant.int 1 loc(#loc1) | |
| %false_2453 = torch.constant.bool false loc(#loc1) | |
| %false_2454 = torch.constant.bool false loc(#loc1) | |
| %true_2455 = torch.constant.bool true loc(#loc1) | |
| %true_2456 = torch.constant.bool true loc(#loc1) | |
| %8047 = torch.aten._convolution %8035, %8042, %5656, %8043, %8044, %8045, %false_2449, %8046, %int1_2452, %false_2453, %false_2454, %true_2455, %true_2456 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,640,1,1],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int0_2457 = torch.constant.int 0 loc(#loc1) | |
| %int2_2458 = torch.constant.int 2 loc(#loc1) | |
| %int3_2459 = torch.constant.int 3 loc(#loc1) | |
| %int1_2460 = torch.constant.int 1 loc(#loc1) | |
| %8048 = torch.prim.ListConstruct %int0_2457, %int2_2458, %int3_2459, %int1_2460 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8049 = torch.aten.permute %8047, %8048 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,32,32,640],f32> loc(#loc1) | |
| %int2_2461 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2462 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2463 = torch.constant.int 640 loc(#loc1) | |
| %8050 = torch.prim.ListConstruct %int2_2461, %int1024_2462, %int640_2463 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int655360 = torch.constant.int 655360 loc(#loc1) | |
| %int640_2464 = torch.constant.int 640 loc(#loc1) | |
| %int1_2465 = torch.constant.int 1 loc(#loc1) | |
| %8051 = torch.prim.ListConstruct %int655360, %int640_2464, %int1_2465 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8052 = torch.aten._reshape_alias %8049, %8050, %8051 : !torch.tensor<[2,32,32,640],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_2466 = torch.constant.int 640 loc(#loc1) | |
| %8053 = torch.prim.ListConstruct %int640_2466 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_2467 = torch.constant.none loc(#loc0) | |
| %none_2468 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05_2469 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0_2470, %result1_2471, %result2_2472 = torch.aten.native_layer_norm %8052, %8053, %none_2467, %none_2468, %float1.000000e-05_2469 : !torch.tensor<[2,1024,640],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,1],f32>, !torch.tensor<[2,1024,1],f32> loc(#loc1) | |
| %8054 = torch.aten.detach %5655 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8055 = torch.aten.detach %8054 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2473 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2474 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2475 = torch.constant.int 40 loc(#loc1) | |
| %int16_2476 = torch.constant.int 16 loc(#loc1) | |
| %8056 = torch.prim.ListConstruct %int2_2473, %int1024_2474, %int40_2475, %int16_2476 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8057 = torch.aten.view %result0_2470, %8056 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %8058 = torch.aten.abs %8057 : !torch.tensor<[2,1024,40,16],f32> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int3_2477 = torch.constant.int 3 loc(#loc1) | |
| %true_2478 = torch.constant.bool true loc(#loc1) | |
| %values_2479, %indices_2480 = torch.aten.max.dim %8058, %int3_2477, %true_2478 : !torch.tensor<[2,1024,40,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,1024,40,1],f32>, !torch.tensor<[2,1024,40,1],si64> loc(#loc1) | |
| %int2_2481 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2482 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2483 = torch.constant.int 40 loc(#loc1) | |
| %int1_2484 = torch.constant.int 1 loc(#loc1) | |
| %8059 = torch.prim.ListConstruct %int2_2481, %int1024_2482, %int40_2483, %int1_2484 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8060 = torch.aten.view %values_2479, %8059 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,1],f32> loc(#loc1) | |
| %int2_2485 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2486 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2487 = torch.constant.int 40 loc(#loc1) | |
| %int16_2488 = torch.constant.int 16 loc(#loc1) | |
| %8061 = torch.prim.ListConstruct %int2_2485, %int1024_2486, %int40_2487, %int16_2488 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2489 = torch.constant.bool false loc(#loc1) | |
| %8062 = torch.aten.expand %8060, %8061, %false_2489 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int0_2490 = torch.constant.int 0 loc(#loc1) | |
| %8063 = torch.aten.clone %8062, %int0_2490 : !torch.tensor<[2,1024,40,16],f32>, !torch.int -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int2_2491 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2492 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2493 = torch.constant.int 640 loc(#loc1) | |
| %8064 = torch.prim.ListConstruct %int2_2491, %int1024_2492, %int640_2493 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8065 = torch.aten._unsafe_view %8063, %8064 : !torch.tensor<[2,1024,40,16],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8066 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2494 = torch.constant.int 1 loc(#loc1) | |
| %8067 = torch.aten.sub.Tensor %8055, %8066, %int1_2494 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8068 = torch.aten.lift_fresh_copy %5654 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8069 = torch.aten.pow.Tensor_Tensor %8068, %8067 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8070 = torch.aten.neg %8069 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8071 = torch.aten.neg %8070 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8072 = torch.aten.div.Tensor %8065, %8071 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8073 = torch.aten.detach %5653 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8074 = torch.aten.detach %8073 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8075 = torch.aten.div.Tensor %result0_2470, %8072 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2495 = torch.constant.int 1 loc(#loc1) | |
| %8076 = torch.aten.add.Tensor %8075, %8074, %int1_2495 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8077 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2496 = torch.constant.int 1 loc(#loc1) | |
| %8078 = torch.aten.sub.Tensor %8055, %8077, %int1_2496 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8079 = torch.aten.lift_fresh_copy %5652 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8080 = torch.aten.pow.Tensor_Tensor %8079, %8078 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8081 = torch.aten.neg %8080 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8082 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2497 = torch.constant.int 1 loc(#loc1) | |
| %8083 = torch.aten.sub.Tensor %8055, %8082, %int1_2497 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8084 = torch.aten.lift_fresh_copy %5651 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8085 = torch.aten.pow.Tensor_Tensor %8084, %8083 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8086 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2498 = torch.constant.int 1 loc(#loc1) | |
| %8087 = torch.aten.sub.Tensor %8085, %8086, %int1_2498 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8088 = torch.aten.gt.Tensor %8076, %8087 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8089 = torch.aten.where.self %8088, %8087, %8076 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8090 = torch.aten.lt.Tensor %8089, %8081 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8091 = torch.aten.where.self %8090, %8081, %8089 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8092 = torch.operator "aten.round"(%8091) : (!torch.tensor<[2,1024,640],f32>) -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2499 = torch.constant.int 1 loc(#loc1) | |
| %8093 = torch.aten.sub.Tensor %8092, %8074, %int1_2499 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8094 = torch.aten.mul.Tensor %8093, %8072 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_2500 = torch.constant.int 640 loc(#loc1) | |
| %int40_2501 = torch.constant.int 40 loc(#loc1) | |
| %int16_2502 = torch.constant.int 16 loc(#loc1) | |
| %8095 = torch.prim.ListConstruct %int640_2500, %int40_2501, %int16_2502 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2503 = torch.constant.bool false loc(#loc1) | |
| %8096 = torch.aten.expand %5650, %8095, %false_2503 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int0_2504 = torch.constant.int 0 loc(#loc1) | |
| %8097 = torch.aten.clone %8096, %int0_2504 : !torch.tensor<[640,40,16],f16>, !torch.int -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int640_2505 = torch.constant.int 640 loc(#loc1) | |
| %int640_2506 = torch.constant.int 640 loc(#loc1) | |
| %8098 = torch.prim.ListConstruct %int640_2505, %int640_2506 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8099 = torch.aten._unsafe_view %8097, %8098 : !torch.tensor<[640,40,16],f16>, !torch.list<int> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %8100 = torch.aten.mul.Tensor %5649, %8099 : !torch.tensor, !torch.tensor<[640,640],f16> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %int6_2507 = torch.constant.int 6 loc(#loc1) | |
| %none_2508 = torch.constant.none loc(#loc0) | |
| %none_2509 = torch.constant.none loc(#loc0) | |
| %none_2510 = torch.constant.none loc(#loc0) | |
| %false_2511 = torch.constant.bool false loc(#loc1) | |
| %none_2512 = torch.constant.none loc(#loc0) | |
| %8101 = torch.aten._to_copy %8100, %int6_2507, %none_2508, %none_2509, %none_2510, %false_2511, %none_2512 : !torch.tensor<[640,640],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %8102 = torch.aten.t %8101 : !torch.tensor<[640,640],f32> -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %int2048 = torch.constant.int 2048 loc(#loc1) | |
| %int640_2513 = torch.constant.int 640 loc(#loc1) | |
| %8103 = torch.prim.ListConstruct %int2048, %int640_2513 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8104 = torch.aten.view %8094, %8103 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int1_2514 = torch.constant.int 1 loc(#loc1) | |
| %int1_2515 = torch.constant.int 1 loc(#loc1) | |
| %8105 = torch.aten.addmm %5648, %8104, %8102, %int1_2514, %int1_2515 : !torch.tensor, !torch.tensor<[2048,640],f32>, !torch.tensor<[640,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int2_2516 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2517 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2518 = torch.constant.int 640 loc(#loc1) | |
| %8106 = torch.prim.ListConstruct %int2_2516, %int1024_2517, %int640_2518 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8107 = torch.aten.view %8105, %8106 : !torch.tensor<[2048,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8108 = torch.aten.detach %5647 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8109 = torch.aten.detach %8108 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2519 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2520 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2521 = torch.constant.int 40 loc(#loc1) | |
| %int16_2522 = torch.constant.int 16 loc(#loc1) | |
| %8110 = torch.prim.ListConstruct %int2_2519, %int1024_2520, %int40_2521, %int16_2522 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8111 = torch.aten.view %result0_2470, %8110 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %8112 = torch.aten.abs %8111 : !torch.tensor<[2,1024,40,16],f32> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int3_2523 = torch.constant.int 3 loc(#loc1) | |
| %true_2524 = torch.constant.bool true loc(#loc1) | |
| %values_2525, %indices_2526 = torch.aten.max.dim %8112, %int3_2523, %true_2524 : !torch.tensor<[2,1024,40,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,1024,40,1],f32>, !torch.tensor<[2,1024,40,1],si64> loc(#loc1) | |
| %int2_2527 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2528 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2529 = torch.constant.int 40 loc(#loc1) | |
| %int1_2530 = torch.constant.int 1 loc(#loc1) | |
| %8113 = torch.prim.ListConstruct %int2_2527, %int1024_2528, %int40_2529, %int1_2530 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8114 = torch.aten.view %values_2525, %8113 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,1],f32> loc(#loc1) | |
| %int2_2531 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2532 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2533 = torch.constant.int 40 loc(#loc1) | |
| %int16_2534 = torch.constant.int 16 loc(#loc1) | |
| %8115 = torch.prim.ListConstruct %int2_2531, %int1024_2532, %int40_2533, %int16_2534 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2535 = torch.constant.bool false loc(#loc1) | |
| %8116 = torch.aten.expand %8114, %8115, %false_2535 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int0_2536 = torch.constant.int 0 loc(#loc1) | |
| %8117 = torch.aten.clone %8116, %int0_2536 : !torch.tensor<[2,1024,40,16],f32>, !torch.int -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int2_2537 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2538 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2539 = torch.constant.int 640 loc(#loc1) | |
| %8118 = torch.prim.ListConstruct %int2_2537, %int1024_2538, %int640_2539 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8119 = torch.aten._unsafe_view %8117, %8118 : !torch.tensor<[2,1024,40,16],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8120 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2540 = torch.constant.int 1 loc(#loc1) | |
| %8121 = torch.aten.sub.Tensor %8109, %8120, %int1_2540 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8122 = torch.aten.lift_fresh_copy %5646 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8123 = torch.aten.pow.Tensor_Tensor %8122, %8121 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8124 = torch.aten.neg %8123 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8125 = torch.aten.neg %8124 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8126 = torch.aten.div.Tensor %8119, %8125 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8127 = torch.aten.detach %5645 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8128 = torch.aten.detach %8127 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8129 = torch.aten.div.Tensor %result0_2470, %8126 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2541 = torch.constant.int 1 loc(#loc1) | |
| %8130 = torch.aten.add.Tensor %8129, %8128, %int1_2541 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8131 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2542 = torch.constant.int 1 loc(#loc1) | |
| %8132 = torch.aten.sub.Tensor %8109, %8131, %int1_2542 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8133 = torch.aten.lift_fresh_copy %5644 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8134 = torch.aten.pow.Tensor_Tensor %8133, %8132 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8135 = torch.aten.neg %8134 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8136 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2543 = torch.constant.int 1 loc(#loc1) | |
| %8137 = torch.aten.sub.Tensor %8109, %8136, %int1_2543 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8138 = torch.aten.lift_fresh_copy %5643 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8139 = torch.aten.pow.Tensor_Tensor %8138, %8137 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8140 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2544 = torch.constant.int 1 loc(#loc1) | |
| %8141 = torch.aten.sub.Tensor %8139, %8140, %int1_2544 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8142 = torch.aten.gt.Tensor %8130, %8141 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8143 = torch.aten.where.self %8142, %8141, %8130 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8144 = torch.aten.lt.Tensor %8143, %8135 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8145 = torch.aten.where.self %8144, %8135, %8143 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8146 = torch.operator "aten.round"(%8145) : (!torch.tensor<[2,1024,640],f32>) -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2545 = torch.constant.int 1 loc(#loc1) | |
| %8147 = torch.aten.sub.Tensor %8146, %8128, %int1_2545 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8148 = torch.aten.mul.Tensor %8147, %8126 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_2546 = torch.constant.int 640 loc(#loc1) | |
| %int40_2547 = torch.constant.int 40 loc(#loc1) | |
| %int16_2548 = torch.constant.int 16 loc(#loc1) | |
| %8149 = torch.prim.ListConstruct %int640_2546, %int40_2547, %int16_2548 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2549 = torch.constant.bool false loc(#loc1) | |
| %8150 = torch.aten.expand %5642, %8149, %false_2549 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int0_2550 = torch.constant.int 0 loc(#loc1) | |
| %8151 = torch.aten.clone %8150, %int0_2550 : !torch.tensor<[640,40,16],f16>, !torch.int -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int640_2551 = torch.constant.int 640 loc(#loc1) | |
| %int640_2552 = torch.constant.int 640 loc(#loc1) | |
| %8152 = torch.prim.ListConstruct %int640_2551, %int640_2552 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8153 = torch.aten._unsafe_view %8151, %8152 : !torch.tensor<[640,40,16],f16>, !torch.list<int> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %8154 = torch.aten.mul.Tensor %5641, %8153 : !torch.tensor, !torch.tensor<[640,640],f16> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %int6_2553 = torch.constant.int 6 loc(#loc1) | |
| %none_2554 = torch.constant.none loc(#loc0) | |
| %none_2555 = torch.constant.none loc(#loc0) | |
| %none_2556 = torch.constant.none loc(#loc0) | |
| %false_2557 = torch.constant.bool false loc(#loc1) | |
| %none_2558 = torch.constant.none loc(#loc0) | |
| %8155 = torch.aten._to_copy %8154, %int6_2553, %none_2554, %none_2555, %none_2556, %false_2557, %none_2558 : !torch.tensor<[640,640],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %8156 = torch.aten.t %8155 : !torch.tensor<[640,640],f32> -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %int2048_2559 = torch.constant.int 2048 loc(#loc1) | |
| %int640_2560 = torch.constant.int 640 loc(#loc1) | |
| %8157 = torch.prim.ListConstruct %int2048_2559, %int640_2560 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8158 = torch.aten.view %8148, %8157 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int1_2561 = torch.constant.int 1 loc(#loc1) | |
| %int1_2562 = torch.constant.int 1 loc(#loc1) | |
| %8159 = torch.aten.addmm %5640, %8158, %8156, %int1_2561, %int1_2562 : !torch.tensor, !torch.tensor<[2048,640],f32>, !torch.tensor<[640,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int2_2563 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2564 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2565 = torch.constant.int 640 loc(#loc1) | |
| %8160 = torch.prim.ListConstruct %int2_2563, %int1024_2564, %int640_2565 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8161 = torch.aten.view %8159, %8160 : !torch.tensor<[2048,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8162 = torch.aten.detach %5639 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8163 = torch.aten.detach %8162 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2566 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2567 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2568 = torch.constant.int 40 loc(#loc1) | |
| %int16_2569 = torch.constant.int 16 loc(#loc1) | |
| %8164 = torch.prim.ListConstruct %int2_2566, %int1024_2567, %int40_2568, %int16_2569 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8165 = torch.aten.view %result0_2470, %8164 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %8166 = torch.aten.abs %8165 : !torch.tensor<[2,1024,40,16],f32> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int3_2570 = torch.constant.int 3 loc(#loc1) | |
| %true_2571 = torch.constant.bool true loc(#loc1) | |
| %values_2572, %indices_2573 = torch.aten.max.dim %8166, %int3_2570, %true_2571 : !torch.tensor<[2,1024,40,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,1024,40,1],f32>, !torch.tensor<[2,1024,40,1],si64> loc(#loc1) | |
| %int2_2574 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2575 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2576 = torch.constant.int 40 loc(#loc1) | |
| %int1_2577 = torch.constant.int 1 loc(#loc1) | |
| %8167 = torch.prim.ListConstruct %int2_2574, %int1024_2575, %int40_2576, %int1_2577 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8168 = torch.aten.view %values_2572, %8167 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,1],f32> loc(#loc1) | |
| %int2_2578 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2579 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2580 = torch.constant.int 40 loc(#loc1) | |
| %int16_2581 = torch.constant.int 16 loc(#loc1) | |
| %8169 = torch.prim.ListConstruct %int2_2578, %int1024_2579, %int40_2580, %int16_2581 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2582 = torch.constant.bool false loc(#loc1) | |
| %8170 = torch.aten.expand %8168, %8169, %false_2582 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int0_2583 = torch.constant.int 0 loc(#loc1) | |
| %8171 = torch.aten.clone %8170, %int0_2583 : !torch.tensor<[2,1024,40,16],f32>, !torch.int -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int2_2584 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2585 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2586 = torch.constant.int 640 loc(#loc1) | |
| %8172 = torch.prim.ListConstruct %int2_2584, %int1024_2585, %int640_2586 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8173 = torch.aten._unsafe_view %8171, %8172 : !torch.tensor<[2,1024,40,16],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8174 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2587 = torch.constant.int 1 loc(#loc1) | |
| %8175 = torch.aten.sub.Tensor %8163, %8174, %int1_2587 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8176 = torch.aten.lift_fresh_copy %5638 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8177 = torch.aten.pow.Tensor_Tensor %8176, %8175 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8178 = torch.aten.neg %8177 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8179 = torch.aten.neg %8178 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8180 = torch.aten.div.Tensor %8173, %8179 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8181 = torch.aten.detach %5637 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8182 = torch.aten.detach %8181 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8183 = torch.aten.div.Tensor %result0_2470, %8180 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2588 = torch.constant.int 1 loc(#loc1) | |
| %8184 = torch.aten.add.Tensor %8183, %8182, %int1_2588 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8185 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2589 = torch.constant.int 1 loc(#loc1) | |
| %8186 = torch.aten.sub.Tensor %8163, %8185, %int1_2589 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8187 = torch.aten.lift_fresh_copy %5636 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8188 = torch.aten.pow.Tensor_Tensor %8187, %8186 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8189 = torch.aten.neg %8188 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8190 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2590 = torch.constant.int 1 loc(#loc1) | |
| %8191 = torch.aten.sub.Tensor %8163, %8190, %int1_2590 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8192 = torch.aten.lift_fresh_copy %5635 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8193 = torch.aten.pow.Tensor_Tensor %8192, %8191 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8194 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2591 = torch.constant.int 1 loc(#loc1) | |
| %8195 = torch.aten.sub.Tensor %8193, %8194, %int1_2591 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8196 = torch.aten.gt.Tensor %8184, %8195 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8197 = torch.aten.where.self %8196, %8195, %8184 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8198 = torch.aten.lt.Tensor %8197, %8189 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8199 = torch.aten.where.self %8198, %8189, %8197 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8200 = torch.operator "aten.round"(%8199) : (!torch.tensor<[2,1024,640],f32>) -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2592 = torch.constant.int 1 loc(#loc1) | |
| %8201 = torch.aten.sub.Tensor %8200, %8182, %int1_2592 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8202 = torch.aten.mul.Tensor %8201, %8180 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_2593 = torch.constant.int 640 loc(#loc1) | |
| %int40_2594 = torch.constant.int 40 loc(#loc1) | |
| %int16_2595 = torch.constant.int 16 loc(#loc1) | |
| %8203 = torch.prim.ListConstruct %int640_2593, %int40_2594, %int16_2595 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2596 = torch.constant.bool false loc(#loc1) | |
| %8204 = torch.aten.expand %5634, %8203, %false_2596 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int0_2597 = torch.constant.int 0 loc(#loc1) | |
| %8205 = torch.aten.clone %8204, %int0_2597 : !torch.tensor<[640,40,16],f16>, !torch.int -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int640_2598 = torch.constant.int 640 loc(#loc1) | |
| %int640_2599 = torch.constant.int 640 loc(#loc1) | |
| %8206 = torch.prim.ListConstruct %int640_2598, %int640_2599 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8207 = torch.aten._unsafe_view %8205, %8206 : !torch.tensor<[640,40,16],f16>, !torch.list<int> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %8208 = torch.aten.mul.Tensor %5633, %8207 : !torch.tensor, !torch.tensor<[640,640],f16> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %int6_2600 = torch.constant.int 6 loc(#loc1) | |
| %none_2601 = torch.constant.none loc(#loc0) | |
| %none_2602 = torch.constant.none loc(#loc0) | |
| %none_2603 = torch.constant.none loc(#loc0) | |
| %false_2604 = torch.constant.bool false loc(#loc1) | |
| %none_2605 = torch.constant.none loc(#loc0) | |
| %8209 = torch.aten._to_copy %8208, %int6_2600, %none_2601, %none_2602, %none_2603, %false_2604, %none_2605 : !torch.tensor<[640,640],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %8210 = torch.aten.t %8209 : !torch.tensor<[640,640],f32> -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %int2048_2606 = torch.constant.int 2048 loc(#loc1) | |
| %int640_2607 = torch.constant.int 640 loc(#loc1) | |
| %8211 = torch.prim.ListConstruct %int2048_2606, %int640_2607 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8212 = torch.aten.view %8202, %8211 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int1_2608 = torch.constant.int 1 loc(#loc1) | |
| %int1_2609 = torch.constant.int 1 loc(#loc1) | |
| %8213 = torch.aten.addmm %5632, %8212, %8210, %int1_2608, %int1_2609 : !torch.tensor, !torch.tensor<[2048,640],f32>, !torch.tensor<[640,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int2_2610 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2611 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2612 = torch.constant.int 640 loc(#loc1) | |
| %8214 = torch.prim.ListConstruct %int2_2610, %int1024_2611, %int640_2612 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8215 = torch.aten.view %8213, %8214 : !torch.tensor<[2048,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int2_2613 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2614 = torch.constant.int 1024 loc(#loc1) | |
| %int8_2615 = torch.constant.int 8 loc(#loc1) | |
| %int80_2616 = torch.constant.int 80 loc(#loc1) | |
| %8216 = torch.prim.ListConstruct %int2_2613, %int1024_2614, %int8_2615, %int80_2616 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int655360_2617 = torch.constant.int 655360 loc(#loc1) | |
| %int640_2618 = torch.constant.int 640 loc(#loc1) | |
| %int80_2619 = torch.constant.int 80 loc(#loc1) | |
| %int1_2620 = torch.constant.int 1 loc(#loc1) | |
| %8217 = torch.prim.ListConstruct %int655360_2617, %int640_2618, %int80_2619, %int1_2620 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8218 = torch.aten._reshape_alias %8107, %8216, %8217 : !torch.tensor<[2,1024,640],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,1024,8,80],f32> loc(#loc1) | |
| %int0_2621 = torch.constant.int 0 loc(#loc1) | |
| %int2_2622 = torch.constant.int 2 loc(#loc1) | |
| %int1_2623 = torch.constant.int 1 loc(#loc1) | |
| %int3_2624 = torch.constant.int 3 loc(#loc1) | |
| %8219 = torch.prim.ListConstruct %int0_2621, %int2_2622, %int1_2623, %int3_2624 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8220 = torch.aten.permute %8218, %8219 : !torch.tensor<[2,1024,8,80],f32>, !torch.list<int> -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int0_2625 = torch.constant.int 0 loc(#loc1) | |
| %8221 = torch.aten.clone %8220, %int0_2625 : !torch.tensor<[2,8,1024,80],f32>, !torch.int -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int16_2626 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2627 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2628 = torch.constant.int 80 loc(#loc1) | |
| %8222 = torch.prim.ListConstruct %int16_2626, %int1024_2627, %int80_2628 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8223 = torch.aten._unsafe_view %8221, %8222 : !torch.tensor<[2,8,1024,80],f32>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int2_2629 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2630 = torch.constant.int 1024 loc(#loc1) | |
| %int8_2631 = torch.constant.int 8 loc(#loc1) | |
| %int80_2632 = torch.constant.int 80 loc(#loc1) | |
| %8224 = torch.prim.ListConstruct %int2_2629, %int1024_2630, %int8_2631, %int80_2632 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int655360_2633 = torch.constant.int 655360 loc(#loc1) | |
| %int640_2634 = torch.constant.int 640 loc(#loc1) | |
| %int80_2635 = torch.constant.int 80 loc(#loc1) | |
| %int1_2636 = torch.constant.int 1 loc(#loc1) | |
| %8225 = torch.prim.ListConstruct %int655360_2633, %int640_2634, %int80_2635, %int1_2636 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8226 = torch.aten._reshape_alias %8161, %8224, %8225 : !torch.tensor<[2,1024,640],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,1024,8,80],f32> loc(#loc1) | |
| %int0_2637 = torch.constant.int 0 loc(#loc1) | |
| %int2_2638 = torch.constant.int 2 loc(#loc1) | |
| %int1_2639 = torch.constant.int 1 loc(#loc1) | |
| %int3_2640 = torch.constant.int 3 loc(#loc1) | |
| %8227 = torch.prim.ListConstruct %int0_2637, %int2_2638, %int1_2639, %int3_2640 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8228 = torch.aten.permute %8226, %8227 : !torch.tensor<[2,1024,8,80],f32>, !torch.list<int> -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int0_2641 = torch.constant.int 0 loc(#loc1) | |
| %8229 = torch.aten.clone %8228, %int0_2641 : !torch.tensor<[2,8,1024,80],f32>, !torch.int -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int16_2642 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2643 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2644 = torch.constant.int 80 loc(#loc1) | |
| %8230 = torch.prim.ListConstruct %int16_2642, %int1024_2643, %int80_2644 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8231 = torch.aten._unsafe_view %8229, %8230 : !torch.tensor<[2,8,1024,80],f32>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int2_2645 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2646 = torch.constant.int 1024 loc(#loc1) | |
| %int8_2647 = torch.constant.int 8 loc(#loc1) | |
| %int80_2648 = torch.constant.int 80 loc(#loc1) | |
| %8232 = torch.prim.ListConstruct %int2_2645, %int1024_2646, %int8_2647, %int80_2648 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int655360_2649 = torch.constant.int 655360 loc(#loc1) | |
| %int640_2650 = torch.constant.int 640 loc(#loc1) | |
| %int80_2651 = torch.constant.int 80 loc(#loc1) | |
| %int1_2652 = torch.constant.int 1 loc(#loc1) | |
| %8233 = torch.prim.ListConstruct %int655360_2649, %int640_2650, %int80_2651, %int1_2652 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8234 = torch.aten._reshape_alias %8215, %8232, %8233 : !torch.tensor<[2,1024,640],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,1024,8,80],f32> loc(#loc1) | |
| %int0_2653 = torch.constant.int 0 loc(#loc1) | |
| %int2_2654 = torch.constant.int 2 loc(#loc1) | |
| %int1_2655 = torch.constant.int 1 loc(#loc1) | |
| %int3_2656 = torch.constant.int 3 loc(#loc1) | |
| %8235 = torch.prim.ListConstruct %int0_2653, %int2_2654, %int1_2655, %int3_2656 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8236 = torch.aten.permute %8234, %8235 : !torch.tensor<[2,1024,8,80],f32>, !torch.list<int> -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int0_2657 = torch.constant.int 0 loc(#loc1) | |
| %8237 = torch.aten.clone %8236, %int0_2657 : !torch.tensor<[2,8,1024,80],f32>, !torch.int -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int16_2658 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2659 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2660 = torch.constant.int 80 loc(#loc1) | |
| %8238 = torch.prim.ListConstruct %int16_2658, %int1024_2659, %int80_2660 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8239 = torch.aten._unsafe_view %8237, %8238 : !torch.tensor<[2,8,1024,80],f32>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int-1_2661 = torch.constant.int -1 loc(#loc1) | |
| %int-2_2662 = torch.constant.int -2 loc(#loc1) | |
| %8240 = torch.aten.transpose.int %8231, %int-1_2661, %int-2_2662 : !torch.tensor<[16,1024,80],f32>, !torch.int, !torch.int -> !torch.tensor<[16,80,1024],f32> loc(#loc1) | |
| %int16_2663 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2664 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2665 = torch.constant.int 80 loc(#loc1) | |
| %8241 = torch.prim.ListConstruct %int16_2663, %int1024_2664, %int80_2665 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2666 = torch.constant.bool false loc(#loc1) | |
| %8242 = torch.aten.expand %8223, %8241, %false_2666 : !torch.tensor<[16,1024,80],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int16_2667 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2668 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2669 = torch.constant.int 80 loc(#loc1) | |
| %8243 = torch.prim.ListConstruct %int16_2667, %int1024_2668, %int80_2669 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int81920 = torch.constant.int 81920 loc(#loc1) | |
| %int80_2670 = torch.constant.int 80 loc(#loc1) | |
| %int1_2671 = torch.constant.int 1 loc(#loc1) | |
| %8244 = torch.prim.ListConstruct %int81920, %int80_2670, %int1_2671 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8245 = torch.aten._reshape_alias %8242, %8243, %8244 : !torch.tensor<[16,1024,80],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int16_2672 = torch.constant.int 16 loc(#loc1) | |
| %int80_2673 = torch.constant.int 80 loc(#loc1) | |
| %int1024_2674 = torch.constant.int 1024 loc(#loc1) | |
| %8246 = torch.prim.ListConstruct %int16_2672, %int80_2673, %int1024_2674 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2675 = torch.constant.bool false loc(#loc1) | |
| %8247 = torch.aten.expand %8240, %8246, %false_2675 : !torch.tensor<[16,80,1024],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,80,1024],f32> loc(#loc1) | |
| %int16_2676 = torch.constant.int 16 loc(#loc1) | |
| %int80_2677 = torch.constant.int 80 loc(#loc1) | |
| %int1024_2678 = torch.constant.int 1024 loc(#loc1) | |
| %8248 = torch.prim.ListConstruct %int16_2676, %int80_2677, %int1024_2678 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int81920_2679 = torch.constant.int 81920 loc(#loc1) | |
| %int1_2680 = torch.constant.int 1 loc(#loc1) | |
| %int80_2681 = torch.constant.int 80 loc(#loc1) | |
| %8249 = torch.prim.ListConstruct %int81920_2679, %int1_2680, %int80_2681 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8250 = torch.aten._reshape_alias %8247, %8248, %8249 : !torch.tensor<[16,80,1024],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,80,1024],f32> loc(#loc1) | |
| %8251 = torch.aten.bmm %8245, %8250 : !torch.tensor<[16,1024,80],f32>, !torch.tensor<[16,80,1024],f32> -> !torch.tensor<[16,1024,1024],f32> loc(#loc1) | |
| %int16_2682 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2683 = torch.constant.int 1024 loc(#loc1) | |
| %int1024_2684 = torch.constant.int 1024 loc(#loc1) | |
| %8252 = torch.prim.ListConstruct %int16_2682, %int1024_2683, %int1024_2684 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8253 = torch.aten._unsafe_view %8251, %8252 : !torch.tensor<[16,1024,1024],f32>, !torch.list<int> -> !torch.tensor<[16,1024,1024],f32> loc(#loc1) | |
| %8254 = torch.tensor.literal(dense<0.11180339887498948> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %8255 = torch.aten.mul.Tensor %8253, %8254 : !torch.tensor<[16,1024,1024],f32>, !torch.tensor<[],f64> -> !torch.tensor<[16,1024,1024],f32> loc(#loc1) | |
| %int-1_2685 = torch.constant.int -1 loc(#loc1) | |
| %false_2686 = torch.constant.bool false loc(#loc1) | |
| %8256 = torch.aten._softmax %8255, %int-1_2685, %false_2686 : !torch.tensor<[16,1024,1024],f32>, !torch.int, !torch.bool -> !torch.tensor<[16,1024,1024],f32> loc(#loc1) | |
| %int16_2687 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2688 = torch.constant.int 1024 loc(#loc1) | |
| %int1024_2689 = torch.constant.int 1024 loc(#loc1) | |
| %8257 = torch.prim.ListConstruct %int16_2687, %int1024_2688, %int1024_2689 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2690 = torch.constant.bool false loc(#loc1) | |
| %8258 = torch.aten.expand %8256, %8257, %false_2690 : !torch.tensor<[16,1024,1024],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,1024,1024],f32> loc(#loc1) | |
| %int16_2691 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2692 = torch.constant.int 1024 loc(#loc1) | |
| %int1024_2693 = torch.constant.int 1024 loc(#loc1) | |
| %8259 = torch.prim.ListConstruct %int16_2691, %int1024_2692, %int1024_2693 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1048576 = torch.constant.int 1048576 loc(#loc1) | |
| %int1024_2694 = torch.constant.int 1024 loc(#loc1) | |
| %int1_2695 = torch.constant.int 1 loc(#loc1) | |
| %8260 = torch.prim.ListConstruct %int1048576, %int1024_2694, %int1_2695 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8261 = torch.aten._reshape_alias %8258, %8259, %8260 : !torch.tensor<[16,1024,1024],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,1024,1024],f32> loc(#loc1) | |
| %int16_2696 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2697 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2698 = torch.constant.int 80 loc(#loc1) | |
| %8262 = torch.prim.ListConstruct %int16_2696, %int1024_2697, %int80_2698 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2699 = torch.constant.bool false loc(#loc1) | |
| %8263 = torch.aten.expand %8239, %8262, %false_2699 : !torch.tensor<[16,1024,80],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int16_2700 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2701 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2702 = torch.constant.int 80 loc(#loc1) | |
| %8264 = torch.prim.ListConstruct %int16_2700, %int1024_2701, %int80_2702 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int81920_2703 = torch.constant.int 81920 loc(#loc1) | |
| %int80_2704 = torch.constant.int 80 loc(#loc1) | |
| %int1_2705 = torch.constant.int 1 loc(#loc1) | |
| %8265 = torch.prim.ListConstruct %int81920_2703, %int80_2704, %int1_2705 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8266 = torch.aten._reshape_alias %8263, %8264, %8265 : !torch.tensor<[16,1024,80],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %8267 = torch.aten.bmm %8261, %8266 : !torch.tensor<[16,1024,1024],f32>, !torch.tensor<[16,1024,80],f32> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int16_2706 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2707 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2708 = torch.constant.int 80 loc(#loc1) | |
| %8268 = torch.prim.ListConstruct %int16_2706, %int1024_2707, %int80_2708 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8269 = torch.aten._unsafe_view %8267, %8268 : !torch.tensor<[16,1024,80],f32>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int2_2709 = torch.constant.int 2 loc(#loc1) | |
| %int8_2710 = torch.constant.int 8 loc(#loc1) | |
| %int1024_2711 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2712 = torch.constant.int 80 loc(#loc1) | |
| %8270 = torch.prim.ListConstruct %int2_2709, %int8_2710, %int1024_2711, %int80_2712 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int655360_2713 = torch.constant.int 655360 loc(#loc1) | |
| %int81920_2714 = torch.constant.int 81920 loc(#loc1) | |
| %int80_2715 = torch.constant.int 80 loc(#loc1) | |
| %int1_2716 = torch.constant.int 1 loc(#loc1) | |
| %8271 = torch.prim.ListConstruct %int655360_2713, %int81920_2714, %int80_2715, %int1_2716 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8272 = torch.aten._reshape_alias %8269, %8270, %8271 : !torch.tensor<[16,1024,80],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int0_2717 = torch.constant.int 0 loc(#loc1) | |
| %int2_2718 = torch.constant.int 2 loc(#loc1) | |
| %int1_2719 = torch.constant.int 1 loc(#loc1) | |
| %int3_2720 = torch.constant.int 3 loc(#loc1) | |
| %8273 = torch.prim.ListConstruct %int0_2717, %int2_2718, %int1_2719, %int3_2720 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8274 = torch.aten.permute %8272, %8273 : !torch.tensor<[2,8,1024,80],f32>, !torch.list<int> -> !torch.tensor<[2,1024,8,80],f32> loc(#loc1) | |
| %int0_2721 = torch.constant.int 0 loc(#loc1) | |
| %8275 = torch.aten.clone %8274, %int0_2721 : !torch.tensor<[2,1024,8,80],f32>, !torch.int -> !torch.tensor<[2,1024,8,80],f32> loc(#loc1) | |
| %int2_2722 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2723 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2724 = torch.constant.int 640 loc(#loc1) | |
| %8276 = torch.prim.ListConstruct %int2_2722, %int1024_2723, %int640_2724 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8277 = torch.aten._unsafe_view %8275, %8276 : !torch.tensor<[2,1024,8,80],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8278 = torch.aten.detach %5631 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8279 = torch.aten.detach %8278 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2725 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2726 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2727 = torch.constant.int 40 loc(#loc1) | |
| %int16_2728 = torch.constant.int 16 loc(#loc1) | |
| %8280 = torch.prim.ListConstruct %int2_2725, %int1024_2726, %int40_2727, %int16_2728 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8281 = torch.aten.view %8277, %8280 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %8282 = torch.aten.abs %8281 : !torch.tensor<[2,1024,40,16],f32> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int3_2729 = torch.constant.int 3 loc(#loc1) | |
| %true_2730 = torch.constant.bool true loc(#loc1) | |
| %values_2731, %indices_2732 = torch.aten.max.dim %8282, %int3_2729, %true_2730 : !torch.tensor<[2,1024,40,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,1024,40,1],f32>, !torch.tensor<[2,1024,40,1],si64> loc(#loc1) | |
| %int2_2733 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2734 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2735 = torch.constant.int 40 loc(#loc1) | |
| %int1_2736 = torch.constant.int 1 loc(#loc1) | |
| %8283 = torch.prim.ListConstruct %int2_2733, %int1024_2734, %int40_2735, %int1_2736 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8284 = torch.aten.view %values_2731, %8283 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,1],f32> loc(#loc1) | |
| %int2_2737 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2738 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2739 = torch.constant.int 40 loc(#loc1) | |
| %int16_2740 = torch.constant.int 16 loc(#loc1) | |
| %8285 = torch.prim.ListConstruct %int2_2737, %int1024_2738, %int40_2739, %int16_2740 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2741 = torch.constant.bool false loc(#loc1) | |
| %8286 = torch.aten.expand %8284, %8285, %false_2741 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int0_2742 = torch.constant.int 0 loc(#loc1) | |
| %8287 = torch.aten.clone %8286, %int0_2742 : !torch.tensor<[2,1024,40,16],f32>, !torch.int -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int2_2743 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2744 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2745 = torch.constant.int 640 loc(#loc1) | |
| %8288 = torch.prim.ListConstruct %int2_2743, %int1024_2744, %int640_2745 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8289 = torch.aten._unsafe_view %8287, %8288 : !torch.tensor<[2,1024,40,16],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8290 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2746 = torch.constant.int 1 loc(#loc1) | |
| %8291 = torch.aten.sub.Tensor %8279, %8290, %int1_2746 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8292 = torch.aten.lift_fresh_copy %5630 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8293 = torch.aten.pow.Tensor_Tensor %8292, %8291 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8294 = torch.aten.neg %8293 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8295 = torch.aten.neg %8294 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8296 = torch.aten.div.Tensor %8289, %8295 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8297 = torch.aten.detach %5629 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8298 = torch.aten.detach %8297 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8299 = torch.aten.div.Tensor %8277, %8296 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2747 = torch.constant.int 1 loc(#loc1) | |
| %8300 = torch.aten.add.Tensor %8299, %8298, %int1_2747 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8301 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2748 = torch.constant.int 1 loc(#loc1) | |
| %8302 = torch.aten.sub.Tensor %8279, %8301, %int1_2748 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8303 = torch.aten.lift_fresh_copy %5628 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8304 = torch.aten.pow.Tensor_Tensor %8303, %8302 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8305 = torch.aten.neg %8304 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8306 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2749 = torch.constant.int 1 loc(#loc1) | |
| %8307 = torch.aten.sub.Tensor %8279, %8306, %int1_2749 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8308 = torch.aten.lift_fresh_copy %5627 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8309 = torch.aten.pow.Tensor_Tensor %8308, %8307 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8310 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2750 = torch.constant.int 1 loc(#loc1) | |
| %8311 = torch.aten.sub.Tensor %8309, %8310, %int1_2750 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8312 = torch.aten.gt.Tensor %8300, %8311 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8313 = torch.aten.where.self %8312, %8311, %8300 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8314 = torch.aten.lt.Tensor %8313, %8305 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8315 = torch.aten.where.self %8314, %8305, %8313 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8316 = torch.operator "aten.round"(%8315) : (!torch.tensor<[2,1024,640],f32>) -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2751 = torch.constant.int 1 loc(#loc1) | |
| %8317 = torch.aten.sub.Tensor %8316, %8298, %int1_2751 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8318 = torch.aten.mul.Tensor %8317, %8296 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_2752 = torch.constant.int 640 loc(#loc1) | |
| %int40_2753 = torch.constant.int 40 loc(#loc1) | |
| %int16_2754 = torch.constant.int 16 loc(#loc1) | |
| %8319 = torch.prim.ListConstruct %int640_2752, %int40_2753, %int16_2754 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2755 = torch.constant.bool false loc(#loc1) | |
| %8320 = torch.aten.expand %5626, %8319, %false_2755 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int0_2756 = torch.constant.int 0 loc(#loc1) | |
| %8321 = torch.aten.clone %8320, %int0_2756 : !torch.tensor<[640,40,16],f16>, !torch.int -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int640_2757 = torch.constant.int 640 loc(#loc1) | |
| %int640_2758 = torch.constant.int 640 loc(#loc1) | |
| %8322 = torch.prim.ListConstruct %int640_2757, %int640_2758 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8323 = torch.aten._unsafe_view %8321, %8322 : !torch.tensor<[640,40,16],f16>, !torch.list<int> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %8324 = torch.aten.mul.Tensor %5625, %8323 : !torch.tensor, !torch.tensor<[640,640],f16> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %int6_2759 = torch.constant.int 6 loc(#loc1) | |
| %none_2760 = torch.constant.none loc(#loc0) | |
| %none_2761 = torch.constant.none loc(#loc0) | |
| %none_2762 = torch.constant.none loc(#loc0) | |
| %false_2763 = torch.constant.bool false loc(#loc1) | |
| %none_2764 = torch.constant.none loc(#loc0) | |
| %8325 = torch.aten._to_copy %8324, %int6_2759, %none_2760, %none_2761, %none_2762, %false_2763, %none_2764 : !torch.tensor<[640,640],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %8326 = torch.aten.t %8325 : !torch.tensor<[640,640],f32> -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %int2048_2765 = torch.constant.int 2048 loc(#loc1) | |
| %int640_2766 = torch.constant.int 640 loc(#loc1) | |
| %8327 = torch.prim.ListConstruct %int2048_2765, %int640_2766 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8328 = torch.aten.view %8318, %8327 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int1_2767 = torch.constant.int 1 loc(#loc1) | |
| %int1_2768 = torch.constant.int 1 loc(#loc1) | |
| %8329 = torch.aten.addmm %5624, %8328, %8326, %int1_2767, %int1_2768 : !torch.tensor, !torch.tensor<[2048,640],f32>, !torch.tensor<[640,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int2_2769 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2770 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2771 = torch.constant.int 640 loc(#loc1) | |
| %8330 = torch.prim.ListConstruct %int2_2769, %int1024_2770, %int640_2771 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8331 = torch.aten.view %8329, %8330 : !torch.tensor<[2048,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2772 = torch.constant.int 1 loc(#loc1) | |
| %8332 = torch.aten.add.Tensor %8331, %8052, %int1_2772 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_2773 = torch.constant.int 640 loc(#loc1) | |
| %8333 = torch.prim.ListConstruct %int640_2773 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_2774 = torch.constant.none loc(#loc0) | |
| %none_2775 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05_2776 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0_2777, %result1_2778, %result2_2779 = torch.aten.native_layer_norm %8332, %8333, %none_2774, %none_2775, %float1.000000e-05_2776 : !torch.tensor<[2,1024,640],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,1],f32>, !torch.tensor<[2,1024,1],f32> loc(#loc1) | |
| %8334 = torch.aten.detach %5623 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8335 = torch.aten.detach %8334 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2780 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2781 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2782 = torch.constant.int 40 loc(#loc1) | |
| %int16_2783 = torch.constant.int 16 loc(#loc1) | |
| %8336 = torch.prim.ListConstruct %int2_2780, %int1024_2781, %int40_2782, %int16_2783 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8337 = torch.aten.view %result0_2777, %8336 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %8338 = torch.aten.abs %8337 : !torch.tensor<[2,1024,40,16],f32> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int3_2784 = torch.constant.int 3 loc(#loc1) | |
| %true_2785 = torch.constant.bool true loc(#loc1) | |
| %values_2786, %indices_2787 = torch.aten.max.dim %8338, %int3_2784, %true_2785 : !torch.tensor<[2,1024,40,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,1024,40,1],f32>, !torch.tensor<[2,1024,40,1],si64> loc(#loc1) | |
| %int2_2788 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2789 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2790 = torch.constant.int 40 loc(#loc1) | |
| %int1_2791 = torch.constant.int 1 loc(#loc1) | |
| %8339 = torch.prim.ListConstruct %int2_2788, %int1024_2789, %int40_2790, %int1_2791 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8340 = torch.aten.view %values_2786, %8339 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,1],f32> loc(#loc1) | |
| %int2_2792 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2793 = torch.constant.int 1024 loc(#loc1) | |
| %int40_2794 = torch.constant.int 40 loc(#loc1) | |
| %int16_2795 = torch.constant.int 16 loc(#loc1) | |
| %8341 = torch.prim.ListConstruct %int2_2792, %int1024_2793, %int40_2794, %int16_2795 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2796 = torch.constant.bool false loc(#loc1) | |
| %8342 = torch.aten.expand %8340, %8341, %false_2796 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int0_2797 = torch.constant.int 0 loc(#loc1) | |
| %8343 = torch.aten.clone %8342, %int0_2797 : !torch.tensor<[2,1024,40,16],f32>, !torch.int -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int2_2798 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2799 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2800 = torch.constant.int 640 loc(#loc1) | |
| %8344 = torch.prim.ListConstruct %int2_2798, %int1024_2799, %int640_2800 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8345 = torch.aten._unsafe_view %8343, %8344 : !torch.tensor<[2,1024,40,16],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8346 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2801 = torch.constant.int 1 loc(#loc1) | |
| %8347 = torch.aten.sub.Tensor %8335, %8346, %int1_2801 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8348 = torch.aten.lift_fresh_copy %5622 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8349 = torch.aten.pow.Tensor_Tensor %8348, %8347 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8350 = torch.aten.neg %8349 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8351 = torch.aten.neg %8350 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8352 = torch.aten.div.Tensor %8345, %8351 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8353 = torch.aten.detach %5621 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8354 = torch.aten.detach %8353 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8355 = torch.aten.div.Tensor %result0_2777, %8352 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2802 = torch.constant.int 1 loc(#loc1) | |
| %8356 = torch.aten.add.Tensor %8355, %8354, %int1_2802 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8357 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2803 = torch.constant.int 1 loc(#loc1) | |
| %8358 = torch.aten.sub.Tensor %8335, %8357, %int1_2803 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8359 = torch.aten.lift_fresh_copy %5620 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8360 = torch.aten.pow.Tensor_Tensor %8359, %8358 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8361 = torch.aten.neg %8360 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8362 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2804 = torch.constant.int 1 loc(#loc1) | |
| %8363 = torch.aten.sub.Tensor %8335, %8362, %int1_2804 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8364 = torch.aten.lift_fresh_copy %5619 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8365 = torch.aten.pow.Tensor_Tensor %8364, %8363 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8366 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2805 = torch.constant.int 1 loc(#loc1) | |
| %8367 = torch.aten.sub.Tensor %8365, %8366, %int1_2805 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8368 = torch.aten.gt.Tensor %8356, %8367 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8369 = torch.aten.where.self %8368, %8367, %8356 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8370 = torch.aten.lt.Tensor %8369, %8361 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8371 = torch.aten.where.self %8370, %8361, %8369 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8372 = torch.operator "aten.round"(%8371) : (!torch.tensor<[2,1024,640],f32>) -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_2806 = torch.constant.int 1 loc(#loc1) | |
| %8373 = torch.aten.sub.Tensor %8372, %8354, %int1_2806 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8374 = torch.aten.mul.Tensor %8373, %8352 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_2807 = torch.constant.int 640 loc(#loc1) | |
| %int40_2808 = torch.constant.int 40 loc(#loc1) | |
| %int16_2809 = torch.constant.int 16 loc(#loc1) | |
| %8375 = torch.prim.ListConstruct %int640_2807, %int40_2808, %int16_2809 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2810 = torch.constant.bool false loc(#loc1) | |
| %8376 = torch.aten.expand %5618, %8375, %false_2810 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int0_2811 = torch.constant.int 0 loc(#loc1) | |
| %8377 = torch.aten.clone %8376, %int0_2811 : !torch.tensor<[640,40,16],f16>, !torch.int -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int640_2812 = torch.constant.int 640 loc(#loc1) | |
| %int640_2813 = torch.constant.int 640 loc(#loc1) | |
| %8378 = torch.prim.ListConstruct %int640_2812, %int640_2813 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8379 = torch.aten._unsafe_view %8377, %8378 : !torch.tensor<[640,40,16],f16>, !torch.list<int> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %8380 = torch.aten.mul.Tensor %5617, %8379 : !torch.tensor, !torch.tensor<[640,640],f16> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %int6_2814 = torch.constant.int 6 loc(#loc1) | |
| %none_2815 = torch.constant.none loc(#loc0) | |
| %none_2816 = torch.constant.none loc(#loc0) | |
| %none_2817 = torch.constant.none loc(#loc0) | |
| %false_2818 = torch.constant.bool false loc(#loc1) | |
| %none_2819 = torch.constant.none loc(#loc0) | |
| %8381 = torch.aten._to_copy %8380, %int6_2814, %none_2815, %none_2816, %none_2817, %false_2818, %none_2819 : !torch.tensor<[640,640],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %8382 = torch.aten.t %8381 : !torch.tensor<[640,640],f32> -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %int2048_2820 = torch.constant.int 2048 loc(#loc1) | |
| %int640_2821 = torch.constant.int 640 loc(#loc1) | |
| %8383 = torch.prim.ListConstruct %int2048_2820, %int640_2821 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8384 = torch.aten.view %8374, %8383 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int1_2822 = torch.constant.int 1 loc(#loc1) | |
| %int1_2823 = torch.constant.int 1 loc(#loc1) | |
| %8385 = torch.aten.addmm %5616, %8384, %8382, %int1_2822, %int1_2823 : !torch.tensor, !torch.tensor<[2048,640],f32>, !torch.tensor<[640,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int2_2824 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2825 = torch.constant.int 1024 loc(#loc1) | |
| %int640_2826 = torch.constant.int 640 loc(#loc1) | |
| %8386 = torch.prim.ListConstruct %int2_2824, %int1024_2825, %int640_2826 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8387 = torch.aten.view %8385, %8386 : !torch.tensor<[2048,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8388 = torch.aten.detach %5615 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8389 = torch.aten.detach %8388 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2827 = torch.constant.int 2 loc(#loc1) | |
| %int77_2828 = torch.constant.int 77 loc(#loc1) | |
| %int48_2829 = torch.constant.int 48 loc(#loc1) | |
| %int16_2830 = torch.constant.int 16 loc(#loc1) | |
| %8390 = torch.prim.ListConstruct %int2_2827, %int77_2828, %int48_2829, %int16_2830 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8391 = torch.aten.view %3921, %8390 : !torch.tensor<[2,77,768],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %8392 = torch.aten.abs %8391 : !torch.tensor<[2,77,48,16],f32> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int3_2831 = torch.constant.int 3 loc(#loc1) | |
| %true_2832 = torch.constant.bool true loc(#loc1) | |
| %values_2833, %indices_2834 = torch.aten.max.dim %8392, %int3_2831, %true_2832 : !torch.tensor<[2,77,48,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,77,48,1],f32>, !torch.tensor<[2,77,48,1],si64> loc(#loc1) | |
| %int2_2835 = torch.constant.int 2 loc(#loc1) | |
| %int77_2836 = torch.constant.int 77 loc(#loc1) | |
| %int48_2837 = torch.constant.int 48 loc(#loc1) | |
| %int1_2838 = torch.constant.int 1 loc(#loc1) | |
| %8393 = torch.prim.ListConstruct %int2_2835, %int77_2836, %int48_2837, %int1_2838 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8394 = torch.aten.view %values_2833, %8393 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,1],f32> loc(#loc1) | |
| %int2_2839 = torch.constant.int 2 loc(#loc1) | |
| %int77_2840 = torch.constant.int 77 loc(#loc1) | |
| %int48_2841 = torch.constant.int 48 loc(#loc1) | |
| %int16_2842 = torch.constant.int 16 loc(#loc1) | |
| %8395 = torch.prim.ListConstruct %int2_2839, %int77_2840, %int48_2841, %int16_2842 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2843 = torch.constant.bool false loc(#loc1) | |
| %8396 = torch.aten.expand %8394, %8395, %false_2843 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int0_2844 = torch.constant.int 0 loc(#loc1) | |
| %8397 = torch.aten.clone %8396, %int0_2844 : !torch.tensor<[2,77,48,16],f32>, !torch.int -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int2_2845 = torch.constant.int 2 loc(#loc1) | |
| %int77_2846 = torch.constant.int 77 loc(#loc1) | |
| %int768_2847 = torch.constant.int 768 loc(#loc1) | |
| %8398 = torch.prim.ListConstruct %int2_2845, %int77_2846, %int768_2847 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8399 = torch.aten._unsafe_view %8397, %8398 : !torch.tensor<[2,77,48,16],f32>, !torch.list<int> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8400 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2848 = torch.constant.int 1 loc(#loc1) | |
| %8401 = torch.aten.sub.Tensor %8389, %8400, %int1_2848 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8402 = torch.aten.lift_fresh_copy %5614 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8403 = torch.aten.pow.Tensor_Tensor %8402, %8401 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8404 = torch.aten.neg %8403 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8405 = torch.aten.neg %8404 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8406 = torch.aten.div.Tensor %8399, %8405 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8407 = torch.aten.detach %5613 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8408 = torch.aten.detach %8407 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8409 = torch.aten.div.Tensor %3921, %8406 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_2849 = torch.constant.int 1 loc(#loc1) | |
| %8410 = torch.aten.add.Tensor %8409, %8408, %int1_2849 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8411 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2850 = torch.constant.int 1 loc(#loc1) | |
| %8412 = torch.aten.sub.Tensor %8389, %8411, %int1_2850 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8413 = torch.aten.lift_fresh_copy %5612 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8414 = torch.aten.pow.Tensor_Tensor %8413, %8412 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8415 = torch.aten.neg %8414 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8416 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2851 = torch.constant.int 1 loc(#loc1) | |
| %8417 = torch.aten.sub.Tensor %8389, %8416, %int1_2851 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8418 = torch.aten.lift_fresh_copy %5611 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8419 = torch.aten.pow.Tensor_Tensor %8418, %8417 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8420 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2852 = torch.constant.int 1 loc(#loc1) | |
| %8421 = torch.aten.sub.Tensor %8419, %8420, %int1_2852 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8422 = torch.aten.gt.Tensor %8410, %8421 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %8423 = torch.aten.where.self %8422, %8421, %8410 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8424 = torch.aten.lt.Tensor %8423, %8415 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %8425 = torch.aten.where.self %8424, %8415, %8423 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8426 = torch.operator "aten.round"(%8425) : (!torch.tensor<[2,77,768],f32>) -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_2853 = torch.constant.int 1 loc(#loc1) | |
| %8427 = torch.aten.sub.Tensor %8426, %8408, %int1_2853 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8428 = torch.aten.mul.Tensor %8427, %8406 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int640_2854 = torch.constant.int 640 loc(#loc1) | |
| %int48_2855 = torch.constant.int 48 loc(#loc1) | |
| %int16_2856 = torch.constant.int 16 loc(#loc1) | |
| %8429 = torch.prim.ListConstruct %int640_2854, %int48_2855, %int16_2856 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2857 = torch.constant.bool false loc(#loc1) | |
| %8430 = torch.aten.expand %5610, %8429, %false_2857 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,48,16],f16> loc(#loc1) | |
| %int0_2858 = torch.constant.int 0 loc(#loc1) | |
| %8431 = torch.aten.clone %8430, %int0_2858 : !torch.tensor<[640,48,16],f16>, !torch.int -> !torch.tensor<[640,48,16],f16> loc(#loc1) | |
| %int640_2859 = torch.constant.int 640 loc(#loc1) | |
| %int768_2860 = torch.constant.int 768 loc(#loc1) | |
| %8432 = torch.prim.ListConstruct %int640_2859, %int768_2860 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8433 = torch.aten._unsafe_view %8431, %8432 : !torch.tensor<[640,48,16],f16>, !torch.list<int> -> !torch.tensor<[640,768],f16> loc(#loc1) | |
| %8434 = torch.aten.mul.Tensor %5609, %8433 : !torch.tensor, !torch.tensor<[640,768],f16> -> !torch.tensor<[640,768],f16> loc(#loc1) | |
| %int6_2861 = torch.constant.int 6 loc(#loc1) | |
| %none_2862 = torch.constant.none loc(#loc0) | |
| %none_2863 = torch.constant.none loc(#loc0) | |
| %none_2864 = torch.constant.none loc(#loc0) | |
| %false_2865 = torch.constant.bool false loc(#loc1) | |
| %none_2866 = torch.constant.none loc(#loc0) | |
| %8435 = torch.aten._to_copy %8434, %int6_2861, %none_2862, %none_2863, %none_2864, %false_2865, %none_2866 : !torch.tensor<[640,768],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,768],f32> loc(#loc1) | |
| %8436 = torch.aten.t %8435 : !torch.tensor<[640,768],f32> -> !torch.tensor<[768,640],f32> loc(#loc1) | |
| %int154_2867 = torch.constant.int 154 loc(#loc1) | |
| %int768_2868 = torch.constant.int 768 loc(#loc1) | |
| %8437 = torch.prim.ListConstruct %int154_2867, %int768_2868 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int768_2869 = torch.constant.int 768 loc(#loc1) | |
| %int1_2870 = torch.constant.int 1 loc(#loc1) | |
| %8438 = torch.prim.ListConstruct %int768_2869, %int1_2870 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8439 = torch.aten._reshape_alias %8428, %8437, %8438 : !torch.tensor<[2,77,768],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[154,768],f32> loc(#loc1) | |
| %8440 = torch.aten.mm %8439, %8436 : !torch.tensor<[154,768],f32>, !torch.tensor<[768,640],f32> -> !torch.tensor<[154,640],f32> loc(#loc1) | |
| %int2_2871 = torch.constant.int 2 loc(#loc1) | |
| %int77_2872 = torch.constant.int 77 loc(#loc1) | |
| %int640_2873 = torch.constant.int 640 loc(#loc1) | |
| %8441 = torch.prim.ListConstruct %int2_2871, %int77_2872, %int640_2873 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8442 = torch.aten._unsafe_view %8440, %8441 : !torch.tensor<[154,640],f32>, !torch.list<int> -> !torch.tensor<[2,77,640],f32> loc(#loc1) | |
| %8443 = torch.aten.detach %5608 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8444 = torch.aten.detach %8443 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_2874 = torch.constant.int 2 loc(#loc1) | |
| %int77_2875 = torch.constant.int 77 loc(#loc1) | |
| %int48_2876 = torch.constant.int 48 loc(#loc1) | |
| %int16_2877 = torch.constant.int 16 loc(#loc1) | |
| %8445 = torch.prim.ListConstruct %int2_2874, %int77_2875, %int48_2876, %int16_2877 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8446 = torch.aten.view %3921, %8445 : !torch.tensor<[2,77,768],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %8447 = torch.aten.abs %8446 : !torch.tensor<[2,77,48,16],f32> -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int3_2878 = torch.constant.int 3 loc(#loc1) | |
| %true_2879 = torch.constant.bool true loc(#loc1) | |
| %values_2880, %indices_2881 = torch.aten.max.dim %8447, %int3_2878, %true_2879 : !torch.tensor<[2,77,48,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,77,48,1],f32>, !torch.tensor<[2,77,48,1],si64> loc(#loc1) | |
| %int2_2882 = torch.constant.int 2 loc(#loc1) | |
| %int77_2883 = torch.constant.int 77 loc(#loc1) | |
| %int48_2884 = torch.constant.int 48 loc(#loc1) | |
| %int1_2885 = torch.constant.int 1 loc(#loc1) | |
| %8448 = torch.prim.ListConstruct %int2_2882, %int77_2883, %int48_2884, %int1_2885 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8449 = torch.aten.view %values_2880, %8448 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int> -> !torch.tensor<[2,77,48,1],f32> loc(#loc1) | |
| %int2_2886 = torch.constant.int 2 loc(#loc1) | |
| %int77_2887 = torch.constant.int 77 loc(#loc1) | |
| %int48_2888 = torch.constant.int 48 loc(#loc1) | |
| %int16_2889 = torch.constant.int 16 loc(#loc1) | |
| %8450 = torch.prim.ListConstruct %int2_2886, %int77_2887, %int48_2888, %int16_2889 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2890 = torch.constant.bool false loc(#loc1) | |
| %8451 = torch.aten.expand %8449, %8450, %false_2890 : !torch.tensor<[2,77,48,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int0_2891 = torch.constant.int 0 loc(#loc1) | |
| %8452 = torch.aten.clone %8451, %int0_2891 : !torch.tensor<[2,77,48,16],f32>, !torch.int -> !torch.tensor<[2,77,48,16],f32> loc(#loc1) | |
| %int2_2892 = torch.constant.int 2 loc(#loc1) | |
| %int77_2893 = torch.constant.int 77 loc(#loc1) | |
| %int768_2894 = torch.constant.int 768 loc(#loc1) | |
| %8453 = torch.prim.ListConstruct %int2_2892, %int77_2893, %int768_2894 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8454 = torch.aten._unsafe_view %8452, %8453 : !torch.tensor<[2,77,48,16],f32>, !torch.list<int> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8455 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2895 = torch.constant.int 1 loc(#loc1) | |
| %8456 = torch.aten.sub.Tensor %8444, %8455, %int1_2895 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8457 = torch.aten.lift_fresh_copy %5607 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8458 = torch.aten.pow.Tensor_Tensor %8457, %8456 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8459 = torch.aten.neg %8458 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8460 = torch.aten.neg %8459 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8461 = torch.aten.div.Tensor %8454, %8460 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8462 = torch.aten.detach %5606 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8463 = torch.aten.detach %8462 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8464 = torch.aten.div.Tensor %3921, %8461 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_2896 = torch.constant.int 1 loc(#loc1) | |
| %8465 = torch.aten.add.Tensor %8464, %8463, %int1_2896 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8466 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2897 = torch.constant.int 1 loc(#loc1) | |
| %8467 = torch.aten.sub.Tensor %8444, %8466, %int1_2897 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8468 = torch.aten.lift_fresh_copy %5605 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8469 = torch.aten.pow.Tensor_Tensor %8468, %8467 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8470 = torch.aten.neg %8469 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8471 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2898 = torch.constant.int 1 loc(#loc1) | |
| %8472 = torch.aten.sub.Tensor %8444, %8471, %int1_2898 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8473 = torch.aten.lift_fresh_copy %5604 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8474 = torch.aten.pow.Tensor_Tensor %8473, %8472 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8475 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_2899 = torch.constant.int 1 loc(#loc1) | |
| %8476 = torch.aten.sub.Tensor %8474, %8475, %int1_2899 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8477 = torch.aten.gt.Tensor %8465, %8476 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %8478 = torch.aten.where.self %8477, %8476, %8465 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8479 = torch.aten.lt.Tensor %8478, %8470 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,77,768],i1> loc(#loc1) | |
| %8480 = torch.aten.where.self %8479, %8470, %8478 : !torch.tensor<[2,77,768],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8481 = torch.operator "aten.round"(%8480) : (!torch.tensor<[2,77,768],f32>) -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int1_2900 = torch.constant.int 1 loc(#loc1) | |
| %8482 = torch.aten.sub.Tensor %8481, %8463, %int1_2900 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %8483 = torch.aten.mul.Tensor %8482, %8461 : !torch.tensor<[2,77,768],f32>, !torch.tensor<[2,77,768],f32> -> !torch.tensor<[2,77,768],f32> loc(#loc1) | |
| %int640_2901 = torch.constant.int 640 loc(#loc1) | |
| %int48_2902 = torch.constant.int 48 loc(#loc1) | |
| %int16_2903 = torch.constant.int 16 loc(#loc1) | |
| %8484 = torch.prim.ListConstruct %int640_2901, %int48_2902, %int16_2903 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2904 = torch.constant.bool false loc(#loc1) | |
| %8485 = torch.aten.expand %5603, %8484, %false_2904 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,48,16],f16> loc(#loc1) | |
| %int0_2905 = torch.constant.int 0 loc(#loc1) | |
| %8486 = torch.aten.clone %8485, %int0_2905 : !torch.tensor<[640,48,16],f16>, !torch.int -> !torch.tensor<[640,48,16],f16> loc(#loc1) | |
| %int640_2906 = torch.constant.int 640 loc(#loc1) | |
| %int768_2907 = torch.constant.int 768 loc(#loc1) | |
| %8487 = torch.prim.ListConstruct %int640_2906, %int768_2907 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8488 = torch.aten._unsafe_view %8486, %8487 : !torch.tensor<[640,48,16],f16>, !torch.list<int> -> !torch.tensor<[640,768],f16> loc(#loc1) | |
| %8489 = torch.aten.mul.Tensor %5602, %8488 : !torch.tensor, !torch.tensor<[640,768],f16> -> !torch.tensor<[640,768],f16> loc(#loc1) | |
| %int6_2908 = torch.constant.int 6 loc(#loc1) | |
| %none_2909 = torch.constant.none loc(#loc0) | |
| %none_2910 = torch.constant.none loc(#loc0) | |
| %none_2911 = torch.constant.none loc(#loc0) | |
| %false_2912 = torch.constant.bool false loc(#loc1) | |
| %none_2913 = torch.constant.none loc(#loc0) | |
| %8490 = torch.aten._to_copy %8489, %int6_2908, %none_2909, %none_2910, %none_2911, %false_2912, %none_2913 : !torch.tensor<[640,768],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,768],f32> loc(#loc1) | |
| %8491 = torch.aten.t %8490 : !torch.tensor<[640,768],f32> -> !torch.tensor<[768,640],f32> loc(#loc1) | |
| %int154_2914 = torch.constant.int 154 loc(#loc1) | |
| %int768_2915 = torch.constant.int 768 loc(#loc1) | |
| %8492 = torch.prim.ListConstruct %int154_2914, %int768_2915 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int768_2916 = torch.constant.int 768 loc(#loc1) | |
| %int1_2917 = torch.constant.int 1 loc(#loc1) | |
| %8493 = torch.prim.ListConstruct %int768_2916, %int1_2917 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8494 = torch.aten._reshape_alias %8483, %8492, %8493 : !torch.tensor<[2,77,768],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[154,768],f32> loc(#loc1) | |
| %8495 = torch.aten.mm %8494, %8491 : !torch.tensor<[154,768],f32>, !torch.tensor<[768,640],f32> -> !torch.tensor<[154,640],f32> loc(#loc1) | |
| %int2_2918 = torch.constant.int 2 loc(#loc1) | |
| %int77_2919 = torch.constant.int 77 loc(#loc1) | |
| %int640_2920 = torch.constant.int 640 loc(#loc1) | |
| %8496 = torch.prim.ListConstruct %int2_2918, %int77_2919, %int640_2920 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8497 = torch.aten._unsafe_view %8495, %8496 : !torch.tensor<[154,640],f32>, !torch.list<int> -> !torch.tensor<[2,77,640],f32> loc(#loc1) | |
| %int2_2921 = torch.constant.int 2 loc(#loc1) | |
| %int1024_2922 = torch.constant.int 1024 loc(#loc1) | |
| %int8_2923 = torch.constant.int 8 loc(#loc1) | |
| %int80_2924 = torch.constant.int 80 loc(#loc1) | |
| %8498 = torch.prim.ListConstruct %int2_2921, %int1024_2922, %int8_2923, %int80_2924 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int655360_2925 = torch.constant.int 655360 loc(#loc1) | |
| %int640_2926 = torch.constant.int 640 loc(#loc1) | |
| %int80_2927 = torch.constant.int 80 loc(#loc1) | |
| %int1_2928 = torch.constant.int 1 loc(#loc1) | |
| %8499 = torch.prim.ListConstruct %int655360_2925, %int640_2926, %int80_2927, %int1_2928 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8500 = torch.aten._reshape_alias %8387, %8498, %8499 : !torch.tensor<[2,1024,640],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,1024,8,80],f32> loc(#loc1) | |
| %int0_2929 = torch.constant.int 0 loc(#loc1) | |
| %int2_2930 = torch.constant.int 2 loc(#loc1) | |
| %int1_2931 = torch.constant.int 1 loc(#loc1) | |
| %int3_2932 = torch.constant.int 3 loc(#loc1) | |
| %8501 = torch.prim.ListConstruct %int0_2929, %int2_2930, %int1_2931, %int3_2932 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8502 = torch.aten.permute %8500, %8501 : !torch.tensor<[2,1024,8,80],f32>, !torch.list<int> -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int0_2933 = torch.constant.int 0 loc(#loc1) | |
| %8503 = torch.aten.clone %8502, %int0_2933 : !torch.tensor<[2,8,1024,80],f32>, !torch.int -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int16_2934 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2935 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2936 = torch.constant.int 80 loc(#loc1) | |
| %8504 = torch.prim.ListConstruct %int16_2934, %int1024_2935, %int80_2936 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8505 = torch.aten._unsafe_view %8503, %8504 : !torch.tensor<[2,8,1024,80],f32>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int2_2937 = torch.constant.int 2 loc(#loc1) | |
| %int77_2938 = torch.constant.int 77 loc(#loc1) | |
| %int8_2939 = torch.constant.int 8 loc(#loc1) | |
| %int80_2940 = torch.constant.int 80 loc(#loc1) | |
| %8506 = torch.prim.ListConstruct %int2_2937, %int77_2938, %int8_2939, %int80_2940 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int49280 = torch.constant.int 49280 loc(#loc1) | |
| %int640_2941 = torch.constant.int 640 loc(#loc1) | |
| %int80_2942 = torch.constant.int 80 loc(#loc1) | |
| %int1_2943 = torch.constant.int 1 loc(#loc1) | |
| %8507 = torch.prim.ListConstruct %int49280, %int640_2941, %int80_2942, %int1_2943 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8508 = torch.aten._reshape_alias %8442, %8506, %8507 : !torch.tensor<[2,77,640],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,77,8,80],f32> loc(#loc1) | |
| %int0_2944 = torch.constant.int 0 loc(#loc1) | |
| %int2_2945 = torch.constant.int 2 loc(#loc1) | |
| %int1_2946 = torch.constant.int 1 loc(#loc1) | |
| %int3_2947 = torch.constant.int 3 loc(#loc1) | |
| %8509 = torch.prim.ListConstruct %int0_2944, %int2_2945, %int1_2946, %int3_2947 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8510 = torch.aten.permute %8508, %8509 : !torch.tensor<[2,77,8,80],f32>, !torch.list<int> -> !torch.tensor<[2,8,77,80],f32> loc(#loc1) | |
| %int0_2948 = torch.constant.int 0 loc(#loc1) | |
| %8511 = torch.aten.clone %8510, %int0_2948 : !torch.tensor<[2,8,77,80],f32>, !torch.int -> !torch.tensor<[2,8,77,80],f32> loc(#loc1) | |
| %int16_2949 = torch.constant.int 16 loc(#loc1) | |
| %int77_2950 = torch.constant.int 77 loc(#loc1) | |
| %int80_2951 = torch.constant.int 80 loc(#loc1) | |
| %8512 = torch.prim.ListConstruct %int16_2949, %int77_2950, %int80_2951 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8513 = torch.aten._unsafe_view %8511, %8512 : !torch.tensor<[2,8,77,80],f32>, !torch.list<int> -> !torch.tensor<[16,77,80],f32> loc(#loc1) | |
| %int2_2952 = torch.constant.int 2 loc(#loc1) | |
| %int77_2953 = torch.constant.int 77 loc(#loc1) | |
| %int8_2954 = torch.constant.int 8 loc(#loc1) | |
| %int80_2955 = torch.constant.int 80 loc(#loc1) | |
| %8514 = torch.prim.ListConstruct %int2_2952, %int77_2953, %int8_2954, %int80_2955 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int49280_2956 = torch.constant.int 49280 loc(#loc1) | |
| %int640_2957 = torch.constant.int 640 loc(#loc1) | |
| %int80_2958 = torch.constant.int 80 loc(#loc1) | |
| %int1_2959 = torch.constant.int 1 loc(#loc1) | |
| %8515 = torch.prim.ListConstruct %int49280_2956, %int640_2957, %int80_2958, %int1_2959 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8516 = torch.aten._reshape_alias %8497, %8514, %8515 : !torch.tensor<[2,77,640],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,77,8,80],f32> loc(#loc1) | |
| %int0_2960 = torch.constant.int 0 loc(#loc1) | |
| %int2_2961 = torch.constant.int 2 loc(#loc1) | |
| %int1_2962 = torch.constant.int 1 loc(#loc1) | |
| %int3_2963 = torch.constant.int 3 loc(#loc1) | |
| %8517 = torch.prim.ListConstruct %int0_2960, %int2_2961, %int1_2962, %int3_2963 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8518 = torch.aten.permute %8516, %8517 : !torch.tensor<[2,77,8,80],f32>, !torch.list<int> -> !torch.tensor<[2,8,77,80],f32> loc(#loc1) | |
| %int0_2964 = torch.constant.int 0 loc(#loc1) | |
| %8519 = torch.aten.clone %8518, %int0_2964 : !torch.tensor<[2,8,77,80],f32>, !torch.int -> !torch.tensor<[2,8,77,80],f32> loc(#loc1) | |
| %int16_2965 = torch.constant.int 16 loc(#loc1) | |
| %int77_2966 = torch.constant.int 77 loc(#loc1) | |
| %int80_2967 = torch.constant.int 80 loc(#loc1) | |
| %8520 = torch.prim.ListConstruct %int16_2965, %int77_2966, %int80_2967 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8521 = torch.aten._unsafe_view %8519, %8520 : !torch.tensor<[2,8,77,80],f32>, !torch.list<int> -> !torch.tensor<[16,77,80],f32> loc(#loc1) | |
| %int-1_2968 = torch.constant.int -1 loc(#loc1) | |
| %int-2_2969 = torch.constant.int -2 loc(#loc1) | |
| %8522 = torch.aten.transpose.int %8513, %int-1_2968, %int-2_2969 : !torch.tensor<[16,77,80],f32>, !torch.int, !torch.int -> !torch.tensor<[16,80,77],f32> loc(#loc1) | |
| %int16_2970 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2971 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2972 = torch.constant.int 80 loc(#loc1) | |
| %8523 = torch.prim.ListConstruct %int16_2970, %int1024_2971, %int80_2972 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2973 = torch.constant.bool false loc(#loc1) | |
| %8524 = torch.aten.expand %8505, %8523, %false_2973 : !torch.tensor<[16,1024,80],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int16_2974 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2975 = torch.constant.int 1024 loc(#loc1) | |
| %int80_2976 = torch.constant.int 80 loc(#loc1) | |
| %8525 = torch.prim.ListConstruct %int16_2974, %int1024_2975, %int80_2976 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int81920_2977 = torch.constant.int 81920 loc(#loc1) | |
| %int80_2978 = torch.constant.int 80 loc(#loc1) | |
| %int1_2979 = torch.constant.int 1 loc(#loc1) | |
| %8526 = torch.prim.ListConstruct %int81920_2977, %int80_2978, %int1_2979 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8527 = torch.aten._reshape_alias %8524, %8525, %8526 : !torch.tensor<[16,1024,80],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int16_2980 = torch.constant.int 16 loc(#loc1) | |
| %int80_2981 = torch.constant.int 80 loc(#loc1) | |
| %int77_2982 = torch.constant.int 77 loc(#loc1) | |
| %8528 = torch.prim.ListConstruct %int16_2980, %int80_2981, %int77_2982 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2983 = torch.constant.bool false loc(#loc1) | |
| %8529 = torch.aten.expand %8522, %8528, %false_2983 : !torch.tensor<[16,80,77],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,80,77],f32> loc(#loc1) | |
| %int16_2984 = torch.constant.int 16 loc(#loc1) | |
| %int80_2985 = torch.constant.int 80 loc(#loc1) | |
| %int77_2986 = torch.constant.int 77 loc(#loc1) | |
| %8530 = torch.prim.ListConstruct %int16_2984, %int80_2985, %int77_2986 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int6160 = torch.constant.int 6160 loc(#loc1) | |
| %int1_2987 = torch.constant.int 1 loc(#loc1) | |
| %int80_2988 = torch.constant.int 80 loc(#loc1) | |
| %8531 = torch.prim.ListConstruct %int6160, %int1_2987, %int80_2988 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8532 = torch.aten._reshape_alias %8529, %8530, %8531 : !torch.tensor<[16,80,77],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,80,77],f32> loc(#loc1) | |
| %8533 = torch.aten.bmm %8527, %8532 : !torch.tensor<[16,1024,80],f32>, !torch.tensor<[16,80,77],f32> -> !torch.tensor<[16,1024,77],f32> loc(#loc1) | |
| %int16_2989 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2990 = torch.constant.int 1024 loc(#loc1) | |
| %int77_2991 = torch.constant.int 77 loc(#loc1) | |
| %8534 = torch.prim.ListConstruct %int16_2989, %int1024_2990, %int77_2991 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8535 = torch.aten._unsafe_view %8533, %8534 : !torch.tensor<[16,1024,77],f32>, !torch.list<int> -> !torch.tensor<[16,1024,77],f32> loc(#loc1) | |
| %8536 = torch.tensor.literal(dense<0.11180339887498948> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %8537 = torch.aten.mul.Tensor %8535, %8536 : !torch.tensor<[16,1024,77],f32>, !torch.tensor<[],f64> -> !torch.tensor<[16,1024,77],f32> loc(#loc1) | |
| %int-1_2992 = torch.constant.int -1 loc(#loc1) | |
| %false_2993 = torch.constant.bool false loc(#loc1) | |
| %8538 = torch.aten._softmax %8537, %int-1_2992, %false_2993 : !torch.tensor<[16,1024,77],f32>, !torch.int, !torch.bool -> !torch.tensor<[16,1024,77],f32> loc(#loc1) | |
| %int16_2994 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2995 = torch.constant.int 1024 loc(#loc1) | |
| %int77_2996 = torch.constant.int 77 loc(#loc1) | |
| %8539 = torch.prim.ListConstruct %int16_2994, %int1024_2995, %int77_2996 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_2997 = torch.constant.bool false loc(#loc1) | |
| %8540 = torch.aten.expand %8538, %8539, %false_2997 : !torch.tensor<[16,1024,77],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,1024,77],f32> loc(#loc1) | |
| %int16_2998 = torch.constant.int 16 loc(#loc1) | |
| %int1024_2999 = torch.constant.int 1024 loc(#loc1) | |
| %int77_3000 = torch.constant.int 77 loc(#loc1) | |
| %8541 = torch.prim.ListConstruct %int16_2998, %int1024_2999, %int77_3000 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int78848 = torch.constant.int 78848 loc(#loc1) | |
| %int77_3001 = torch.constant.int 77 loc(#loc1) | |
| %int1_3002 = torch.constant.int 1 loc(#loc1) | |
| %8542 = torch.prim.ListConstruct %int78848, %int77_3001, %int1_3002 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8543 = torch.aten._reshape_alias %8540, %8541, %8542 : !torch.tensor<[16,1024,77],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,1024,77],f32> loc(#loc1) | |
| %int16_3003 = torch.constant.int 16 loc(#loc1) | |
| %int77_3004 = torch.constant.int 77 loc(#loc1) | |
| %int80_3005 = torch.constant.int 80 loc(#loc1) | |
| %8544 = torch.prim.ListConstruct %int16_3003, %int77_3004, %int80_3005 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3006 = torch.constant.bool false loc(#loc1) | |
| %8545 = torch.aten.expand %8521, %8544, %false_3006 : !torch.tensor<[16,77,80],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[16,77,80],f32> loc(#loc1) | |
| %int16_3007 = torch.constant.int 16 loc(#loc1) | |
| %int77_3008 = torch.constant.int 77 loc(#loc1) | |
| %int80_3009 = torch.constant.int 80 loc(#loc1) | |
| %8546 = torch.prim.ListConstruct %int16_3007, %int77_3008, %int80_3009 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int6160_3010 = torch.constant.int 6160 loc(#loc1) | |
| %int80_3011 = torch.constant.int 80 loc(#loc1) | |
| %int1_3012 = torch.constant.int 1 loc(#loc1) | |
| %8547 = torch.prim.ListConstruct %int6160_3010, %int80_3011, %int1_3012 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8548 = torch.aten._reshape_alias %8545, %8546, %8547 : !torch.tensor<[16,77,80],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[16,77,80],f32> loc(#loc1) | |
| %8549 = torch.aten.bmm %8543, %8548 : !torch.tensor<[16,1024,77],f32>, !torch.tensor<[16,77,80],f32> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int16_3013 = torch.constant.int 16 loc(#loc1) | |
| %int1024_3014 = torch.constant.int 1024 loc(#loc1) | |
| %int80_3015 = torch.constant.int 80 loc(#loc1) | |
| %8550 = torch.prim.ListConstruct %int16_3013, %int1024_3014, %int80_3015 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8551 = torch.aten._unsafe_view %8549, %8550 : !torch.tensor<[16,1024,80],f32>, !torch.list<int> -> !torch.tensor<[16,1024,80],f32> loc(#loc1) | |
| %int2_3016 = torch.constant.int 2 loc(#loc1) | |
| %int8_3017 = torch.constant.int 8 loc(#loc1) | |
| %int1024_3018 = torch.constant.int 1024 loc(#loc1) | |
| %int80_3019 = torch.constant.int 80 loc(#loc1) | |
| %8552 = torch.prim.ListConstruct %int2_3016, %int8_3017, %int1024_3018, %int80_3019 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int655360_3020 = torch.constant.int 655360 loc(#loc1) | |
| %int81920_3021 = torch.constant.int 81920 loc(#loc1) | |
| %int80_3022 = torch.constant.int 80 loc(#loc1) | |
| %int1_3023 = torch.constant.int 1 loc(#loc1) | |
| %8553 = torch.prim.ListConstruct %int655360_3020, %int81920_3021, %int80_3022, %int1_3023 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8554 = torch.aten._reshape_alias %8551, %8552, %8553 : !torch.tensor<[16,1024,80],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,8,1024,80],f32> loc(#loc1) | |
| %int0_3024 = torch.constant.int 0 loc(#loc1) | |
| %int2_3025 = torch.constant.int 2 loc(#loc1) | |
| %int1_3026 = torch.constant.int 1 loc(#loc1) | |
| %int3_3027 = torch.constant.int 3 loc(#loc1) | |
| %8555 = torch.prim.ListConstruct %int0_3024, %int2_3025, %int1_3026, %int3_3027 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8556 = torch.aten.permute %8554, %8555 : !torch.tensor<[2,8,1024,80],f32>, !torch.list<int> -> !torch.tensor<[2,1024,8,80],f32> loc(#loc1) | |
| %int0_3028 = torch.constant.int 0 loc(#loc1) | |
| %8557 = torch.aten.clone %8556, %int0_3028 : !torch.tensor<[2,1024,8,80],f32>, !torch.int -> !torch.tensor<[2,1024,8,80],f32> loc(#loc1) | |
| %int2_3029 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3030 = torch.constant.int 1024 loc(#loc1) | |
| %int640_3031 = torch.constant.int 640 loc(#loc1) | |
| %8558 = torch.prim.ListConstruct %int2_3029, %int1024_3030, %int640_3031 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8559 = torch.aten._unsafe_view %8557, %8558 : !torch.tensor<[2,1024,8,80],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8560 = torch.aten.detach %5601 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8561 = torch.aten.detach %8560 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_3032 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3033 = torch.constant.int 1024 loc(#loc1) | |
| %int40_3034 = torch.constant.int 40 loc(#loc1) | |
| %int16_3035 = torch.constant.int 16 loc(#loc1) | |
| %8562 = torch.prim.ListConstruct %int2_3032, %int1024_3033, %int40_3034, %int16_3035 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8563 = torch.aten.view %8559, %8562 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %8564 = torch.aten.abs %8563 : !torch.tensor<[2,1024,40,16],f32> -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int3_3036 = torch.constant.int 3 loc(#loc1) | |
| %true_3037 = torch.constant.bool true loc(#loc1) | |
| %values_3038, %indices_3039 = torch.aten.max.dim %8564, %int3_3036, %true_3037 : !torch.tensor<[2,1024,40,16],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,1024,40,1],f32>, !torch.tensor<[2,1024,40,1],si64> loc(#loc1) | |
| %int2_3040 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3041 = torch.constant.int 1024 loc(#loc1) | |
| %int40_3042 = torch.constant.int 40 loc(#loc1) | |
| %int1_3043 = torch.constant.int 1 loc(#loc1) | |
| %8565 = torch.prim.ListConstruct %int2_3040, %int1024_3041, %int40_3042, %int1_3043 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8566 = torch.aten.view %values_3038, %8565 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int> -> !torch.tensor<[2,1024,40,1],f32> loc(#loc1) | |
| %int2_3044 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3045 = torch.constant.int 1024 loc(#loc1) | |
| %int40_3046 = torch.constant.int 40 loc(#loc1) | |
| %int16_3047 = torch.constant.int 16 loc(#loc1) | |
| %8567 = torch.prim.ListConstruct %int2_3044, %int1024_3045, %int40_3046, %int16_3047 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3048 = torch.constant.bool false loc(#loc1) | |
| %8568 = torch.aten.expand %8566, %8567, %false_3048 : !torch.tensor<[2,1024,40,1],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int0_3049 = torch.constant.int 0 loc(#loc1) | |
| %8569 = torch.aten.clone %8568, %int0_3049 : !torch.tensor<[2,1024,40,16],f32>, !torch.int -> !torch.tensor<[2,1024,40,16],f32> loc(#loc1) | |
| %int2_3050 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3051 = torch.constant.int 1024 loc(#loc1) | |
| %int640_3052 = torch.constant.int 640 loc(#loc1) | |
| %8570 = torch.prim.ListConstruct %int2_3050, %int1024_3051, %int640_3052 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8571 = torch.aten._unsafe_view %8569, %8570 : !torch.tensor<[2,1024,40,16],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8572 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3053 = torch.constant.int 1 loc(#loc1) | |
| %8573 = torch.aten.sub.Tensor %8561, %8572, %int1_3053 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8574 = torch.aten.lift_fresh_copy %5600 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8575 = torch.aten.pow.Tensor_Tensor %8574, %8573 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8576 = torch.aten.neg %8575 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8577 = torch.aten.neg %8576 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8578 = torch.aten.div.Tensor %8571, %8577 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8579 = torch.aten.detach %5599 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8580 = torch.aten.detach %8579 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8581 = torch.aten.div.Tensor %8559, %8578 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_3054 = torch.constant.int 1 loc(#loc1) | |
| %8582 = torch.aten.add.Tensor %8581, %8580, %int1_3054 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8583 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3055 = torch.constant.int 1 loc(#loc1) | |
| %8584 = torch.aten.sub.Tensor %8561, %8583, %int1_3055 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8585 = torch.aten.lift_fresh_copy %5598 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8586 = torch.aten.pow.Tensor_Tensor %8585, %8584 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8587 = torch.aten.neg %8586 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8588 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3056 = torch.constant.int 1 loc(#loc1) | |
| %8589 = torch.aten.sub.Tensor %8561, %8588, %int1_3056 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8590 = torch.aten.lift_fresh_copy %5597 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8591 = torch.aten.pow.Tensor_Tensor %8590, %8589 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8592 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3057 = torch.constant.int 1 loc(#loc1) | |
| %8593 = torch.aten.sub.Tensor %8591, %8592, %int1_3057 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8594 = torch.aten.gt.Tensor %8582, %8593 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8595 = torch.aten.where.self %8594, %8593, %8582 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8596 = torch.aten.lt.Tensor %8595, %8587 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,1024,640],i1> loc(#loc1) | |
| %8597 = torch.aten.where.self %8596, %8587, %8595 : !torch.tensor<[2,1024,640],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8598 = torch.operator "aten.round"(%8597) : (!torch.tensor<[2,1024,640],f32>) -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_3058 = torch.constant.int 1 loc(#loc1) | |
| %8599 = torch.aten.sub.Tensor %8598, %8580, %int1_3058 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %8600 = torch.aten.mul.Tensor %8599, %8578 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_3059 = torch.constant.int 640 loc(#loc1) | |
| %int40_3060 = torch.constant.int 40 loc(#loc1) | |
| %int16_3061 = torch.constant.int 16 loc(#loc1) | |
| %8601 = torch.prim.ListConstruct %int640_3059, %int40_3060, %int16_3061 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3062 = torch.constant.bool false loc(#loc1) | |
| %8602 = torch.aten.expand %5596, %8601, %false_3062 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int0_3063 = torch.constant.int 0 loc(#loc1) | |
| %8603 = torch.aten.clone %8602, %int0_3063 : !torch.tensor<[640,40,16],f16>, !torch.int -> !torch.tensor<[640,40,16],f16> loc(#loc1) | |
| %int640_3064 = torch.constant.int 640 loc(#loc1) | |
| %int640_3065 = torch.constant.int 640 loc(#loc1) | |
| %8604 = torch.prim.ListConstruct %int640_3064, %int640_3065 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8605 = torch.aten._unsafe_view %8603, %8604 : !torch.tensor<[640,40,16],f16>, !torch.list<int> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %8606 = torch.aten.mul.Tensor %5595, %8605 : !torch.tensor, !torch.tensor<[640,640],f16> -> !torch.tensor<[640,640],f16> loc(#loc1) | |
| %int6_3066 = torch.constant.int 6 loc(#loc1) | |
| %none_3067 = torch.constant.none loc(#loc0) | |
| %none_3068 = torch.constant.none loc(#loc0) | |
| %none_3069 = torch.constant.none loc(#loc0) | |
| %false_3070 = torch.constant.bool false loc(#loc1) | |
| %none_3071 = torch.constant.none loc(#loc0) | |
| %8607 = torch.aten._to_copy %8606, %int6_3066, %none_3067, %none_3068, %none_3069, %false_3070, %none_3071 : !torch.tensor<[640,640],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %8608 = torch.aten.t %8607 : !torch.tensor<[640,640],f32> -> !torch.tensor<[640,640],f32> loc(#loc1) | |
| %int2048_3072 = torch.constant.int 2048 loc(#loc1) | |
| %int640_3073 = torch.constant.int 640 loc(#loc1) | |
| %8609 = torch.prim.ListConstruct %int2048_3072, %int640_3073 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8610 = torch.aten.view %8600, %8609 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int1_3074 = torch.constant.int 1 loc(#loc1) | |
| %int1_3075 = torch.constant.int 1 loc(#loc1) | |
| %8611 = torch.aten.addmm %5594, %8610, %8608, %int1_3074, %int1_3075 : !torch.tensor, !torch.tensor<[2048,640],f32>, !torch.tensor<[640,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int2_3076 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3077 = torch.constant.int 1024 loc(#loc1) | |
| %int640_3078 = torch.constant.int 640 loc(#loc1) | |
| %8612 = torch.prim.ListConstruct %int2_3076, %int1024_3077, %int640_3078 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8613 = torch.aten.view %8611, %8612 : !torch.tensor<[2048,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_3079 = torch.constant.int 1 loc(#loc1) | |
| %8614 = torch.aten.add.Tensor %8613, %8332, %int1_3079 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int640_3080 = torch.constant.int 640 loc(#loc1) | |
| %8615 = torch.prim.ListConstruct %int640_3080 : (!torch.int) -> !torch.list<int> loc(#loc0) | |
| %none_3081 = torch.constant.none loc(#loc0) | |
| %none_3082 = torch.constant.none loc(#loc0) | |
| %float1.000000e-05_3083 = torch.constant.float 1.000000e-05 loc(#loc1) | |
| %result0_3084, %result1_3085, %result2_3086 = torch.aten.native_layer_norm %8614, %8615, %none_3081, %none_3082, %float1.000000e-05_3083 : !torch.tensor<[2,1024,640],f32>, !torch.list<int>, !torch.none, !torch.none, !torch.float -> !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,1],f32>, !torch.tensor<[2,1024,1],f32> loc(#loc1) | |
| %int5120 = torch.constant.int 5120 loc(#loc1) | |
| %int40_3087 = torch.constant.int 40 loc(#loc1) | |
| %int16_3088 = torch.constant.int 16 loc(#loc1) | |
| %8616 = torch.prim.ListConstruct %int5120, %int40_3087, %int16_3088 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3089 = torch.constant.bool false loc(#loc1) | |
| %8617 = torch.aten.expand %5593, %8616, %false_3089 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[5120,40,16],f16> loc(#loc1) | |
| %int0_3090 = torch.constant.int 0 loc(#loc1) | |
| %8618 = torch.aten.clone %8617, %int0_3090 : !torch.tensor<[5120,40,16],f16>, !torch.int -> !torch.tensor<[5120,40,16],f16> loc(#loc1) | |
| %int5120_3091 = torch.constant.int 5120 loc(#loc1) | |
| %int640_3092 = torch.constant.int 640 loc(#loc1) | |
| %8619 = torch.prim.ListConstruct %int5120_3091, %int640_3092 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8620 = torch.aten._unsafe_view %8618, %8619 : !torch.tensor<[5120,40,16],f16>, !torch.list<int> -> !torch.tensor<[5120,640],f16> loc(#loc1) | |
| %8621 = torch.aten.mul.Tensor %5592, %8620 : !torch.tensor, !torch.tensor<[5120,640],f16> -> !torch.tensor<[5120,640],f16> loc(#loc1) | |
| %int6_3093 = torch.constant.int 6 loc(#loc1) | |
| %none_3094 = torch.constant.none loc(#loc0) | |
| %none_3095 = torch.constant.none loc(#loc0) | |
| %none_3096 = torch.constant.none loc(#loc0) | |
| %false_3097 = torch.constant.bool false loc(#loc1) | |
| %none_3098 = torch.constant.none loc(#loc0) | |
| %8622 = torch.aten._to_copy %8621, %int6_3093, %none_3094, %none_3095, %none_3096, %false_3097, %none_3098 : !torch.tensor<[5120,640],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[5120,640],f32> loc(#loc1) | |
| %8623 = torch.aten.t %8622 : !torch.tensor<[5120,640],f32> -> !torch.tensor<[640,5120],f32> loc(#loc1) | |
| %int2048_3099 = torch.constant.int 2048 loc(#loc1) | |
| %int640_3100 = torch.constant.int 640 loc(#loc1) | |
| %8624 = torch.prim.ListConstruct %int2048_3099, %int640_3100 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8625 = torch.aten.view %result0_3084, %8624 : !torch.tensor<[2,1024,640],f32>, !torch.list<int> -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int1_3101 = torch.constant.int 1 loc(#loc1) | |
| %int1_3102 = torch.constant.int 1 loc(#loc1) | |
| %8626 = torch.aten.addmm %5591, %8625, %8623, %int1_3101, %int1_3102 : !torch.tensor, !torch.tensor<[2048,640],f32>, !torch.tensor<[640,5120],f32>, !torch.int, !torch.int -> !torch.tensor<[2048,5120],f32> loc(#loc1) | |
| %int2_3103 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3104 = torch.constant.int 1024 loc(#loc1) | |
| %int5120_3105 = torch.constant.int 5120 loc(#loc1) | |
| %8627 = torch.prim.ListConstruct %int2_3103, %int1024_3104, %int5120_3105 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8628 = torch.aten.view %8626, %8627 : !torch.tensor<[2048,5120],f32>, !torch.list<int> -> !torch.tensor<[2,1024,5120],f32> loc(#loc1) | |
| %int-1_3106 = torch.constant.int -1 loc(#loc1) | |
| %int0_3107 = torch.constant.int 0 loc(#loc1) | |
| %int2560_3108 = torch.constant.int 2560 loc(#loc1) | |
| %int1_3109 = torch.constant.int 1 loc(#loc1) | |
| %8629 = torch.aten.slice.Tensor %8628, %int-1_3106, %int0_3107, %int2560_3108, %int1_3109 : !torch.tensor<[2,1024,5120],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,1024,2560],f32> loc(#loc1) | |
| %int-1_3110 = torch.constant.int -1 loc(#loc1) | |
| %int2560_3111 = torch.constant.int 2560 loc(#loc1) | |
| %int5120_3112 = torch.constant.int 5120 loc(#loc1) | |
| %int1_3113 = torch.constant.int 1 loc(#loc1) | |
| %8630 = torch.aten.slice.Tensor %8628, %int-1_3110, %int2560_3111, %int5120_3112, %int1_3113 : !torch.tensor<[2,1024,5120],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,1024,2560],f32> loc(#loc1) | |
| %str_3114 = torch.constant.str "none" loc(#loc1) | |
| %8631 = torch.aten.gelu %8630, %str_3114 : !torch.tensor<[2,1024,2560],f32>, !torch.str -> !torch.tensor<[2,1024,2560],f32> loc(#loc1) | |
| %8632 = torch.aten.mul.Tensor %8629, %8631 : !torch.tensor<[2,1024,2560],f32>, !torch.tensor<[2,1024,2560],f32> -> !torch.tensor<[2,1024,2560],f32> loc(#loc1) | |
| %int640_3115 = torch.constant.int 640 loc(#loc1) | |
| %int160_3116 = torch.constant.int 160 loc(#loc1) | |
| %int16_3117 = torch.constant.int 16 loc(#loc1) | |
| %8633 = torch.prim.ListConstruct %int640_3115, %int160_3116, %int16_3117 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3118 = torch.constant.bool false loc(#loc1) | |
| %8634 = torch.aten.expand %5590, %8633, %false_3118 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,160,16],f16> loc(#loc1) | |
| %int0_3119 = torch.constant.int 0 loc(#loc1) | |
| %8635 = torch.aten.clone %8634, %int0_3119 : !torch.tensor<[640,160,16],f16>, !torch.int -> !torch.tensor<[640,160,16],f16> loc(#loc1) | |
| %int640_3120 = torch.constant.int 640 loc(#loc1) | |
| %int2560_3121 = torch.constant.int 2560 loc(#loc1) | |
| %8636 = torch.prim.ListConstruct %int640_3120, %int2560_3121 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8637 = torch.aten._unsafe_view %8635, %8636 : !torch.tensor<[640,160,16],f16>, !torch.list<int> -> !torch.tensor<[640,2560],f16> loc(#loc1) | |
| %8638 = torch.aten.mul.Tensor %5589, %8637 : !torch.tensor, !torch.tensor<[640,2560],f16> -> !torch.tensor<[640,2560],f16> loc(#loc1) | |
| %int6_3122 = torch.constant.int 6 loc(#loc1) | |
| %none_3123 = torch.constant.none loc(#loc0) | |
| %none_3124 = torch.constant.none loc(#loc0) | |
| %none_3125 = torch.constant.none loc(#loc0) | |
| %false_3126 = torch.constant.bool false loc(#loc1) | |
| %none_3127 = torch.constant.none loc(#loc0) | |
| %8639 = torch.aten._to_copy %8638, %int6_3122, %none_3123, %none_3124, %none_3125, %false_3126, %none_3127 : !torch.tensor<[640,2560],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,2560],f32> loc(#loc1) | |
| %8640 = torch.aten.t %8639 : !torch.tensor<[640,2560],f32> -> !torch.tensor<[2560,640],f32> loc(#loc1) | |
| %int2048_3128 = torch.constant.int 2048 loc(#loc1) | |
| %int2560_3129 = torch.constant.int 2560 loc(#loc1) | |
| %8641 = torch.prim.ListConstruct %int2048_3128, %int2560_3129 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8642 = torch.aten.view %8632, %8641 : !torch.tensor<[2,1024,2560],f32>, !torch.list<int> -> !torch.tensor<[2048,2560],f32> loc(#loc1) | |
| %int1_3130 = torch.constant.int 1 loc(#loc1) | |
| %int1_3131 = torch.constant.int 1 loc(#loc1) | |
| %8643 = torch.aten.addmm %5588, %8642, %8640, %int1_3130, %int1_3131 : !torch.tensor, !torch.tensor<[2048,2560],f32>, !torch.tensor<[2560,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2048,640],f32> loc(#loc1) | |
| %int2_3132 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3133 = torch.constant.int 1024 loc(#loc1) | |
| %int640_3134 = torch.constant.int 640 loc(#loc1) | |
| %8644 = torch.prim.ListConstruct %int2_3132, %int1024_3133, %int640_3134 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8645 = torch.aten.view %8643, %8644 : !torch.tensor<[2048,640],f32>, !torch.list<int> -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int1_3135 = torch.constant.int 1 loc(#loc1) | |
| %8646 = torch.aten.add.Tensor %8645, %8614, %int1_3135 : !torch.tensor<[2,1024,640],f32>, !torch.tensor<[2,1024,640],f32>, !torch.int -> !torch.tensor<[2,1024,640],f32> loc(#loc1) | |
| %int2_3136 = torch.constant.int 2 loc(#loc1) | |
| %int32_3137 = torch.constant.int 32 loc(#loc1) | |
| %int32_3138 = torch.constant.int 32 loc(#loc1) | |
| %int640_3139 = torch.constant.int 640 loc(#loc1) | |
| %8647 = torch.prim.ListConstruct %int2_3136, %int32_3137, %int32_3138, %int640_3139 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int655360_3140 = torch.constant.int 655360 loc(#loc1) | |
| %int20480_3141 = torch.constant.int 20480 loc(#loc1) | |
| %int640_3142 = torch.constant.int 640 loc(#loc1) | |
| %int1_3143 = torch.constant.int 1 loc(#loc1) | |
| %8648 = torch.prim.ListConstruct %int655360_3140, %int20480_3141, %int640_3142, %int1_3143 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8649 = torch.aten._reshape_alias %8646, %8647, %8648 : !torch.tensor<[2,1024,640],f32>, !torch.list<int>, !torch.list<int> -> !torch.tensor<[2,32,32,640],f32> loc(#loc1) | |
| %int0_3144 = torch.constant.int 0 loc(#loc1) | |
| %int3_3145 = torch.constant.int 3 loc(#loc1) | |
| %int1_3146 = torch.constant.int 1 loc(#loc1) | |
| %int2_3147 = torch.constant.int 2 loc(#loc1) | |
| %8650 = torch.prim.ListConstruct %int0_3144, %int3_3145, %int1_3146, %int2_3147 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8651 = torch.aten.permute %8649, %8650 : !torch.tensor<[2,32,32,640],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8652 = torch.aten.detach %5587 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8653 = torch.aten.detach %8652 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_3148 = torch.constant.int 2 loc(#loc1) | |
| %int40_3149 = torch.constant.int 40 loc(#loc1) | |
| %int16_3150 = torch.constant.int 16 loc(#loc1) | |
| %int32_3151 = torch.constant.int 32 loc(#loc1) | |
| %int32_3152 = torch.constant.int 32 loc(#loc1) | |
| %8654 = torch.prim.ListConstruct %int2_3148, %int40_3149, %int16_3150, %int32_3151, %int32_3152 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8655 = torch.aten.view %8651, %8654 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %8656 = torch.aten.abs %8655 : !torch.tensor<[2,40,16,32,32],f32> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_3153 = torch.constant.int 2 loc(#loc1) | |
| %true_3154 = torch.constant.bool true loc(#loc1) | |
| %values_3155, %indices_3156 = torch.aten.max.dim %8656, %int2_3153, %true_3154 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,40,1,32,32],f32>, !torch.tensor<[2,40,1,32,32],si64> loc(#loc1) | |
| %int2_3157 = torch.constant.int 2 loc(#loc1) | |
| %int40_3158 = torch.constant.int 40 loc(#loc1) | |
| %int1_3159 = torch.constant.int 1 loc(#loc1) | |
| %int32_3160 = torch.constant.int 32 loc(#loc1) | |
| %int32_3161 = torch.constant.int 32 loc(#loc1) | |
| %8657 = torch.prim.ListConstruct %int2_3157, %int40_3158, %int1_3159, %int32_3160, %int32_3161 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8658 = torch.aten.view %values_3155, %8657 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,1,32,32],f32> loc(#loc1) | |
| %int2_3162 = torch.constant.int 2 loc(#loc1) | |
| %int40_3163 = torch.constant.int 40 loc(#loc1) | |
| %int16_3164 = torch.constant.int 16 loc(#loc1) | |
| %int32_3165 = torch.constant.int 32 loc(#loc1) | |
| %int32_3166 = torch.constant.int 32 loc(#loc1) | |
| %8659 = torch.prim.ListConstruct %int2_3162, %int40_3163, %int16_3164, %int32_3165, %int32_3166 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3167 = torch.constant.bool false loc(#loc1) | |
| %8660 = torch.aten.expand %8658, %8659, %false_3167 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int0_3168 = torch.constant.int 0 loc(#loc1) | |
| %8661 = torch.aten.clone %8660, %int0_3168 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_3169 = torch.constant.int 2 loc(#loc1) | |
| %int640_3170 = torch.constant.int 640 loc(#loc1) | |
| %int32_3171 = torch.constant.int 32 loc(#loc1) | |
| %int32_3172 = torch.constant.int 32 loc(#loc1) | |
| %8662 = torch.prim.ListConstruct %int2_3169, %int640_3170, %int32_3171, %int32_3172 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8663 = torch.aten._unsafe_view %8661, %8662 : !torch.tensor<[2,40,16,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8664 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3173 = torch.constant.int 1 loc(#loc1) | |
| %8665 = torch.aten.sub.Tensor %8653, %8664, %int1_3173 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8666 = torch.aten.lift_fresh_copy %5586 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8667 = torch.aten.pow.Tensor_Tensor %8666, %8665 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8668 = torch.aten.neg %8667 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8669 = torch.aten.neg %8668 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8670 = torch.aten.div.Tensor %8663, %8669 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8671 = torch.aten.detach %5585 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8672 = torch.aten.detach %8671 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8673 = torch.aten.div.Tensor %8651, %8670 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3174 = torch.constant.int 1 loc(#loc1) | |
| %8674 = torch.aten.add.Tensor %8673, %8672, %int1_3174 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8675 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3175 = torch.constant.int 1 loc(#loc1) | |
| %8676 = torch.aten.sub.Tensor %8653, %8675, %int1_3175 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8677 = torch.aten.lift_fresh_copy %5584 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8678 = torch.aten.pow.Tensor_Tensor %8677, %8676 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8679 = torch.aten.neg %8678 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8680 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3176 = torch.constant.int 1 loc(#loc1) | |
| %8681 = torch.aten.sub.Tensor %8653, %8680, %int1_3176 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8682 = torch.aten.lift_fresh_copy %5583 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8683 = torch.aten.pow.Tensor_Tensor %8682, %8681 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8684 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3177 = torch.constant.int 1 loc(#loc1) | |
| %8685 = torch.aten.sub.Tensor %8683, %8684, %int1_3177 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8686 = torch.aten.gt.Tensor %8674, %8685 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8687 = torch.aten.where.self %8686, %8685, %8674 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8688 = torch.aten.lt.Tensor %8687, %8679 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8689 = torch.aten.where.self %8688, %8679, %8687 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8690 = torch.operator "aten.round"(%8689) : (!torch.tensor<[2,640,32,32],f32>) -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3178 = torch.constant.int 1 loc(#loc1) | |
| %8691 = torch.aten.sub.Tensor %8690, %8672, %int1_3178 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8692 = torch.aten.mul.Tensor %8691, %8670 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int640_3179 = torch.constant.int 640 loc(#loc1) | |
| %int40_3180 = torch.constant.int 40 loc(#loc1) | |
| %int16_3181 = torch.constant.int 16 loc(#loc1) | |
| %int1_3182 = torch.constant.int 1 loc(#loc1) | |
| %int1_3183 = torch.constant.int 1 loc(#loc1) | |
| %8693 = torch.prim.ListConstruct %int640_3179, %int40_3180, %int16_3181, %int1_3182, %int1_3183 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3184 = torch.constant.bool false loc(#loc1) | |
| %8694 = torch.aten.expand %5582, %8693, %false_3184 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16,1,1],f16> loc(#loc1) | |
| %int0_3185 = torch.constant.int 0 loc(#loc1) | |
| %8695 = torch.aten.clone %8694, %int0_3185 : !torch.tensor<[640,40,16,1,1],f16>, !torch.int -> !torch.tensor<[640,40,16,1,1],f16> loc(#loc1) | |
| %int640_3186 = torch.constant.int 640 loc(#loc1) | |
| %int640_3187 = torch.constant.int 640 loc(#loc1) | |
| %int1_3188 = torch.constant.int 1 loc(#loc1) | |
| %int1_3189 = torch.constant.int 1 loc(#loc1) | |
| %8696 = torch.prim.ListConstruct %int640_3186, %int640_3187, %int1_3188, %int1_3189 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8697 = torch.aten._unsafe_view %8695, %8696 : !torch.tensor<[640,40,16,1,1],f16>, !torch.list<int> -> !torch.tensor<[640,640,1,1],f16> loc(#loc1) | |
| %8698 = torch.aten.mul.Tensor %5581, %8697 : !torch.tensor, !torch.tensor<[640,640,1,1],f16> -> !torch.tensor<[640,640,1,1],f16> loc(#loc1) | |
| %int6_3190 = torch.constant.int 6 loc(#loc1) | |
| %none_3191 = torch.constant.none loc(#loc0) | |
| %none_3192 = torch.constant.none loc(#loc0) | |
| %none_3193 = torch.constant.none loc(#loc0) | |
| %false_3194 = torch.constant.bool false loc(#loc1) | |
| %none_3195 = torch.constant.none loc(#loc0) | |
| %8699 = torch.aten._to_copy %8698, %int6_3190, %none_3191, %none_3192, %none_3193, %false_3194, %none_3195 : !torch.tensor<[640,640,1,1],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640,1,1],f32> loc(#loc1) | |
| %int1_3196 = torch.constant.int 1 loc(#loc1) | |
| %int1_3197 = torch.constant.int 1 loc(#loc1) | |
| %8700 = torch.prim.ListConstruct %int1_3196, %int1_3197 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_3198 = torch.constant.int 0 loc(#loc1) | |
| %int0_3199 = torch.constant.int 0 loc(#loc1) | |
| %8701 = torch.prim.ListConstruct %int0_3198, %int0_3199 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3200 = torch.constant.int 1 loc(#loc1) | |
| %int1_3201 = torch.constant.int 1 loc(#loc1) | |
| %8702 = torch.prim.ListConstruct %int1_3200, %int1_3201 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3202 = torch.constant.bool false loc(#loc1) | |
| %int0_3203 = torch.constant.int 0 loc(#loc1) | |
| %int0_3204 = torch.constant.int 0 loc(#loc1) | |
| %8703 = torch.prim.ListConstruct %int0_3203, %int0_3204 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3205 = torch.constant.int 1 loc(#loc1) | |
| %false_3206 = torch.constant.bool false loc(#loc1) | |
| %false_3207 = torch.constant.bool false loc(#loc1) | |
| %true_3208 = torch.constant.bool true loc(#loc1) | |
| %true_3209 = torch.constant.bool true loc(#loc1) | |
| %8704 = torch.aten._convolution %8692, %8699, %5580, %8700, %8701, %8702, %false_3202, %8703, %int1_3205, %false_3206, %false_3207, %true_3208, %true_3209 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,640,1,1],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3210 = torch.constant.int 1 loc(#loc1) | |
| %8705 = torch.aten.add.Tensor %8704, %7981, %int1_3210 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int2_3211 = torch.constant.int 2 loc(#loc1) | |
| %int32_3212 = torch.constant.int 32 loc(#loc1) | |
| %int20_3213 = torch.constant.int 20 loc(#loc1) | |
| %int1024_3214 = torch.constant.int 1024 loc(#loc1) | |
| %8706 = torch.prim.ListConstruct %int2_3211, %int32_3212, %int20_3213, %int1024_3214 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8707 = torch.aten.view %8705, %8706 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_3215 = torch.constant.int 2 loc(#loc1) | |
| %int3_3216 = torch.constant.int 3 loc(#loc1) | |
| %8708 = torch.prim.ListConstruct %int2_3215, %int3_3216 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_3217 = torch.constant.int 0 loc(#loc1) | |
| %true_3218 = torch.constant.bool true loc(#loc1) | |
| %8709 = torch.aten.var.correction %8707, %8708, %int0_3217, %true_3218 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_3219 = torch.constant.int 2 loc(#loc1) | |
| %int3_3220 = torch.constant.int 3 loc(#loc1) | |
| %8710 = torch.prim.ListConstruct %int2_3219, %int3_3220 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_3221 = torch.constant.bool true loc(#loc1) | |
| %none_3222 = torch.constant.none loc(#loc0) | |
| %8711 = torch.aten.mean.dim %8707, %8710, %true_3221, %none_3222 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %8712 = torch.tensor.literal(dense<1.000000e-05> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_3223 = torch.constant.int 1 loc(#loc1) | |
| %8713 = torch.aten.add.Tensor %8709, %8712, %int1_3223 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %8714 = torch.aten.rsqrt %8713 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_3224 = torch.constant.int 1 loc(#loc1) | |
| %8715 = torch.aten.sub.Tensor %8707, %8711, %int1_3224 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %8716 = torch.aten.mul.Tensor %8715, %8714 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_3225 = torch.constant.int 2 loc(#loc1) | |
| %int640_3226 = torch.constant.int 640 loc(#loc1) | |
| %int32_3227 = torch.constant.int 32 loc(#loc1) | |
| %int32_3228 = torch.constant.int 32 loc(#loc1) | |
| %8717 = torch.prim.ListConstruct %int2_3225, %int640_3226, %int32_3227, %int32_3228 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8718 = torch.aten.view %8716, %8717 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int-1_3229 = torch.constant.int -1 loc(#loc1) | |
| %8719 = torch.aten.unsqueeze %5579, %int-1_3229 : !torch.tensor, !torch.int -> !torch.tensor<[640,1],f32> loc(#loc1) | |
| %int-1_3230 = torch.constant.int -1 loc(#loc1) | |
| %8720 = torch.aten.unsqueeze %8719, %int-1_3230 : !torch.tensor<[640,1],f32>, !torch.int -> !torch.tensor<[640,1,1],f32> loc(#loc1) | |
| %8721 = torch.aten.mul.Tensor %8718, %8720 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,1,1],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int-1_3231 = torch.constant.int -1 loc(#loc1) | |
| %8722 = torch.aten.unsqueeze %5578, %int-1_3231 : !torch.tensor, !torch.int -> !torch.tensor<[640,1],f32> loc(#loc1) | |
| %int-1_3232 = torch.constant.int -1 loc(#loc1) | |
| %8723 = torch.aten.unsqueeze %8722, %int-1_3232 : !torch.tensor<[640,1],f32>, !torch.int -> !torch.tensor<[640,1,1],f32> loc(#loc1) | |
| %int1_3233 = torch.constant.int 1 loc(#loc1) | |
| %8724 = torch.aten.add.Tensor %8721, %8723, %int1_3233 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,1,1],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8725 = torch.aten.silu %8724 : !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8726 = torch.aten.detach %5577 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8727 = torch.aten.detach %8726 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_3234 = torch.constant.int 2 loc(#loc1) | |
| %int40_3235 = torch.constant.int 40 loc(#loc1) | |
| %int16_3236 = torch.constant.int 16 loc(#loc1) | |
| %int32_3237 = torch.constant.int 32 loc(#loc1) | |
| %int32_3238 = torch.constant.int 32 loc(#loc1) | |
| %8728 = torch.prim.ListConstruct %int2_3234, %int40_3235, %int16_3236, %int32_3237, %int32_3238 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8729 = torch.aten.view %8725, %8728 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %8730 = torch.aten.abs %8729 : !torch.tensor<[2,40,16,32,32],f32> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_3239 = torch.constant.int 2 loc(#loc1) | |
| %true_3240 = torch.constant.bool true loc(#loc1) | |
| %values_3241, %indices_3242 = torch.aten.max.dim %8730, %int2_3239, %true_3240 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,40,1,32,32],f32>, !torch.tensor<[2,40,1,32,32],si64> loc(#loc1) | |
| %int2_3243 = torch.constant.int 2 loc(#loc1) | |
| %int40_3244 = torch.constant.int 40 loc(#loc1) | |
| %int1_3245 = torch.constant.int 1 loc(#loc1) | |
| %int32_3246 = torch.constant.int 32 loc(#loc1) | |
| %int32_3247 = torch.constant.int 32 loc(#loc1) | |
| %8731 = torch.prim.ListConstruct %int2_3243, %int40_3244, %int1_3245, %int32_3246, %int32_3247 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8732 = torch.aten.view %values_3241, %8731 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,1,32,32],f32> loc(#loc1) | |
| %int2_3248 = torch.constant.int 2 loc(#loc1) | |
| %int40_3249 = torch.constant.int 40 loc(#loc1) | |
| %int16_3250 = torch.constant.int 16 loc(#loc1) | |
| %int32_3251 = torch.constant.int 32 loc(#loc1) | |
| %int32_3252 = torch.constant.int 32 loc(#loc1) | |
| %8733 = torch.prim.ListConstruct %int2_3248, %int40_3249, %int16_3250, %int32_3251, %int32_3252 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3253 = torch.constant.bool false loc(#loc1) | |
| %8734 = torch.aten.expand %8732, %8733, %false_3253 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int0_3254 = torch.constant.int 0 loc(#loc1) | |
| %8735 = torch.aten.clone %8734, %int0_3254 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_3255 = torch.constant.int 2 loc(#loc1) | |
| %int640_3256 = torch.constant.int 640 loc(#loc1) | |
| %int32_3257 = torch.constant.int 32 loc(#loc1) | |
| %int32_3258 = torch.constant.int 32 loc(#loc1) | |
| %8736 = torch.prim.ListConstruct %int2_3255, %int640_3256, %int32_3257, %int32_3258 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8737 = torch.aten._unsafe_view %8735, %8736 : !torch.tensor<[2,40,16,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8738 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3259 = torch.constant.int 1 loc(#loc1) | |
| %8739 = torch.aten.sub.Tensor %8727, %8738, %int1_3259 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8740 = torch.aten.lift_fresh_copy %5576 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8741 = torch.aten.pow.Tensor_Tensor %8740, %8739 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8742 = torch.aten.neg %8741 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8743 = torch.aten.neg %8742 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8744 = torch.aten.div.Tensor %8737, %8743 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8745 = torch.aten.detach %5575 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8746 = torch.aten.detach %8745 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8747 = torch.aten.div.Tensor %8725, %8744 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3260 = torch.constant.int 1 loc(#loc1) | |
| %8748 = torch.aten.add.Tensor %8747, %8746, %int1_3260 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8749 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3261 = torch.constant.int 1 loc(#loc1) | |
| %8750 = torch.aten.sub.Tensor %8727, %8749, %int1_3261 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8751 = torch.aten.lift_fresh_copy %5574 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8752 = torch.aten.pow.Tensor_Tensor %8751, %8750 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8753 = torch.aten.neg %8752 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8754 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3262 = torch.constant.int 1 loc(#loc1) | |
| %8755 = torch.aten.sub.Tensor %8727, %8754, %int1_3262 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8756 = torch.aten.lift_fresh_copy %5573 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8757 = torch.aten.pow.Tensor_Tensor %8756, %8755 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8758 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3263 = torch.constant.int 1 loc(#loc1) | |
| %8759 = torch.aten.sub.Tensor %8757, %8758, %int1_3263 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8760 = torch.aten.gt.Tensor %8748, %8759 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8761 = torch.aten.where.self %8760, %8759, %8748 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8762 = torch.aten.lt.Tensor %8761, %8753 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8763 = torch.aten.where.self %8762, %8753, %8761 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8764 = torch.operator "aten.round"(%8763) : (!torch.tensor<[2,640,32,32],f32>) -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3264 = torch.constant.int 1 loc(#loc1) | |
| %8765 = torch.aten.sub.Tensor %8764, %8746, %int1_3264 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8766 = torch.aten.mul.Tensor %8765, %8744 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int640_3265 = torch.constant.int 640 loc(#loc1) | |
| %int40_3266 = torch.constant.int 40 loc(#loc1) | |
| %int16_3267 = torch.constant.int 16 loc(#loc1) | |
| %int3_3268 = torch.constant.int 3 loc(#loc1) | |
| %int3_3269 = torch.constant.int 3 loc(#loc1) | |
| %8767 = torch.prim.ListConstruct %int640_3265, %int40_3266, %int16_3267, %int3_3268, %int3_3269 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3270 = torch.constant.bool false loc(#loc1) | |
| %8768 = torch.aten.expand %5572, %8767, %false_3270 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16,3,3],f16> loc(#loc1) | |
| %int0_3271 = torch.constant.int 0 loc(#loc1) | |
| %8769 = torch.aten.clone %8768, %int0_3271 : !torch.tensor<[640,40,16,3,3],f16>, !torch.int -> !torch.tensor<[640,40,16,3,3],f16> loc(#loc1) | |
| %int640_3272 = torch.constant.int 640 loc(#loc1) | |
| %int640_3273 = torch.constant.int 640 loc(#loc1) | |
| %int3_3274 = torch.constant.int 3 loc(#loc1) | |
| %int3_3275 = torch.constant.int 3 loc(#loc1) | |
| %8770 = torch.prim.ListConstruct %int640_3272, %int640_3273, %int3_3274, %int3_3275 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8771 = torch.aten._unsafe_view %8769, %8770 : !torch.tensor<[640,40,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[640,640,3,3],f16> loc(#loc1) | |
| %8772 = torch.aten.mul.Tensor %5571, %8771 : !torch.tensor, !torch.tensor<[640,640,3,3],f16> -> !torch.tensor<[640,640,3,3],f16> loc(#loc1) | |
| %int6_3276 = torch.constant.int 6 loc(#loc1) | |
| %none_3277 = torch.constant.none loc(#loc0) | |
| %none_3278 = torch.constant.none loc(#loc0) | |
| %none_3279 = torch.constant.none loc(#loc0) | |
| %false_3280 = torch.constant.bool false loc(#loc1) | |
| %none_3281 = torch.constant.none loc(#loc0) | |
| %8773 = torch.aten._to_copy %8772, %int6_3276, %none_3277, %none_3278, %none_3279, %false_3280, %none_3281 : !torch.tensor<[640,640,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640,3,3],f32> loc(#loc1) | |
| %int1_3282 = torch.constant.int 1 loc(#loc1) | |
| %int1_3283 = torch.constant.int 1 loc(#loc1) | |
| %8774 = torch.prim.ListConstruct %int1_3282, %int1_3283 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3284 = torch.constant.int 1 loc(#loc1) | |
| %int1_3285 = torch.constant.int 1 loc(#loc1) | |
| %8775 = torch.prim.ListConstruct %int1_3284, %int1_3285 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3286 = torch.constant.int 1 loc(#loc1) | |
| %int1_3287 = torch.constant.int 1 loc(#loc1) | |
| %8776 = torch.prim.ListConstruct %int1_3286, %int1_3287 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3288 = torch.constant.bool false loc(#loc1) | |
| %int0_3289 = torch.constant.int 0 loc(#loc1) | |
| %int0_3290 = torch.constant.int 0 loc(#loc1) | |
| %8777 = torch.prim.ListConstruct %int0_3289, %int0_3290 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3291 = torch.constant.int 1 loc(#loc1) | |
| %false_3292 = torch.constant.bool false loc(#loc1) | |
| %false_3293 = torch.constant.bool false loc(#loc1) | |
| %true_3294 = torch.constant.bool true loc(#loc1) | |
| %true_3295 = torch.constant.bool true loc(#loc1) | |
| %8778 = torch.aten._convolution %8766, %8773, %5570, %8774, %8775, %8776, %false_3288, %8777, %int1_3291, %false_3292, %false_3293, %true_3294, %true_3295 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,640,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8779 = torch.aten.silu %5951 : !torch.tensor<[2,1280],f32> -> !torch.tensor<[2,1280],f32> loc(#loc1) | |
| %8780 = torch.aten.t %5569 : !torch.tensor -> !torch.tensor<[1280,640],f32> loc(#loc1) | |
| %int1_3296 = torch.constant.int 1 loc(#loc1) | |
| %int1_3297 = torch.constant.int 1 loc(#loc1) | |
| %8781 = torch.aten.addmm %5568, %8779, %8780, %int1_3296, %int1_3297 : !torch.tensor, !torch.tensor<[2,1280],f32>, !torch.tensor<[1280,640],f32>, !torch.int, !torch.int -> !torch.tensor<[2,640],f32> loc(#loc1) | |
| %int0_3298 = torch.constant.int 0 loc(#loc1) | |
| %int0_3299 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_3300 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_3301 = torch.constant.int 1 loc(#loc1) | |
| %8782 = torch.aten.slice.Tensor %8781, %int0_3298, %int0_3299, %int9223372036854775807_3300, %int1_3301 : !torch.tensor<[2,640],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,640],f32> loc(#loc1) | |
| %int1_3302 = torch.constant.int 1 loc(#loc1) | |
| %int0_3303 = torch.constant.int 0 loc(#loc1) | |
| %int9223372036854775807_3304 = torch.constant.int 9223372036854775807 loc(#loc1) | |
| %int1_3305 = torch.constant.int 1 loc(#loc1) | |
| %8783 = torch.aten.slice.Tensor %8782, %int1_3302, %int0_3303, %int9223372036854775807_3304, %int1_3305 : !torch.tensor<[2,640],f32>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.tensor<[2,640],f32> loc(#loc1) | |
| %int2_3306 = torch.constant.int 2 loc(#loc1) | |
| %8784 = torch.aten.unsqueeze %8783, %int2_3306 : !torch.tensor<[2,640],f32>, !torch.int -> !torch.tensor<[2,640,1],f32> loc(#loc1) | |
| %int3_3307 = torch.constant.int 3 loc(#loc1) | |
| %8785 = torch.aten.unsqueeze %8784, %int3_3307 : !torch.tensor<[2,640,1],f32>, !torch.int -> !torch.tensor<[2,640,1,1],f32> loc(#loc1) | |
| %int1_3308 = torch.constant.int 1 loc(#loc1) | |
| %8786 = torch.aten.add.Tensor %8778, %8785, %int1_3308 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,1,1],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int2_3309 = torch.constant.int 2 loc(#loc1) | |
| %int32_3310 = torch.constant.int 32 loc(#loc1) | |
| %int20_3311 = torch.constant.int 20 loc(#loc1) | |
| %int1024_3312 = torch.constant.int 1024 loc(#loc1) | |
| %8787 = torch.prim.ListConstruct %int2_3309, %int32_3310, %int20_3311, %int1024_3312 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8788 = torch.aten.view %8786, %8787 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_3313 = torch.constant.int 2 loc(#loc1) | |
| %int3_3314 = torch.constant.int 3 loc(#loc1) | |
| %8789 = torch.prim.ListConstruct %int2_3313, %int3_3314 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_3315 = torch.constant.int 0 loc(#loc1) | |
| %true_3316 = torch.constant.bool true loc(#loc1) | |
| %8790 = torch.aten.var.correction %8788, %8789, %int0_3315, %true_3316 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_3317 = torch.constant.int 2 loc(#loc1) | |
| %int3_3318 = torch.constant.int 3 loc(#loc1) | |
| %8791 = torch.prim.ListConstruct %int2_3317, %int3_3318 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_3319 = torch.constant.bool true loc(#loc1) | |
| %none_3320 = torch.constant.none loc(#loc0) | |
| %8792 = torch.aten.mean.dim %8788, %8791, %true_3319, %none_3320 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %8793 = torch.tensor.literal(dense<1.000000e-05> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_3321 = torch.constant.int 1 loc(#loc1) | |
| %8794 = torch.aten.add.Tensor %8790, %8793, %int1_3321 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %8795 = torch.aten.rsqrt %8794 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_3322 = torch.constant.int 1 loc(#loc1) | |
| %8796 = torch.aten.sub.Tensor %8788, %8792, %int1_3322 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %8797 = torch.aten.mul.Tensor %8796, %8795 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_3323 = torch.constant.int 2 loc(#loc1) | |
| %int640_3324 = torch.constant.int 640 loc(#loc1) | |
| %int32_3325 = torch.constant.int 32 loc(#loc1) | |
| %int32_3326 = torch.constant.int 32 loc(#loc1) | |
| %8798 = torch.prim.ListConstruct %int2_3323, %int640_3324, %int32_3325, %int32_3326 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8799 = torch.aten.view %8797, %8798 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int-1_3327 = torch.constant.int -1 loc(#loc1) | |
| %8800 = torch.aten.unsqueeze %5567, %int-1_3327 : !torch.tensor, !torch.int -> !torch.tensor<[640,1],f32> loc(#loc1) | |
| %int-1_3328 = torch.constant.int -1 loc(#loc1) | |
| %8801 = torch.aten.unsqueeze %8800, %int-1_3328 : !torch.tensor<[640,1],f32>, !torch.int -> !torch.tensor<[640,1,1],f32> loc(#loc1) | |
| %8802 = torch.aten.mul.Tensor %8799, %8801 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,1,1],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int-1_3329 = torch.constant.int -1 loc(#loc1) | |
| %8803 = torch.aten.unsqueeze %5566, %int-1_3329 : !torch.tensor, !torch.int -> !torch.tensor<[640,1],f32> loc(#loc1) | |
| %int-1_3330 = torch.constant.int -1 loc(#loc1) | |
| %8804 = torch.aten.unsqueeze %8803, %int-1_3330 : !torch.tensor<[640,1],f32>, !torch.int -> !torch.tensor<[640,1,1],f32> loc(#loc1) | |
| %int1_3331 = torch.constant.int 1 loc(#loc1) | |
| %8805 = torch.aten.add.Tensor %8802, %8804, %int1_3331 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,1,1],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8806 = torch.aten.silu %8805 : !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8807 = torch.aten.detach %5565 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8808 = torch.aten.detach %8807 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_3332 = torch.constant.int 2 loc(#loc1) | |
| %int40_3333 = torch.constant.int 40 loc(#loc1) | |
| %int16_3334 = torch.constant.int 16 loc(#loc1) | |
| %int32_3335 = torch.constant.int 32 loc(#loc1) | |
| %int32_3336 = torch.constant.int 32 loc(#loc1) | |
| %8809 = torch.prim.ListConstruct %int2_3332, %int40_3333, %int16_3334, %int32_3335, %int32_3336 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8810 = torch.aten.view %8806, %8809 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %8811 = torch.aten.abs %8810 : !torch.tensor<[2,40,16,32,32],f32> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_3337 = torch.constant.int 2 loc(#loc1) | |
| %true_3338 = torch.constant.bool true loc(#loc1) | |
| %values_3339, %indices_3340 = torch.aten.max.dim %8811, %int2_3337, %true_3338 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,40,1,32,32],f32>, !torch.tensor<[2,40,1,32,32],si64> loc(#loc1) | |
| %int2_3341 = torch.constant.int 2 loc(#loc1) | |
| %int40_3342 = torch.constant.int 40 loc(#loc1) | |
| %int1_3343 = torch.constant.int 1 loc(#loc1) | |
| %int32_3344 = torch.constant.int 32 loc(#loc1) | |
| %int32_3345 = torch.constant.int 32 loc(#loc1) | |
| %8812 = torch.prim.ListConstruct %int2_3341, %int40_3342, %int1_3343, %int32_3344, %int32_3345 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8813 = torch.aten.view %values_3339, %8812 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,1,32,32],f32> loc(#loc1) | |
| %int2_3346 = torch.constant.int 2 loc(#loc1) | |
| %int40_3347 = torch.constant.int 40 loc(#loc1) | |
| %int16_3348 = torch.constant.int 16 loc(#loc1) | |
| %int32_3349 = torch.constant.int 32 loc(#loc1) | |
| %int32_3350 = torch.constant.int 32 loc(#loc1) | |
| %8814 = torch.prim.ListConstruct %int2_3346, %int40_3347, %int16_3348, %int32_3349, %int32_3350 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3351 = torch.constant.bool false loc(#loc1) | |
| %8815 = torch.aten.expand %8813, %8814, %false_3351 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int0_3352 = torch.constant.int 0 loc(#loc1) | |
| %8816 = torch.aten.clone %8815, %int0_3352 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_3353 = torch.constant.int 2 loc(#loc1) | |
| %int640_3354 = torch.constant.int 640 loc(#loc1) | |
| %int32_3355 = torch.constant.int 32 loc(#loc1) | |
| %int32_3356 = torch.constant.int 32 loc(#loc1) | |
| %8817 = torch.prim.ListConstruct %int2_3353, %int640_3354, %int32_3355, %int32_3356 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8818 = torch.aten._unsafe_view %8816, %8817 : !torch.tensor<[2,40,16,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8819 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3357 = torch.constant.int 1 loc(#loc1) | |
| %8820 = torch.aten.sub.Tensor %8808, %8819, %int1_3357 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8821 = torch.aten.lift_fresh_copy %5564 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8822 = torch.aten.pow.Tensor_Tensor %8821, %8820 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8823 = torch.aten.neg %8822 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8824 = torch.aten.neg %8823 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8825 = torch.aten.div.Tensor %8818, %8824 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8826 = torch.aten.detach %5563 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8827 = torch.aten.detach %8826 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8828 = torch.aten.div.Tensor %8806, %8825 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3358 = torch.constant.int 1 loc(#loc1) | |
| %8829 = torch.aten.add.Tensor %8828, %8827, %int1_3358 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8830 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3359 = torch.constant.int 1 loc(#loc1) | |
| %8831 = torch.aten.sub.Tensor %8808, %8830, %int1_3359 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8832 = torch.aten.lift_fresh_copy %5562 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8833 = torch.aten.pow.Tensor_Tensor %8832, %8831 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8834 = torch.aten.neg %8833 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8835 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3360 = torch.constant.int 1 loc(#loc1) | |
| %8836 = torch.aten.sub.Tensor %8808, %8835, %int1_3360 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8837 = torch.aten.lift_fresh_copy %5561 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8838 = torch.aten.pow.Tensor_Tensor %8837, %8836 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8839 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3361 = torch.constant.int 1 loc(#loc1) | |
| %8840 = torch.aten.sub.Tensor %8838, %8839, %int1_3361 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8841 = torch.aten.gt.Tensor %8829, %8840 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8842 = torch.aten.where.self %8841, %8840, %8829 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8843 = torch.aten.lt.Tensor %8842, %8834 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8844 = torch.aten.where.self %8843, %8834, %8842 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8845 = torch.operator "aten.round"(%8844) : (!torch.tensor<[2,640,32,32],f32>) -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3362 = torch.constant.int 1 loc(#loc1) | |
| %8846 = torch.aten.sub.Tensor %8845, %8827, %int1_3362 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8847 = torch.aten.mul.Tensor %8846, %8825 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int640_3363 = torch.constant.int 640 loc(#loc1) | |
| %int40_3364 = torch.constant.int 40 loc(#loc1) | |
| %int16_3365 = torch.constant.int 16 loc(#loc1) | |
| %int3_3366 = torch.constant.int 3 loc(#loc1) | |
| %int3_3367 = torch.constant.int 3 loc(#loc1) | |
| %8848 = torch.prim.ListConstruct %int640_3363, %int40_3364, %int16_3365, %int3_3366, %int3_3367 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3368 = torch.constant.bool false loc(#loc1) | |
| %8849 = torch.aten.expand %5560, %8848, %false_3368 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16,3,3],f16> loc(#loc1) | |
| %int0_3369 = torch.constant.int 0 loc(#loc1) | |
| %8850 = torch.aten.clone %8849, %int0_3369 : !torch.tensor<[640,40,16,3,3],f16>, !torch.int -> !torch.tensor<[640,40,16,3,3],f16> loc(#loc1) | |
| %int640_3370 = torch.constant.int 640 loc(#loc1) | |
| %int640_3371 = torch.constant.int 640 loc(#loc1) | |
| %int3_3372 = torch.constant.int 3 loc(#loc1) | |
| %int3_3373 = torch.constant.int 3 loc(#loc1) | |
| %8851 = torch.prim.ListConstruct %int640_3370, %int640_3371, %int3_3372, %int3_3373 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8852 = torch.aten._unsafe_view %8850, %8851 : !torch.tensor<[640,40,16,3,3],f16>, !torch.list<int> -> !torch.tensor<[640,640,3,3],f16> loc(#loc1) | |
| %8853 = torch.aten.mul.Tensor %5559, %8852 : !torch.tensor, !torch.tensor<[640,640,3,3],f16> -> !torch.tensor<[640,640,3,3],f16> loc(#loc1) | |
| %int6_3374 = torch.constant.int 6 loc(#loc1) | |
| %none_3375 = torch.constant.none loc(#loc0) | |
| %none_3376 = torch.constant.none loc(#loc0) | |
| %none_3377 = torch.constant.none loc(#loc0) | |
| %false_3378 = torch.constant.bool false loc(#loc1) | |
| %none_3379 = torch.constant.none loc(#loc0) | |
| %8854 = torch.aten._to_copy %8853, %int6_3374, %none_3375, %none_3376, %none_3377, %false_3378, %none_3379 : !torch.tensor<[640,640,3,3],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640,3,3],f32> loc(#loc1) | |
| %int1_3380 = torch.constant.int 1 loc(#loc1) | |
| %int1_3381 = torch.constant.int 1 loc(#loc1) | |
| %8855 = torch.prim.ListConstruct %int1_3380, %int1_3381 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3382 = torch.constant.int 1 loc(#loc1) | |
| %int1_3383 = torch.constant.int 1 loc(#loc1) | |
| %8856 = torch.prim.ListConstruct %int1_3382, %int1_3383 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3384 = torch.constant.int 1 loc(#loc1) | |
| %int1_3385 = torch.constant.int 1 loc(#loc1) | |
| %8857 = torch.prim.ListConstruct %int1_3384, %int1_3385 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3386 = torch.constant.bool false loc(#loc1) | |
| %int0_3387 = torch.constant.int 0 loc(#loc1) | |
| %int0_3388 = torch.constant.int 0 loc(#loc1) | |
| %8858 = torch.prim.ListConstruct %int0_3387, %int0_3388 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3389 = torch.constant.int 1 loc(#loc1) | |
| %false_3390 = torch.constant.bool false loc(#loc1) | |
| %false_3391 = torch.constant.bool false loc(#loc1) | |
| %true_3392 = torch.constant.bool true loc(#loc1) | |
| %true_3393 = torch.constant.bool true loc(#loc1) | |
| %8859 = torch.aten._convolution %8847, %8854, %5558, %8855, %8856, %8857, %false_3386, %8858, %int1_3389, %false_3390, %false_3391, %true_3392, %true_3393 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,640,3,3],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3394 = torch.constant.int 1 loc(#loc1) | |
| %8860 = torch.aten.add.Tensor %8705, %8859, %int1_3394 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8861 = torch.tensor.literal(dense<1.000000e+00> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %8862 = torch.aten.div.Tensor %8860, %8861 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f64> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int2_3395 = torch.constant.int 2 loc(#loc1) | |
| %int32_3396 = torch.constant.int 32 loc(#loc1) | |
| %int20_3397 = torch.constant.int 20 loc(#loc1) | |
| %int1024_3398 = torch.constant.int 1024 loc(#loc1) | |
| %8863 = torch.prim.ListConstruct %int2_3395, %int32_3396, %int20_3397, %int1024_3398 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8864 = torch.aten.view %8862, %8863 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_3399 = torch.constant.int 2 loc(#loc1) | |
| %int3_3400 = torch.constant.int 3 loc(#loc1) | |
| %8865 = torch.prim.ListConstruct %int2_3399, %int3_3400 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_3401 = torch.constant.int 0 loc(#loc1) | |
| %true_3402 = torch.constant.bool true loc(#loc1) | |
| %8866 = torch.aten.var.correction %8864, %8865, %int0_3401, %true_3402 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.int, !torch.bool -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int2_3403 = torch.constant.int 2 loc(#loc1) | |
| %int3_3404 = torch.constant.int 3 loc(#loc1) | |
| %8867 = torch.prim.ListConstruct %int2_3403, %int3_3404 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %true_3405 = torch.constant.bool true loc(#loc1) | |
| %none_3406 = torch.constant.none loc(#loc0) | |
| %8868 = torch.aten.mean.dim %8864, %8867, %true_3405, %none_3406 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int>, !torch.bool, !torch.none -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %8869 = torch.tensor.literal(dense<9.9999999999999995E-7> : tensor<f64>) : !torch.tensor<[],f64> loc(#loc1) | |
| %int1_3407 = torch.constant.int 1 loc(#loc1) | |
| %8870 = torch.aten.add.Tensor %8866, %8869, %int1_3407 : !torch.tensor<[2,32,1,1],f32>, !torch.tensor<[],f64>, !torch.int -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %8871 = torch.aten.rsqrt %8870 : !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,1,1],f32> loc(#loc1) | |
| %int1_3408 = torch.constant.int 1 loc(#loc1) | |
| %8872 = torch.aten.sub.Tensor %8864, %8868, %int1_3408 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32>, !torch.int -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %8873 = torch.aten.mul.Tensor %8872, %8871 : !torch.tensor<[2,32,20,1024],f32>, !torch.tensor<[2,32,1,1],f32> -> !torch.tensor<[2,32,20,1024],f32> loc(#loc1) | |
| %int2_3409 = torch.constant.int 2 loc(#loc1) | |
| %int640_3410 = torch.constant.int 640 loc(#loc1) | |
| %int32_3411 = torch.constant.int 32 loc(#loc1) | |
| %int32_3412 = torch.constant.int 32 loc(#loc1) | |
| %8874 = torch.prim.ListConstruct %int2_3409, %int640_3410, %int32_3411, %int32_3412 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8875 = torch.aten.view %8873, %8874 : !torch.tensor<[2,32,20,1024],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8876 = torch.aten.detach %5557 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8877 = torch.aten.detach %8876 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %int2_3413 = torch.constant.int 2 loc(#loc1) | |
| %int40_3414 = torch.constant.int 40 loc(#loc1) | |
| %int16_3415 = torch.constant.int 16 loc(#loc1) | |
| %int32_3416 = torch.constant.int 32 loc(#loc1) | |
| %int32_3417 = torch.constant.int 32 loc(#loc1) | |
| %8878 = torch.prim.ListConstruct %int2_3413, %int40_3414, %int16_3415, %int32_3416, %int32_3417 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8879 = torch.aten.view %8875, %8878 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %8880 = torch.aten.abs %8879 : !torch.tensor<[2,40,16,32,32],f32> -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_3418 = torch.constant.int 2 loc(#loc1) | |
| %true_3419 = torch.constant.bool true loc(#loc1) | |
| %values_3420, %indices_3421 = torch.aten.max.dim %8880, %int2_3418, %true_3419 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int, !torch.bool -> !torch.tensor<[2,40,1,32,32],f32>, !torch.tensor<[2,40,1,32,32],si64> loc(#loc1) | |
| %int2_3422 = torch.constant.int 2 loc(#loc1) | |
| %int40_3423 = torch.constant.int 40 loc(#loc1) | |
| %int1_3424 = torch.constant.int 1 loc(#loc1) | |
| %int32_3425 = torch.constant.int 32 loc(#loc1) | |
| %int32_3426 = torch.constant.int 32 loc(#loc1) | |
| %8881 = torch.prim.ListConstruct %int2_3422, %int40_3423, %int1_3424, %int32_3425, %int32_3426 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8882 = torch.aten.view %values_3420, %8881 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,40,1,32,32],f32> loc(#loc1) | |
| %int2_3427 = torch.constant.int 2 loc(#loc1) | |
| %int40_3428 = torch.constant.int 40 loc(#loc1) | |
| %int16_3429 = torch.constant.int 16 loc(#loc1) | |
| %int32_3430 = torch.constant.int 32 loc(#loc1) | |
| %int32_3431 = torch.constant.int 32 loc(#loc1) | |
| %8883 = torch.prim.ListConstruct %int2_3427, %int40_3428, %int16_3429, %int32_3430, %int32_3431 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3432 = torch.constant.bool false loc(#loc1) | |
| %8884 = torch.aten.expand %8882, %8883, %false_3432 : !torch.tensor<[2,40,1,32,32],f32>, !torch.list<int>, !torch.bool -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int0_3433 = torch.constant.int 0 loc(#loc1) | |
| %8885 = torch.aten.clone %8884, %int0_3433 : !torch.tensor<[2,40,16,32,32],f32>, !torch.int -> !torch.tensor<[2,40,16,32,32],f32> loc(#loc1) | |
| %int2_3434 = torch.constant.int 2 loc(#loc1) | |
| %int640_3435 = torch.constant.int 640 loc(#loc1) | |
| %int32_3436 = torch.constant.int 32 loc(#loc1) | |
| %int32_3437 = torch.constant.int 32 loc(#loc1) | |
| %8886 = torch.prim.ListConstruct %int2_3434, %int640_3435, %int32_3436, %int32_3437 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8887 = torch.aten._unsafe_view %8885, %8886 : !torch.tensor<[2,40,16,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8888 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3438 = torch.constant.int 1 loc(#loc1) | |
| %8889 = torch.aten.sub.Tensor %8877, %8888, %int1_3438 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8890 = torch.aten.lift_fresh_copy %5556 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8891 = torch.aten.pow.Tensor_Tensor %8890, %8889 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8892 = torch.aten.neg %8891 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8893 = torch.aten.neg %8892 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8894 = torch.aten.div.Tensor %8887, %8893 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8895 = torch.aten.detach %5555 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8896 = torch.aten.detach %8895 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8897 = torch.aten.div.Tensor %8875, %8894 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3439 = torch.constant.int 1 loc(#loc1) | |
| %8898 = torch.aten.add.Tensor %8897, %8896, %int1_3439 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8899 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3440 = torch.constant.int 1 loc(#loc1) | |
| %8900 = torch.aten.sub.Tensor %8877, %8899, %int1_3440 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8901 = torch.aten.lift_fresh_copy %5554 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8902 = torch.aten.pow.Tensor_Tensor %8901, %8900 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8903 = torch.aten.neg %8902 : !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8904 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3441 = torch.constant.int 1 loc(#loc1) | |
| %8905 = torch.aten.sub.Tensor %8877, %8904, %int1_3441 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8906 = torch.aten.lift_fresh_copy %5553 : !torch.tensor -> !torch.tensor<[],f32> loc(#loc1) | |
| %8907 = torch.aten.pow.Tensor_Tensor %8906, %8905 : !torch.tensor<[],f32>, !torch.tensor<[],f32> -> !torch.tensor<[],f32> loc(#loc1) | |
| %8908 = torch.tensor.literal(dense<1> : tensor<si64>) : !torch.tensor<[],si64> loc(#loc1) | |
| %int1_3442 = torch.constant.int 1 loc(#loc1) | |
| %8909 = torch.aten.sub.Tensor %8907, %8908, %int1_3442 : !torch.tensor<[],f32>, !torch.tensor<[],si64>, !torch.int -> !torch.tensor<[],f32> loc(#loc1) | |
| %8910 = torch.aten.gt.Tensor %8898, %8909 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8911 = torch.aten.where.self %8910, %8909, %8898 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8912 = torch.aten.lt.Tensor %8911, %8903 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32> -> !torch.tensor<[2,640,32,32],i1> loc(#loc1) | |
| %8913 = torch.aten.where.self %8912, %8903, %8911 : !torch.tensor<[2,640,32,32],i1>, !torch.tensor<[],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8914 = torch.operator "aten.round"(%8913) : (!torch.tensor<[2,640,32,32],f32>) -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int1_3443 = torch.constant.int 1 loc(#loc1) | |
| %8915 = torch.aten.sub.Tensor %8914, %8896, %int1_3443 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[],f32>, !torch.int -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %8916 = torch.aten.mul.Tensor %8915, %8894 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[2,640,32,32],f32> -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int640_3444 = torch.constant.int 640 loc(#loc1) | |
| %int40_3445 = torch.constant.int 40 loc(#loc1) | |
| %int16_3446 = torch.constant.int 16 loc(#loc1) | |
| %int1_3447 = torch.constant.int 1 loc(#loc1) | |
| %int1_3448 = torch.constant.int 1 loc(#loc1) | |
| %8917 = torch.prim.ListConstruct %int640_3444, %int40_3445, %int16_3446, %int1_3447, %int1_3448 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3449 = torch.constant.bool false loc(#loc1) | |
| %8918 = torch.aten.expand %5552, %8917, %false_3449 : !torch.tensor, !torch.list<int>, !torch.bool -> !torch.tensor<[640,40,16,1,1],f16> loc(#loc1) | |
| %int0_3450 = torch.constant.int 0 loc(#loc1) | |
| %8919 = torch.aten.clone %8918, %int0_3450 : !torch.tensor<[640,40,16,1,1],f16>, !torch.int -> !torch.tensor<[640,40,16,1,1],f16> loc(#loc1) | |
| %int640_3451 = torch.constant.int 640 loc(#loc1) | |
| %int640_3452 = torch.constant.int 640 loc(#loc1) | |
| %int1_3453 = torch.constant.int 1 loc(#loc1) | |
| %int1_3454 = torch.constant.int 1 loc(#loc1) | |
| %8920 = torch.prim.ListConstruct %int640_3451, %int640_3452, %int1_3453, %int1_3454 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8921 = torch.aten._unsafe_view %8919, %8920 : !torch.tensor<[640,40,16,1,1],f16>, !torch.list<int> -> !torch.tensor<[640,640,1,1],f16> loc(#loc1) | |
| %8922 = torch.aten.mul.Tensor %5551, %8921 : !torch.tensor, !torch.tensor<[640,640,1,1],f16> -> !torch.tensor<[640,640,1,1],f16> loc(#loc1) | |
| %int6_3455 = torch.constant.int 6 loc(#loc1) | |
| %none_3456 = torch.constant.none loc(#loc0) | |
| %none_3457 = torch.constant.none loc(#loc0) | |
| %none_3458 = torch.constant.none loc(#loc0) | |
| %false_3459 = torch.constant.bool false loc(#loc1) | |
| %none_3460 = torch.constant.none loc(#loc0) | |
| %8923 = torch.aten._to_copy %8922, %int6_3455, %none_3456, %none_3457, %none_3458, %false_3459, %none_3460 : !torch.tensor<[640,640,1,1],f16>, !torch.int, !torch.none, !torch.none, !torch.none, !torch.bool, !torch.none -> !torch.tensor<[640,640,1,1],f32> loc(#loc1) | |
| %int1_3461 = torch.constant.int 1 loc(#loc1) | |
| %int1_3462 = torch.constant.int 1 loc(#loc1) | |
| %8924 = torch.prim.ListConstruct %int1_3461, %int1_3462 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int0_3463 = torch.constant.int 0 loc(#loc1) | |
| %int0_3464 = torch.constant.int 0 loc(#loc1) | |
| %8925 = torch.prim.ListConstruct %int0_3463, %int0_3464 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3465 = torch.constant.int 1 loc(#loc1) | |
| %int1_3466 = torch.constant.int 1 loc(#loc1) | |
| %8926 = torch.prim.ListConstruct %int1_3465, %int1_3466 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %false_3467 = torch.constant.bool false loc(#loc1) | |
| %int0_3468 = torch.constant.int 0 loc(#loc1) | |
| %int0_3469 = torch.constant.int 0 loc(#loc1) | |
| %8927 = torch.prim.ListConstruct %int0_3468, %int0_3469 : (!torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %int1_3470 = torch.constant.int 1 loc(#loc1) | |
| %false_3471 = torch.constant.bool false loc(#loc1) | |
| %false_3472 = torch.constant.bool false loc(#loc1) | |
| %true_3473 = torch.constant.bool true loc(#loc1) | |
| %true_3474 = torch.constant.bool true loc(#loc1) | |
| %8928 = torch.aten._convolution %8916, %8923, %5550, %8924, %8925, %8926, %false_3467, %8927, %int1_3470, %false_3471, %false_3472, %true_3473, %true_3474 : !torch.tensor<[2,640,32,32],f32>, !torch.tensor<[640,640,1,1],f32>, !torch.tensor, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int, !torch.bool, !torch.bool, !torch.bool, !torch.bool -> !torch.tensor<[2,640,32,32],f32> loc(#loc1) | |
| %int0_3475 = torch.constant.int 0 loc(#loc1) | |
| %int2_3476 = torch.constant.int 2 loc(#loc1) | |
| %int3_3477 = torch.constant.int 3 loc(#loc1) | |
| %int1_3478 = torch.constant.int 1 loc(#loc1) | |
| %8929 = torch.prim.ListConstruct %int0_3475, %int2_3476, %int3_3477, %int1_3478 : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> loc(#loc0) | |
| %8930 = torch.aten.permute %8928, %8929 : !torch.tensor<[2,640,32,32],f32>, !torch.list<int> -> !torch.tensor<[2,32,32,640],f32> loc(#loc1) | |
| %int2_3479 = torch.constant.int 2 loc(#loc1) | |
| %int1024_3480 = torch.constant.int 1024 loc(#loc1) | |
| %int640_3481 = torch.constant.int 640 loc(#loc1) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment