Created
April 22, 2024 16:51
-
-
Save AmosLewis/9a65b077fe121ae96ade0fd094a28242 to your computer and use it in GitHub Desktop.
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
module { | |
func.func @torch_jit(%arg0: !torch.vtensor<[1,1,64,128,128],f32>) -> !torch.vtensor<[1,1,64,128,128],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "pytorch", torch.onnx_meta.producer_version = "1.13.1"} { | |
%0 = torch.vtensor.literal(dense<8.906250e-01> : tensor<1xf32>) : !torch.vtensor<[1],f32> | |
%1 = torch.vtensor.literal(dense<-0.0849609375> : tensor<1xf32>) : !torch.vtensor<[1],f32> | |
%2 = torch.vtensor.literal(dense_resource<__elided__> : tensor<16x1x3x3x3xf32>) : !torch.vtensor<[16,1,3,3,3],f32> | |
%3 = torch.vtensor.literal(dense_resource<__elided__> : tensor<16xf32>) : !torch.vtensor<[16],f32> | |
%4 = torch.vtensor.literal(dense_resource<__elided__> : tensor<16xf32>) : !torch.vtensor<[16],f32> | |
%5 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32x16x3x3x3xf32>) : !torch.vtensor<[32,16,3,3,3],f32> | |
%6 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32xf32>) : !torch.vtensor<[32],f32> | |
%7 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32xf32>) : !torch.vtensor<[32],f32> | |
%8 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32x32x3x3x3xf32>) : !torch.vtensor<[32,32,3,3,3],f32> | |
%9 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32xf32>) : !torch.vtensor<[32],f32> | |
%10 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32xf32>) : !torch.vtensor<[32],f32> | |
%11 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64x32x3x3x3xf32>) : !torch.vtensor<[64,32,3,3,3],f32> | |
%12 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64xf32>) : !torch.vtensor<[64],f32> | |
%13 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64xf32>) : !torch.vtensor<[64],f32> | |
%14 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64x64x3x3x3xf32>) : !torch.vtensor<[64,64,3,3,3],f32> | |
%15 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64xf32>) : !torch.vtensor<[64],f32> | |
%16 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64xf32>) : !torch.vtensor<[64],f32> | |
%17 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128x64x3x3x3xf32>) : !torch.vtensor<[128,64,3,3,3],f32> | |
%18 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128xf32>) : !torch.vtensor<[128],f32> | |
%19 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128xf32>) : !torch.vtensor<[128],f32> | |
%20 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128x128x3x3x3xf32>) : !torch.vtensor<[128,128,3,3,3],f32> | |
%21 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128xf32>) : !torch.vtensor<[128],f32> | |
%22 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128xf32>) : !torch.vtensor<[128],f32> | |
%23 = torch.vtensor.literal(dense_resource<__elided__> : tensor<256x128x3x3x3xf32>) : !torch.vtensor<[256,128,3,3,3],f32> | |
%24 = torch.vtensor.literal(dense_resource<__elided__> : tensor<384xf32>) : !torch.vtensor<[384],f32> | |
%25 = torch.vtensor.literal(dense_resource<__elided__> : tensor<384xf32>) : !torch.vtensor<[384],f32> | |
%26 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128x384x3x3x3xf32>) : !torch.vtensor<[128,384,3,3,3],f32> | |
%27 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128xf32>) : !torch.vtensor<[128],f32> | |
%28 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128xf32>) : !torch.vtensor<[128],f32> | |
%29 = torch.vtensor.literal(dense_resource<__elided__> : tensor<128x128x3x3x3xf32>) : !torch.vtensor<[128,128,3,3,3],f32> | |
%30 = torch.vtensor.literal(dense_resource<__elided__> : tensor<192xf32>) : !torch.vtensor<[192],f32> | |
%31 = torch.vtensor.literal(dense_resource<__elided__> : tensor<192xf32>) : !torch.vtensor<[192],f32> | |
%32 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64x192x3x3x3xf32>) : !torch.vtensor<[64,192,3,3,3],f32> | |
%33 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64xf32>) : !torch.vtensor<[64],f32> | |
%34 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64xf32>) : !torch.vtensor<[64],f32> | |
%35 = torch.vtensor.literal(dense_resource<__elided__> : tensor<64x64x3x3x3xf32>) : !torch.vtensor<[64,64,3,3,3],f32> | |
%36 = torch.vtensor.literal(dense_resource<__elided__> : tensor<96xf32>) : !torch.vtensor<[96],f32> | |
%37 = torch.vtensor.literal(dense_resource<__elided__> : tensor<96xf32>) : !torch.vtensor<[96],f32> | |
%38 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32x96x3x3x3xf32>) : !torch.vtensor<[32,96,3,3,3],f32> | |
%39 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32xf32>) : !torch.vtensor<[32],f32> | |
%40 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32xf32>) : !torch.vtensor<[32],f32> | |
%41 = torch.vtensor.literal(dense_resource<__elided__> : tensor<32x32x3x3x3xf32>) : !torch.vtensor<[32,32,3,3,3],f32> | |
%42 = torch.vtensor.literal(dense_resource<__elided__> : tensor<1x32x1x1x1xf32>) : !torch.vtensor<[1,32,1,1,1],f32> | |
%43 = torch.vtensor.literal(dense<-0.142578125> : tensor<1xf32>) : !torch.vtensor<[1],f32> | |
%none = torch.constant.none | |
%44 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%45 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12 = torch.constant.int 12 | |
%46 = torch.aten.item %44 : !torch.vtensor<[],f32> -> !torch.float | |
%47 = torch.aten.item %45 : !torch.vtensor<[],si8> -> !torch.int | |
%48 = torch.aten.quantize_per_tensor %arg0, %46, %47, %int12 : !torch.vtensor<[1,1,64,128,128],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[1,1,64,128,128],!torch.qint8> | |
%49 = torch.aten.int_repr %48 : !torch.vtensor<[1,1,64,128,128],!torch.qint8> -> !torch.vtensor<[1,1,64,128,128],si8> | |
%50 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%51 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%52 = torch.aten.item %50 : !torch.vtensor<[],f32> -> !torch.float | |
%53 = torch.aten.item %51 : !torch.vtensor<[],si8> -> !torch.int | |
%54 = torch.aten._make_per_tensor_quantized_tensor %49, %52, %53 : !torch.vtensor<[1,1,64,128,128],si8>, !torch.float, !torch.int -> !torch.vtensor<[1,1,64,128,128],!torch.qint8> | |
%55 = torch.aten.dequantize.self %54 : !torch.vtensor<[1,1,64,128,128],!torch.qint8> -> !torch.vtensor<[1,1,64,128,128],f32> | |
%56 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%57 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_0 = torch.constant.int 12 | |
%58 = torch.aten.item %56 : !torch.vtensor<[],f32> -> !torch.float | |
%59 = torch.aten.item %57 : !torch.vtensor<[],si8> -> !torch.int | |
%60 = torch.aten.quantize_per_tensor %0, %58, %59, %int12_0 : !torch.vtensor<[1],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[1],!torch.qint8> | |
%61 = torch.aten.int_repr %60 : !torch.vtensor<[1],!torch.qint8> -> !torch.vtensor<[1],si8> | |
%62 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%63 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%64 = torch.aten.item %62 : !torch.vtensor<[],f32> -> !torch.float | |
%65 = torch.aten.item %63 : !torch.vtensor<[],si8> -> !torch.int | |
%66 = torch.aten._make_per_tensor_quantized_tensor %61, %64, %65 : !torch.vtensor<[1],si8>, !torch.float, !torch.int -> !torch.vtensor<[1],!torch.qint8> | |
%67 = torch.aten.dequantize.self %66 : !torch.vtensor<[1],!torch.qint8> -> !torch.vtensor<[1],f32> | |
%68 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%69 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_1 = torch.constant.int 12 | |
%70 = torch.aten.item %68 : !torch.vtensor<[],f32> -> !torch.float | |
%71 = torch.aten.item %69 : !torch.vtensor<[],si8> -> !torch.int | |
%72 = torch.aten.quantize_per_tensor %1, %70, %71, %int12_1 : !torch.vtensor<[1],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[1],!torch.qint8> | |
%73 = torch.aten.int_repr %72 : !torch.vtensor<[1],!torch.qint8> -> !torch.vtensor<[1],si8> | |
%74 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%75 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%76 = torch.aten.item %74 : !torch.vtensor<[],f32> -> !torch.float | |
%77 = torch.aten.item %75 : !torch.vtensor<[],si8> -> !torch.int | |
%78 = torch.aten._make_per_tensor_quantized_tensor %73, %76, %77 : !torch.vtensor<[1],si8>, !torch.float, !torch.int -> !torch.vtensor<[1],!torch.qint8> | |
%79 = torch.aten.dequantize.self %78 : !torch.vtensor<[1],!torch.qint8> -> !torch.vtensor<[1],f32> | |
%80 = torch.vtensor.literal(dense<[0, 1, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int1 = torch.constant.int 1 | |
%int1_2 = torch.constant.int 1 | |
%int1048576 = torch.constant.int 1048576 | |
%81 = torch.prim.ListConstruct %int1, %int1_2, %int1048576 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%82 = torch.aten.reshape %55, %81 : !torch.vtensor<[1,1,64,128,128],f32>, !torch.list<int> -> !torch.vtensor<[1,1,1048576],f32> | |
%83 = torch.vtensor.literal(dense<1.000000e+00> : tensor<1xf32>) : !torch.vtensor<[1],f32> | |
%84 = torch.vtensor.literal(dense<0.000000e+00> : tensor<1xf32>) : !torch.vtensor<[1],f32> | |
%none_3 = torch.constant.none | |
%true = torch.constant.bool true | |
%false = torch.constant.bool false | |
%float9.999990e-06 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00 = torch.constant.float 0.000000e+00 | |
%85 = torch.aten.instance_norm %82, %83, %84, %none_3, %none_3, %true, %float0.000000e00, %float9.999990e-06, %false : !torch.vtensor<[1,1,1048576],f32>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[1,1,1048576],f32> | |
%86 = torch.aten._shape_as_tensor %55 : !torch.vtensor<[1,1,64,128,128],f32> -> !torch.vtensor<[5],si64> | |
%int0 = torch.constant.int 0 | |
%int0_4 = torch.constant.int 0 | |
%87 = torch.aten.select.int %86, %int0, %int0_4 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%88 = torch.aten.item %87 : !torch.vtensor<[1],si64> -> !torch.int | |
%89 = torch.aten.eq.int %88, %int0 : !torch.int, !torch.int -> !torch.bool | |
%90 = torch.aten.Int.bool %89 : !torch.bool -> !torch.int | |
%int0_5 = torch.constant.int 0 | |
%91 = torch.aten.size.int %85, %int0_5 : !torch.vtensor<[1,1,1048576],f32>, !torch.int -> !torch.int | |
%92 = torch.prim.NumToTensor.Scalar %90 : !torch.int -> !torch.vtensor<[],i1> | |
%93 = torch.prim.NumToTensor.Scalar %91 : !torch.int -> !torch.vtensor<[],si64> | |
%94 = torch.prim.NumToTensor.Scalar %88 : !torch.int -> !torch.vtensor<[],si64> | |
%95 = torch.aten.where.self %92, %93, %94 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%96 = torch.aten.item %95 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_6 = torch.constant.int 1 | |
%97 = torch.aten.select.int %86, %int0, %int1_6 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%98 = torch.aten.item %97 : !torch.vtensor<[1],si64> -> !torch.int | |
%99 = torch.aten.eq.int %98, %int0 : !torch.int, !torch.int -> !torch.bool | |
%100 = torch.aten.Int.bool %99 : !torch.bool -> !torch.int | |
%int1_7 = torch.constant.int 1 | |
%101 = torch.aten.size.int %85, %int1_7 : !torch.vtensor<[1,1,1048576],f32>, !torch.int -> !torch.int | |
%102 = torch.prim.NumToTensor.Scalar %100 : !torch.int -> !torch.vtensor<[],i1> | |
%103 = torch.prim.NumToTensor.Scalar %101 : !torch.int -> !torch.vtensor<[],si64> | |
%104 = torch.prim.NumToTensor.Scalar %98 : !torch.int -> !torch.vtensor<[],si64> | |
%105 = torch.aten.where.self %102, %103, %104 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%106 = torch.aten.item %105 : !torch.vtensor<[],si64> -> !torch.int | |
%int2 = torch.constant.int 2 | |
%107 = torch.aten.select.int %86, %int0, %int2 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%108 = torch.aten.item %107 : !torch.vtensor<[1],si64> -> !torch.int | |
%109 = torch.aten.eq.int %108, %int0 : !torch.int, !torch.int -> !torch.bool | |
%110 = torch.aten.Int.bool %109 : !torch.bool -> !torch.int | |
%int2_8 = torch.constant.int 2 | |
%111 = torch.aten.size.int %85, %int2_8 : !torch.vtensor<[1,1,1048576],f32>, !torch.int -> !torch.int | |
%112 = torch.prim.NumToTensor.Scalar %110 : !torch.int -> !torch.vtensor<[],i1> | |
%113 = torch.prim.NumToTensor.Scalar %111 : !torch.int -> !torch.vtensor<[],si64> | |
%114 = torch.prim.NumToTensor.Scalar %108 : !torch.int -> !torch.vtensor<[],si64> | |
%115 = torch.aten.where.self %112, %113, %114 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%116 = torch.aten.item %115 : !torch.vtensor<[],si64> -> !torch.int | |
%int3 = torch.constant.int 3 | |
%117 = torch.aten.select.int %86, %int0, %int3 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%118 = torch.aten.item %117 : !torch.vtensor<[1],si64> -> !torch.int | |
%119 = torch.aten.eq.int %118, %int0 : !torch.int, !torch.int -> !torch.bool | |
%120 = torch.aten.Int.bool %119 : !torch.bool -> !torch.int | |
%int4 = torch.constant.int 4 | |
%121 = torch.aten.select.int %86, %int0, %int4 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%122 = torch.aten.item %121 : !torch.vtensor<[1],si64> -> !torch.int | |
%123 = torch.aten.eq.int %122, %int0 : !torch.int, !torch.int -> !torch.bool | |
%124 = torch.aten.Int.bool %123 : !torch.bool -> !torch.int | |
%125 = torch.prim.ListConstruct %96, %106, %116, %118, %122 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%126 = torch.aten.reshape %85, %125 : !torch.vtensor<[1,1,1048576],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%127 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_9 = torch.constant.int 0 | |
%int1_10 = torch.constant.int 1 | |
%int0_11 = torch.constant.int 0 | |
%128 = torch.aten.select.int %127, %int0_9, %int0_11 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%129 = torch.aten.item %128 : !torch.vtensor<[1],si64> -> !torch.int | |
%130 = torch.aten.lt.int %129, %int0_9 : !torch.int, !torch.int -> !torch.bool | |
%131 = torch.aten.Int.bool %130 : !torch.bool -> !torch.int | |
%132 = torch.aten.mul.int %131, %int1_10 : !torch.int, !torch.int -> !torch.int | |
%133 = torch.aten.add.int %129, %132 : !torch.int, !torch.int -> !torch.int | |
%int1_12 = torch.constant.int 1 | |
%134 = torch.aten.select.int %127, %int0_9, %int1_12 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%135 = torch.aten.item %134 : !torch.vtensor<[1],si64> -> !torch.int | |
%136 = torch.aten.lt.int %135, %int0_9 : !torch.int, !torch.int -> !torch.bool | |
%137 = torch.aten.Int.bool %136 : !torch.bool -> !torch.int | |
%138 = torch.aten.mul.int %137, %int1_10 : !torch.int, !torch.int -> !torch.int | |
%139 = torch.aten.add.int %135, %138 : !torch.int, !torch.int -> !torch.int | |
%int2_13 = torch.constant.int 2 | |
%140 = torch.aten.select.int %127, %int0_9, %int2_13 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%141 = torch.aten.item %140 : !torch.vtensor<[1],si64> -> !torch.int | |
%142 = torch.aten.lt.int %141, %int0_9 : !torch.int, !torch.int -> !torch.bool | |
%143 = torch.aten.Int.bool %142 : !torch.bool -> !torch.int | |
%144 = torch.aten.mul.int %143, %int1_10 : !torch.int, !torch.int -> !torch.int | |
%145 = torch.aten.add.int %141, %144 : !torch.int, !torch.int -> !torch.int | |
%146 = torch.prim.ListConstruct %133, %139, %145 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_14 = torch.constant.bool false | |
%none_15 = torch.constant.none | |
%147 = torch.aten.tensor %146, %none_15, %none_15, %false_14 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values, %indices = torch.aten.sort %147, %int0_9, %false_14 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_16 = torch.constant.int 0 | |
%148 = torch.aten.select.int %values, %int0_9, %int0_16 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%149 = torch.aten.item %148 : !torch.vtensor<[1],si64> -> !torch.int | |
%150 = torch.aten.unsqueeze %67, %149 : !torch.vtensor<[1],f32>, !torch.int -> !torch.vtensor | |
%int1_17 = torch.constant.int 1 | |
%151 = torch.aten.select.int %values, %int0_9, %int1_17 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%152 = torch.aten.item %151 : !torch.vtensor<[1],si64> -> !torch.int | |
%153 = torch.aten.unsqueeze %150, %152 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_18 = torch.constant.int 2 | |
%154 = torch.aten.select.int %values, %int0_9, %int2_18 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%155 = torch.aten.item %154 : !torch.vtensor<[1],si64> -> !torch.int | |
%156 = torch.aten.unsqueeze %153, %155 : !torch.vtensor, !torch.int -> !torch.vtensor<[1,1,1,1],f32> | |
%157 = torch.aten.mul.Tensor %126, %156 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[1,1,1,1],f32> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%158 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_19 = torch.constant.int 0 | |
%int1_20 = torch.constant.int 1 | |
%int0_21 = torch.constant.int 0 | |
%159 = torch.aten.select.int %158, %int0_19, %int0_21 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%160 = torch.aten.item %159 : !torch.vtensor<[1],si64> -> !torch.int | |
%161 = torch.aten.lt.int %160, %int0_19 : !torch.int, !torch.int -> !torch.bool | |
%162 = torch.aten.Int.bool %161 : !torch.bool -> !torch.int | |
%163 = torch.aten.mul.int %162, %int1_20 : !torch.int, !torch.int -> !torch.int | |
%164 = torch.aten.add.int %160, %163 : !torch.int, !torch.int -> !torch.int | |
%int1_22 = torch.constant.int 1 | |
%165 = torch.aten.select.int %158, %int0_19, %int1_22 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%166 = torch.aten.item %165 : !torch.vtensor<[1],si64> -> !torch.int | |
%167 = torch.aten.lt.int %166, %int0_19 : !torch.int, !torch.int -> !torch.bool | |
%168 = torch.aten.Int.bool %167 : !torch.bool -> !torch.int | |
%169 = torch.aten.mul.int %168, %int1_20 : !torch.int, !torch.int -> !torch.int | |
%170 = torch.aten.add.int %166, %169 : !torch.int, !torch.int -> !torch.int | |
%int2_23 = torch.constant.int 2 | |
%171 = torch.aten.select.int %158, %int0_19, %int2_23 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%172 = torch.aten.item %171 : !torch.vtensor<[1],si64> -> !torch.int | |
%173 = torch.aten.lt.int %172, %int0_19 : !torch.int, !torch.int -> !torch.bool | |
%174 = torch.aten.Int.bool %173 : !torch.bool -> !torch.int | |
%175 = torch.aten.mul.int %174, %int1_20 : !torch.int, !torch.int -> !torch.int | |
%176 = torch.aten.add.int %172, %175 : !torch.int, !torch.int -> !torch.int | |
%177 = torch.prim.ListConstruct %164, %170, %176 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_24 = torch.constant.bool false | |
%none_25 = torch.constant.none | |
%178 = torch.aten.tensor %177, %none_25, %none_25, %false_24 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_26, %indices_27 = torch.aten.sort %178, %int0_19, %false_24 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_28 = torch.constant.int 0 | |
%179 = torch.aten.select.int %values_26, %int0_19, %int0_28 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%180 = torch.aten.item %179 : !torch.vtensor<[1],si64> -> !torch.int | |
%181 = torch.aten.unsqueeze %79, %180 : !torch.vtensor<[1],f32>, !torch.int -> !torch.vtensor | |
%int1_29 = torch.constant.int 1 | |
%182 = torch.aten.select.int %values_26, %int0_19, %int1_29 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%183 = torch.aten.item %182 : !torch.vtensor<[1],si64> -> !torch.int | |
%184 = torch.aten.unsqueeze %181, %183 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_30 = torch.constant.int 2 | |
%185 = torch.aten.select.int %values_26, %int0_19, %int2_30 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%186 = torch.aten.item %185 : !torch.vtensor<[1],si64> -> !torch.int | |
%187 = torch.aten.unsqueeze %184, %186 : !torch.vtensor, !torch.int -> !torch.vtensor<[1,1,1,1],f32> | |
%int1_31 = torch.constant.int 1 | |
%188 = torch.aten.add.Tensor %157, %187, %int1_31 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[1,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,?,?,?,?],f32> | |
%189 = torch.vtensor.literal(dense<3.125000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%190 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_32 = torch.constant.int 12 | |
%191 = torch.aten.item %189 : !torch.vtensor<[],f32> -> !torch.float | |
%192 = torch.aten.item %190 : !torch.vtensor<[],si8> -> !torch.int | |
%193 = torch.aten.quantize_per_tensor %188, %191, %192, %int12_32 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,?,?,?,?],!torch.qint8> | |
%194 = torch.aten.int_repr %193 : !torch.vtensor<[?,?,?,?,?],!torch.qint8> -> !torch.vtensor<[?,?,?,?,?],si8> | |
%195 = torch.vtensor.literal(dense<3.125000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%196 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%197 = torch.aten.item %195 : !torch.vtensor<[],f32> -> !torch.float | |
%198 = torch.aten.item %196 : !torch.vtensor<[],si8> -> !torch.int | |
%199 = torch.aten._make_per_tensor_quantized_tensor %194, %197, %198 : !torch.vtensor<[?,?,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,?,?,?,?],!torch.qint8> | |
%200 = torch.aten.dequantize.self %199 : !torch.vtensor<[?,?,?,?,?],!torch.qint8> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%201 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%202 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_33 = torch.constant.int 12 | |
%203 = torch.aten.item %201 : !torch.vtensor<[],f32> -> !torch.float | |
%204 = torch.aten.item %202 : !torch.vtensor<[],si8> -> !torch.int | |
%205 = torch.aten.quantize_per_tensor %2, %203, %204, %int12_33 : !torch.vtensor<[16,1,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[16,1,3,3,3],!torch.qint8> | |
%206 = torch.aten.int_repr %205 : !torch.vtensor<[16,1,3,3,3],!torch.qint8> -> !torch.vtensor<[16,1,3,3,3],si8> | |
%207 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%208 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%209 = torch.aten.item %207 : !torch.vtensor<[],f32> -> !torch.float | |
%210 = torch.aten.item %208 : !torch.vtensor<[],si8> -> !torch.int | |
%211 = torch.aten._make_per_tensor_quantized_tensor %206, %209, %210 : !torch.vtensor<[16,1,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[16,1,3,3,3],!torch.qint8> | |
%212 = torch.aten.dequantize.self %211 : !torch.vtensor<[16,1,3,3,3],!torch.qint8> -> !torch.vtensor<[16,1,3,3,3],f32> | |
%int1_34 = torch.constant.int 1 | |
%int1_35 = torch.constant.int 1 | |
%int1_36 = torch.constant.int 1 | |
%int1_37 = torch.constant.int 1 | |
%int1_38 = torch.constant.int 1 | |
%int1_39 = torch.constant.int 1 | |
%int1_40 = torch.constant.int 1 | |
%int1_41 = torch.constant.int 1 | |
%int1_42 = torch.constant.int 1 | |
%int0_43 = torch.constant.int 0 | |
%213 = torch.prim.ListConstruct %int1_34, %int1_35, %int1_36 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%214 = torch.prim.ListConstruct %int1_37, %int1_38, %int1_39 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%215 = torch.prim.ListConstruct %int1_40, %int1_41, %int1_42 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%216 = torch.prim.ListConstruct %int0_43, %int0_43 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_44 = torch.constant.bool false | |
%none_45 = torch.constant.none | |
%int1_46 = torch.constant.int 1 | |
%217 = torch.aten.convolution %200, %212, %none_45, %215, %213, %214, %false_44, %216, %int1_46 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[16,1,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,16,?,?,?],f32> | |
%218 = torch.aten.relu %217 : !torch.vtensor<[?,16,?,?,?],f32> -> !torch.vtensor<[?,16,?,?,?],f32> | |
%219 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%220 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_47 = torch.constant.int 12 | |
%221 = torch.aten.item %219 : !torch.vtensor<[],f32> -> !torch.float | |
%222 = torch.aten.item %220 : !torch.vtensor<[],si8> -> !torch.int | |
%223 = torch.aten.quantize_per_tensor %218, %221, %222, %int12_47 : !torch.vtensor<[?,16,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,16,?,?,?],!torch.qint8> | |
%224 = torch.aten.int_repr %223 : !torch.vtensor<[?,16,?,?,?],!torch.qint8> -> !torch.vtensor<[?,16,?,?,?],si8> | |
%225 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%226 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%227 = torch.aten.item %225 : !torch.vtensor<[],f32> -> !torch.float | |
%228 = torch.aten.item %226 : !torch.vtensor<[],si8> -> !torch.int | |
%229 = torch.aten._make_per_tensor_quantized_tensor %224, %227, %228 : !torch.vtensor<[?,16,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,16,?,?,?],!torch.qint8> | |
%230 = torch.aten.dequantize.self %229 : !torch.vtensor<[?,16,?,?,?],!torch.qint8> -> !torch.vtensor<[?,16,?,?,?],f32> | |
%231 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%232 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_48 = torch.constant.int 12 | |
%233 = torch.aten.item %231 : !torch.vtensor<[],f32> -> !torch.float | |
%234 = torch.aten.item %232 : !torch.vtensor<[],si8> -> !torch.int | |
%235 = torch.aten.quantize_per_tensor %3, %233, %234, %int12_48 : !torch.vtensor<[16],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[16],!torch.qint8> | |
%236 = torch.aten.int_repr %235 : !torch.vtensor<[16],!torch.qint8> -> !torch.vtensor<[16],si8> | |
%237 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%238 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%239 = torch.aten.item %237 : !torch.vtensor<[],f32> -> !torch.float | |
%240 = torch.aten.item %238 : !torch.vtensor<[],si8> -> !torch.int | |
%241 = torch.aten._make_per_tensor_quantized_tensor %236, %239, %240 : !torch.vtensor<[16],si8>, !torch.float, !torch.int -> !torch.vtensor<[16],!torch.qint8> | |
%242 = torch.aten.dequantize.self %241 : !torch.vtensor<[16],!torch.qint8> -> !torch.vtensor<[16],f32> | |
%243 = torch.vtensor.literal(dense<4.8828125E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%244 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_49 = torch.constant.int 12 | |
%245 = torch.aten.item %243 : !torch.vtensor<[],f32> -> !torch.float | |
%246 = torch.aten.item %244 : !torch.vtensor<[],si8> -> !torch.int | |
%247 = torch.aten.quantize_per_tensor %4, %245, %246, %int12_49 : !torch.vtensor<[16],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[16],!torch.qint8> | |
%248 = torch.aten.int_repr %247 : !torch.vtensor<[16],!torch.qint8> -> !torch.vtensor<[16],si8> | |
%249 = torch.vtensor.literal(dense<4.8828125E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%250 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%251 = torch.aten.item %249 : !torch.vtensor<[],f32> -> !torch.float | |
%252 = torch.aten.item %250 : !torch.vtensor<[],si8> -> !torch.int | |
%253 = torch.aten._make_per_tensor_quantized_tensor %248, %251, %252 : !torch.vtensor<[16],si8>, !torch.float, !torch.int -> !torch.vtensor<[16],!torch.qint8> | |
%254 = torch.aten.dequantize.self %253 : !torch.vtensor<[16],!torch.qint8> -> !torch.vtensor<[16],f32> | |
%255 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_50 = torch.constant.int 0 | |
%int0_51 = torch.constant.int 0 | |
%256 = torch.aten.select.int %255, %int0_50, %int0_51 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%257 = torch.aten.item %256 : !torch.vtensor<[1],si64> -> !torch.int | |
%258 = torch.aten.eq.int %257, %int0_50 : !torch.int, !torch.int -> !torch.bool | |
%259 = torch.aten.Int.bool %258 : !torch.bool -> !torch.int | |
%int0_52 = torch.constant.int 0 | |
%260 = torch.aten.size.int %230, %int0_52 : !torch.vtensor<[?,16,?,?,?],f32>, !torch.int -> !torch.int | |
%261 = torch.prim.NumToTensor.Scalar %259 : !torch.int -> !torch.vtensor<[],i1> | |
%262 = torch.prim.NumToTensor.Scalar %260 : !torch.int -> !torch.vtensor<[],si64> | |
%263 = torch.prim.NumToTensor.Scalar %257 : !torch.int -> !torch.vtensor<[],si64> | |
%264 = torch.aten.where.self %261, %262, %263 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%265 = torch.aten.item %264 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_53 = torch.constant.int 1 | |
%266 = torch.aten.select.int %255, %int0_50, %int1_53 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%267 = torch.aten.item %266 : !torch.vtensor<[1],si64> -> !torch.int | |
%268 = torch.aten.eq.int %267, %int0_50 : !torch.int, !torch.int -> !torch.bool | |
%269 = torch.aten.Int.bool %268 : !torch.bool -> !torch.int | |
%int1_54 = torch.constant.int 1 | |
%270 = torch.aten.size.int %230, %int1_54 : !torch.vtensor<[?,16,?,?,?],f32>, !torch.int -> !torch.int | |
%271 = torch.prim.NumToTensor.Scalar %269 : !torch.int -> !torch.vtensor<[],i1> | |
%272 = torch.prim.NumToTensor.Scalar %270 : !torch.int -> !torch.vtensor<[],si64> | |
%273 = torch.prim.NumToTensor.Scalar %267 : !torch.int -> !torch.vtensor<[],si64> | |
%274 = torch.aten.where.self %271, %272, %273 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%275 = torch.aten.item %274 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_55 = torch.constant.int 2 | |
%276 = torch.aten.select.int %255, %int0_50, %int2_55 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%277 = torch.aten.item %276 : !torch.vtensor<[1],si64> -> !torch.int | |
%278 = torch.aten.eq.int %277, %int0_50 : !torch.int, !torch.int -> !torch.bool | |
%279 = torch.aten.Int.bool %278 : !torch.bool -> !torch.int | |
%int2_56 = torch.constant.int 2 | |
%280 = torch.aten.size.int %230, %int2_56 : !torch.vtensor<[?,16,?,?,?],f32>, !torch.int -> !torch.int | |
%281 = torch.prim.NumToTensor.Scalar %279 : !torch.int -> !torch.vtensor<[],i1> | |
%282 = torch.prim.NumToTensor.Scalar %280 : !torch.int -> !torch.vtensor<[],si64> | |
%283 = torch.prim.NumToTensor.Scalar %277 : !torch.int -> !torch.vtensor<[],si64> | |
%284 = torch.aten.where.self %281, %282, %283 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%285 = torch.aten.item %284 : !torch.vtensor<[],si64> -> !torch.int | |
%286 = torch.prim.ListConstruct %265, %275, %285 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%287 = torch.aten.reshape %230, %286 : !torch.vtensor<[?,16,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%288 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%289 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_57 = torch.constant.none | |
%true_58 = torch.constant.bool true | |
%false_59 = torch.constant.bool false | |
%float9.999990e-06_60 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_61 = torch.constant.float 0.000000e+00 | |
%290 = torch.aten.instance_norm %287, %288, %289, %none_57, %none_57, %true_58, %float0.000000e00_61, %float9.999990e-06_60, %false_59 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%291 = torch.aten._shape_as_tensor %230 : !torch.vtensor<[?,16,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_62 = torch.constant.int 0 | |
%int0_63 = torch.constant.int 0 | |
%292 = torch.aten.select.int %291, %int0_62, %int0_63 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%293 = torch.aten.item %292 : !torch.vtensor<[1],si64> -> !torch.int | |
%294 = torch.aten.eq.int %293, %int0_62 : !torch.int, !torch.int -> !torch.bool | |
%295 = torch.aten.Int.bool %294 : !torch.bool -> !torch.int | |
%int0_64 = torch.constant.int 0 | |
%296 = torch.aten.size.int %290, %int0_64 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%297 = torch.prim.NumToTensor.Scalar %295 : !torch.int -> !torch.vtensor<[],i1> | |
%298 = torch.prim.NumToTensor.Scalar %296 : !torch.int -> !torch.vtensor<[],si64> | |
%299 = torch.prim.NumToTensor.Scalar %293 : !torch.int -> !torch.vtensor<[],si64> | |
%300 = torch.aten.where.self %297, %298, %299 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%301 = torch.aten.item %300 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_65 = torch.constant.int 1 | |
%302 = torch.aten.select.int %291, %int0_62, %int1_65 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%303 = torch.aten.item %302 : !torch.vtensor<[1],si64> -> !torch.int | |
%304 = torch.aten.eq.int %303, %int0_62 : !torch.int, !torch.int -> !torch.bool | |
%305 = torch.aten.Int.bool %304 : !torch.bool -> !torch.int | |
%int1_66 = torch.constant.int 1 | |
%306 = torch.aten.size.int %290, %int1_66 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%307 = torch.prim.NumToTensor.Scalar %305 : !torch.int -> !torch.vtensor<[],i1> | |
%308 = torch.prim.NumToTensor.Scalar %306 : !torch.int -> !torch.vtensor<[],si64> | |
%309 = torch.prim.NumToTensor.Scalar %303 : !torch.int -> !torch.vtensor<[],si64> | |
%310 = torch.aten.where.self %307, %308, %309 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%311 = torch.aten.item %310 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_67 = torch.constant.int 2 | |
%312 = torch.aten.select.int %291, %int0_62, %int2_67 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%313 = torch.aten.item %312 : !torch.vtensor<[1],si64> -> !torch.int | |
%314 = torch.aten.eq.int %313, %int0_62 : !torch.int, !torch.int -> !torch.bool | |
%315 = torch.aten.Int.bool %314 : !torch.bool -> !torch.int | |
%int2_68 = torch.constant.int 2 | |
%316 = torch.aten.size.int %290, %int2_68 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%317 = torch.prim.NumToTensor.Scalar %315 : !torch.int -> !torch.vtensor<[],i1> | |
%318 = torch.prim.NumToTensor.Scalar %316 : !torch.int -> !torch.vtensor<[],si64> | |
%319 = torch.prim.NumToTensor.Scalar %313 : !torch.int -> !torch.vtensor<[],si64> | |
%320 = torch.aten.where.self %317, %318, %319 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%321 = torch.aten.item %320 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_69 = torch.constant.int 3 | |
%322 = torch.aten.select.int %291, %int0_62, %int3_69 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%323 = torch.aten.item %322 : !torch.vtensor<[1],si64> -> !torch.int | |
%324 = torch.aten.eq.int %323, %int0_62 : !torch.int, !torch.int -> !torch.bool | |
%325 = torch.aten.Int.bool %324 : !torch.bool -> !torch.int | |
%int4_70 = torch.constant.int 4 | |
%326 = torch.aten.select.int %291, %int0_62, %int4_70 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%327 = torch.aten.item %326 : !torch.vtensor<[1],si64> -> !torch.int | |
%328 = torch.aten.eq.int %327, %int0_62 : !torch.int, !torch.int -> !torch.bool | |
%329 = torch.aten.Int.bool %328 : !torch.bool -> !torch.int | |
%330 = torch.prim.ListConstruct %301, %311, %321, %323, %327 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%331 = torch.aten.reshape %290, %330 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%332 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_71 = torch.constant.int 0 | |
%int1_72 = torch.constant.int 1 | |
%int0_73 = torch.constant.int 0 | |
%333 = torch.aten.select.int %332, %int0_71, %int0_73 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%334 = torch.aten.item %333 : !torch.vtensor<[1],si64> -> !torch.int | |
%335 = torch.aten.lt.int %334, %int0_71 : !torch.int, !torch.int -> !torch.bool | |
%336 = torch.aten.Int.bool %335 : !torch.bool -> !torch.int | |
%337 = torch.aten.mul.int %336, %int1_72 : !torch.int, !torch.int -> !torch.int | |
%338 = torch.aten.add.int %334, %337 : !torch.int, !torch.int -> !torch.int | |
%int1_74 = torch.constant.int 1 | |
%339 = torch.aten.select.int %332, %int0_71, %int1_74 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%340 = torch.aten.item %339 : !torch.vtensor<[1],si64> -> !torch.int | |
%341 = torch.aten.lt.int %340, %int0_71 : !torch.int, !torch.int -> !torch.bool | |
%342 = torch.aten.Int.bool %341 : !torch.bool -> !torch.int | |
%343 = torch.aten.mul.int %342, %int1_72 : !torch.int, !torch.int -> !torch.int | |
%344 = torch.aten.add.int %340, %343 : !torch.int, !torch.int -> !torch.int | |
%int2_75 = torch.constant.int 2 | |
%345 = torch.aten.select.int %332, %int0_71, %int2_75 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%346 = torch.aten.item %345 : !torch.vtensor<[1],si64> -> !torch.int | |
%347 = torch.aten.lt.int %346, %int0_71 : !torch.int, !torch.int -> !torch.bool | |
%348 = torch.aten.Int.bool %347 : !torch.bool -> !torch.int | |
%349 = torch.aten.mul.int %348, %int1_72 : !torch.int, !torch.int -> !torch.int | |
%350 = torch.aten.add.int %346, %349 : !torch.int, !torch.int -> !torch.int | |
%351 = torch.prim.ListConstruct %338, %344, %350 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_76 = torch.constant.bool false | |
%none_77 = torch.constant.none | |
%352 = torch.aten.tensor %351, %none_77, %none_77, %false_76 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_78, %indices_79 = torch.aten.sort %352, %int0_71, %false_76 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_80 = torch.constant.int 0 | |
%353 = torch.aten.select.int %values_78, %int0_71, %int0_80 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%354 = torch.aten.item %353 : !torch.vtensor<[1],si64> -> !torch.int | |
%355 = torch.aten.unsqueeze %242, %354 : !torch.vtensor<[16],f32>, !torch.int -> !torch.vtensor | |
%int1_81 = torch.constant.int 1 | |
%356 = torch.aten.select.int %values_78, %int0_71, %int1_81 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%357 = torch.aten.item %356 : !torch.vtensor<[1],si64> -> !torch.int | |
%358 = torch.aten.unsqueeze %355, %357 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_82 = torch.constant.int 2 | |
%359 = torch.aten.select.int %values_78, %int0_71, %int2_82 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%360 = torch.aten.item %359 : !torch.vtensor<[1],si64> -> !torch.int | |
%361 = torch.aten.unsqueeze %358, %360 : !torch.vtensor, !torch.int -> !torch.vtensor<[16,1,1,1],f32> | |
%362 = torch.aten.mul.Tensor %331, %361 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[16,1,1,1],f32> -> !torch.vtensor<[?,16,?,?,?],f32> | |
%363 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_83 = torch.constant.int 0 | |
%int1_84 = torch.constant.int 1 | |
%int0_85 = torch.constant.int 0 | |
%364 = torch.aten.select.int %363, %int0_83, %int0_85 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%365 = torch.aten.item %364 : !torch.vtensor<[1],si64> -> !torch.int | |
%366 = torch.aten.lt.int %365, %int0_83 : !torch.int, !torch.int -> !torch.bool | |
%367 = torch.aten.Int.bool %366 : !torch.bool -> !torch.int | |
%368 = torch.aten.mul.int %367, %int1_84 : !torch.int, !torch.int -> !torch.int | |
%369 = torch.aten.add.int %365, %368 : !torch.int, !torch.int -> !torch.int | |
%int1_86 = torch.constant.int 1 | |
%370 = torch.aten.select.int %363, %int0_83, %int1_86 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%371 = torch.aten.item %370 : !torch.vtensor<[1],si64> -> !torch.int | |
%372 = torch.aten.lt.int %371, %int0_83 : !torch.int, !torch.int -> !torch.bool | |
%373 = torch.aten.Int.bool %372 : !torch.bool -> !torch.int | |
%374 = torch.aten.mul.int %373, %int1_84 : !torch.int, !torch.int -> !torch.int | |
%375 = torch.aten.add.int %371, %374 : !torch.int, !torch.int -> !torch.int | |
%int2_87 = torch.constant.int 2 | |
%376 = torch.aten.select.int %363, %int0_83, %int2_87 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%377 = torch.aten.item %376 : !torch.vtensor<[1],si64> -> !torch.int | |
%378 = torch.aten.lt.int %377, %int0_83 : !torch.int, !torch.int -> !torch.bool | |
%379 = torch.aten.Int.bool %378 : !torch.bool -> !torch.int | |
%380 = torch.aten.mul.int %379, %int1_84 : !torch.int, !torch.int -> !torch.int | |
%381 = torch.aten.add.int %377, %380 : !torch.int, !torch.int -> !torch.int | |
%382 = torch.prim.ListConstruct %369, %375, %381 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_88 = torch.constant.bool false | |
%none_89 = torch.constant.none | |
%383 = torch.aten.tensor %382, %none_89, %none_89, %false_88 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_90, %indices_91 = torch.aten.sort %383, %int0_83, %false_88 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_92 = torch.constant.int 0 | |
%384 = torch.aten.select.int %values_90, %int0_83, %int0_92 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%385 = torch.aten.item %384 : !torch.vtensor<[1],si64> -> !torch.int | |
%386 = torch.aten.unsqueeze %254, %385 : !torch.vtensor<[16],f32>, !torch.int -> !torch.vtensor | |
%int1_93 = torch.constant.int 1 | |
%387 = torch.aten.select.int %values_90, %int0_83, %int1_93 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%388 = torch.aten.item %387 : !torch.vtensor<[1],si64> -> !torch.int | |
%389 = torch.aten.unsqueeze %386, %388 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_94 = torch.constant.int 2 | |
%390 = torch.aten.select.int %values_90, %int0_83, %int2_94 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%391 = torch.aten.item %390 : !torch.vtensor<[1],si64> -> !torch.int | |
%392 = torch.aten.unsqueeze %389, %391 : !torch.vtensor, !torch.int -> !torch.vtensor<[16,1,1,1],f32> | |
%int1_95 = torch.constant.int 1 | |
%393 = torch.aten.add.Tensor %362, %392, %int1_95 : !torch.vtensor<[?,16,?,?,?],f32>, !torch.vtensor<[16,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,16,?,?,?],f32> | |
%394 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%395 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_96 = torch.constant.int 12 | |
%396 = torch.aten.item %394 : !torch.vtensor<[],f32> -> !torch.float | |
%397 = torch.aten.item %395 : !torch.vtensor<[],si8> -> !torch.int | |
%398 = torch.aten.quantize_per_tensor %393, %396, %397, %int12_96 : !torch.vtensor<[?,16,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,16,?,?,?],!torch.qint8> | |
%399 = torch.aten.int_repr %398 : !torch.vtensor<[?,16,?,?,?],!torch.qint8> -> !torch.vtensor<[?,16,?,?,?],si8> | |
%400 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%401 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%402 = torch.aten.item %400 : !torch.vtensor<[],f32> -> !torch.float | |
%403 = torch.aten.item %401 : !torch.vtensor<[],si8> -> !torch.int | |
%404 = torch.aten._make_per_tensor_quantized_tensor %399, %402, %403 : !torch.vtensor<[?,16,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,16,?,?,?],!torch.qint8> | |
%405 = torch.aten.dequantize.self %404 : !torch.vtensor<[?,16,?,?,?],!torch.qint8> -> !torch.vtensor<[?,16,?,?,?],f32> | |
%406 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%407 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_97 = torch.constant.int 12 | |
%408 = torch.aten.item %406 : !torch.vtensor<[],f32> -> !torch.float | |
%409 = torch.aten.item %407 : !torch.vtensor<[],si8> -> !torch.int | |
%410 = torch.aten.quantize_per_tensor %5, %408, %409, %int12_97 : !torch.vtensor<[32,16,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32,16,3,3,3],!torch.qint8> | |
%411 = torch.aten.int_repr %410 : !torch.vtensor<[32,16,3,3,3],!torch.qint8> -> !torch.vtensor<[32,16,3,3,3],si8> | |
%412 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%413 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%414 = torch.aten.item %412 : !torch.vtensor<[],f32> -> !torch.float | |
%415 = torch.aten.item %413 : !torch.vtensor<[],si8> -> !torch.int | |
%416 = torch.aten._make_per_tensor_quantized_tensor %411, %414, %415 : !torch.vtensor<[32,16,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[32,16,3,3,3],!torch.qint8> | |
%417 = torch.aten.dequantize.self %416 : !torch.vtensor<[32,16,3,3,3],!torch.qint8> -> !torch.vtensor<[32,16,3,3,3],f32> | |
%int1_98 = torch.constant.int 1 | |
%int1_99 = torch.constant.int 1 | |
%int1_100 = torch.constant.int 1 | |
%int1_101 = torch.constant.int 1 | |
%int1_102 = torch.constant.int 1 | |
%int1_103 = torch.constant.int 1 | |
%int1_104 = torch.constant.int 1 | |
%int1_105 = torch.constant.int 1 | |
%int1_106 = torch.constant.int 1 | |
%int0_107 = torch.constant.int 0 | |
%418 = torch.prim.ListConstruct %int1_98, %int1_99, %int1_100 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%419 = torch.prim.ListConstruct %int1_101, %int1_102, %int1_103 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%420 = torch.prim.ListConstruct %int1_104, %int1_105, %int1_106 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%421 = torch.prim.ListConstruct %int0_107, %int0_107 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_108 = torch.constant.bool false | |
%none_109 = torch.constant.none | |
%int1_110 = torch.constant.int 1 | |
%422 = torch.aten.convolution %405, %417, %none_109, %420, %418, %419, %false_108, %421, %int1_110 : !torch.vtensor<[?,16,?,?,?],f32>, !torch.vtensor<[32,16,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,32,?,?,?],f32> | |
%423 = torch.aten.relu %422 : !torch.vtensor<[?,32,?,?,?],f32> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%424 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%425 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_111 = torch.constant.int 12 | |
%426 = torch.aten.item %424 : !torch.vtensor<[],f32> -> !torch.float | |
%427 = torch.aten.item %425 : !torch.vtensor<[],si8> -> !torch.int | |
%428 = torch.aten.quantize_per_tensor %423, %426, %427, %int12_111 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%429 = torch.aten.int_repr %428 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],si8> | |
%430 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%431 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%432 = torch.aten.item %430 : !torch.vtensor<[],f32> -> !torch.float | |
%433 = torch.aten.item %431 : !torch.vtensor<[],si8> -> !torch.int | |
%434 = torch.aten._make_per_tensor_quantized_tensor %429, %432, %433 : !torch.vtensor<[?,32,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%435 = torch.aten.dequantize.self %434 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%int2_112 = torch.constant.int 2 | |
%int2_113 = torch.constant.int 2 | |
%int2_114 = torch.constant.int 2 | |
%436 = torch.prim.ListConstruct %int2_112, %int2_113, %int2_114 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int0_115 = torch.constant.int 0 | |
%int0_116 = torch.constant.int 0 | |
%int0_117 = torch.constant.int 0 | |
%437 = torch.prim.ListConstruct %int0_115, %int0_116, %int0_117 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int2_118 = torch.constant.int 2 | |
%int2_119 = torch.constant.int 2 | |
%int2_120 = torch.constant.int 2 | |
%438 = torch.prim.ListConstruct %int2_118, %int2_119, %int2_120 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int1_121 = torch.constant.int 1 | |
%int1_122 = torch.constant.int 1 | |
%int1_123 = torch.constant.int 1 | |
%439 = torch.prim.ListConstruct %int1_121, %int1_122, %int1_123 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_124 = torch.constant.bool false | |
%440 = torch.aten.max_pool3d %435, %436, %438, %437, %439, %false_124 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.vtensor<[?,32,?,?,?],f32> | |
%441 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%442 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_125 = torch.constant.int 12 | |
%443 = torch.aten.item %441 : !torch.vtensor<[],f32> -> !torch.float | |
%444 = torch.aten.item %442 : !torch.vtensor<[],si8> -> !torch.int | |
%445 = torch.aten.quantize_per_tensor %440, %443, %444, %int12_125 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%446 = torch.aten.int_repr %445 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],si8> | |
%447 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%448 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%449 = torch.aten.item %447 : !torch.vtensor<[],f32> -> !torch.float | |
%450 = torch.aten.item %448 : !torch.vtensor<[],si8> -> !torch.int | |
%451 = torch.aten._make_per_tensor_quantized_tensor %446, %449, %450 : !torch.vtensor<[?,32,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%452 = torch.aten.dequantize.self %451 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%453 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%454 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_126 = torch.constant.int 12 | |
%455 = torch.aten.item %453 : !torch.vtensor<[],f32> -> !torch.float | |
%456 = torch.aten.item %454 : !torch.vtensor<[],si8> -> !torch.int | |
%457 = torch.aten.quantize_per_tensor %6, %455, %456, %int12_126 : !torch.vtensor<[32],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%458 = torch.aten.int_repr %457 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],si8> | |
%459 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%460 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%461 = torch.aten.item %459 : !torch.vtensor<[],f32> -> !torch.float | |
%462 = torch.aten.item %460 : !torch.vtensor<[],si8> -> !torch.int | |
%463 = torch.aten._make_per_tensor_quantized_tensor %458, %461, %462 : !torch.vtensor<[32],si8>, !torch.float, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%464 = torch.aten.dequantize.self %463 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],f32> | |
%465 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%466 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_127 = torch.constant.int 12 | |
%467 = torch.aten.item %465 : !torch.vtensor<[],f32> -> !torch.float | |
%468 = torch.aten.item %466 : !torch.vtensor<[],si8> -> !torch.int | |
%469 = torch.aten.quantize_per_tensor %7, %467, %468, %int12_127 : !torch.vtensor<[32],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%470 = torch.aten.int_repr %469 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],si8> | |
%471 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%472 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%473 = torch.aten.item %471 : !torch.vtensor<[],f32> -> !torch.float | |
%474 = torch.aten.item %472 : !torch.vtensor<[],si8> -> !torch.int | |
%475 = torch.aten._make_per_tensor_quantized_tensor %470, %473, %474 : !torch.vtensor<[32],si8>, !torch.float, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%476 = torch.aten.dequantize.self %475 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],f32> | |
%477 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_128 = torch.constant.int 0 | |
%int0_129 = torch.constant.int 0 | |
%478 = torch.aten.select.int %477, %int0_128, %int0_129 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%479 = torch.aten.item %478 : !torch.vtensor<[1],si64> -> !torch.int | |
%480 = torch.aten.eq.int %479, %int0_128 : !torch.int, !torch.int -> !torch.bool | |
%481 = torch.aten.Int.bool %480 : !torch.bool -> !torch.int | |
%int0_130 = torch.constant.int 0 | |
%482 = torch.aten.size.int %452, %int0_130 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%483 = torch.prim.NumToTensor.Scalar %481 : !torch.int -> !torch.vtensor<[],i1> | |
%484 = torch.prim.NumToTensor.Scalar %482 : !torch.int -> !torch.vtensor<[],si64> | |
%485 = torch.prim.NumToTensor.Scalar %479 : !torch.int -> !torch.vtensor<[],si64> | |
%486 = torch.aten.where.self %483, %484, %485 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%487 = torch.aten.item %486 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_131 = torch.constant.int 1 | |
%488 = torch.aten.select.int %477, %int0_128, %int1_131 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%489 = torch.aten.item %488 : !torch.vtensor<[1],si64> -> !torch.int | |
%490 = torch.aten.eq.int %489, %int0_128 : !torch.int, !torch.int -> !torch.bool | |
%491 = torch.aten.Int.bool %490 : !torch.bool -> !torch.int | |
%int1_132 = torch.constant.int 1 | |
%492 = torch.aten.size.int %452, %int1_132 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%493 = torch.prim.NumToTensor.Scalar %491 : !torch.int -> !torch.vtensor<[],i1> | |
%494 = torch.prim.NumToTensor.Scalar %492 : !torch.int -> !torch.vtensor<[],si64> | |
%495 = torch.prim.NumToTensor.Scalar %489 : !torch.int -> !torch.vtensor<[],si64> | |
%496 = torch.aten.where.self %493, %494, %495 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%497 = torch.aten.item %496 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_133 = torch.constant.int 2 | |
%498 = torch.aten.select.int %477, %int0_128, %int2_133 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%499 = torch.aten.item %498 : !torch.vtensor<[1],si64> -> !torch.int | |
%500 = torch.aten.eq.int %499, %int0_128 : !torch.int, !torch.int -> !torch.bool | |
%501 = torch.aten.Int.bool %500 : !torch.bool -> !torch.int | |
%int2_134 = torch.constant.int 2 | |
%502 = torch.aten.size.int %452, %int2_134 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%503 = torch.prim.NumToTensor.Scalar %501 : !torch.int -> !torch.vtensor<[],i1> | |
%504 = torch.prim.NumToTensor.Scalar %502 : !torch.int -> !torch.vtensor<[],si64> | |
%505 = torch.prim.NumToTensor.Scalar %499 : !torch.int -> !torch.vtensor<[],si64> | |
%506 = torch.aten.where.self %503, %504, %505 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%507 = torch.aten.item %506 : !torch.vtensor<[],si64> -> !torch.int | |
%508 = torch.prim.ListConstruct %487, %497, %507 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%509 = torch.aten.reshape %452, %508 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%510 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%511 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_135 = torch.constant.none | |
%true_136 = torch.constant.bool true | |
%false_137 = torch.constant.bool false | |
%float9.999990e-06_138 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_139 = torch.constant.float 0.000000e+00 | |
%512 = torch.aten.instance_norm %509, %510, %511, %none_135, %none_135, %true_136, %float0.000000e00_139, %float9.999990e-06_138, %false_137 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%513 = torch.aten._shape_as_tensor %452 : !torch.vtensor<[?,32,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_140 = torch.constant.int 0 | |
%int0_141 = torch.constant.int 0 | |
%514 = torch.aten.select.int %513, %int0_140, %int0_141 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%515 = torch.aten.item %514 : !torch.vtensor<[1],si64> -> !torch.int | |
%516 = torch.aten.eq.int %515, %int0_140 : !torch.int, !torch.int -> !torch.bool | |
%517 = torch.aten.Int.bool %516 : !torch.bool -> !torch.int | |
%int0_142 = torch.constant.int 0 | |
%518 = torch.aten.size.int %512, %int0_142 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%519 = torch.prim.NumToTensor.Scalar %517 : !torch.int -> !torch.vtensor<[],i1> | |
%520 = torch.prim.NumToTensor.Scalar %518 : !torch.int -> !torch.vtensor<[],si64> | |
%521 = torch.prim.NumToTensor.Scalar %515 : !torch.int -> !torch.vtensor<[],si64> | |
%522 = torch.aten.where.self %519, %520, %521 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%523 = torch.aten.item %522 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_143 = torch.constant.int 1 | |
%524 = torch.aten.select.int %513, %int0_140, %int1_143 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%525 = torch.aten.item %524 : !torch.vtensor<[1],si64> -> !torch.int | |
%526 = torch.aten.eq.int %525, %int0_140 : !torch.int, !torch.int -> !torch.bool | |
%527 = torch.aten.Int.bool %526 : !torch.bool -> !torch.int | |
%int1_144 = torch.constant.int 1 | |
%528 = torch.aten.size.int %512, %int1_144 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%529 = torch.prim.NumToTensor.Scalar %527 : !torch.int -> !torch.vtensor<[],i1> | |
%530 = torch.prim.NumToTensor.Scalar %528 : !torch.int -> !torch.vtensor<[],si64> | |
%531 = torch.prim.NumToTensor.Scalar %525 : !torch.int -> !torch.vtensor<[],si64> | |
%532 = torch.aten.where.self %529, %530, %531 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%533 = torch.aten.item %532 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_145 = torch.constant.int 2 | |
%534 = torch.aten.select.int %513, %int0_140, %int2_145 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%535 = torch.aten.item %534 : !torch.vtensor<[1],si64> -> !torch.int | |
%536 = torch.aten.eq.int %535, %int0_140 : !torch.int, !torch.int -> !torch.bool | |
%537 = torch.aten.Int.bool %536 : !torch.bool -> !torch.int | |
%int2_146 = torch.constant.int 2 | |
%538 = torch.aten.size.int %512, %int2_146 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%539 = torch.prim.NumToTensor.Scalar %537 : !torch.int -> !torch.vtensor<[],i1> | |
%540 = torch.prim.NumToTensor.Scalar %538 : !torch.int -> !torch.vtensor<[],si64> | |
%541 = torch.prim.NumToTensor.Scalar %535 : !torch.int -> !torch.vtensor<[],si64> | |
%542 = torch.aten.where.self %539, %540, %541 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%543 = torch.aten.item %542 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_147 = torch.constant.int 3 | |
%544 = torch.aten.select.int %513, %int0_140, %int3_147 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%545 = torch.aten.item %544 : !torch.vtensor<[1],si64> -> !torch.int | |
%546 = torch.aten.eq.int %545, %int0_140 : !torch.int, !torch.int -> !torch.bool | |
%547 = torch.aten.Int.bool %546 : !torch.bool -> !torch.int | |
%int4_148 = torch.constant.int 4 | |
%548 = torch.aten.select.int %513, %int0_140, %int4_148 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%549 = torch.aten.item %548 : !torch.vtensor<[1],si64> -> !torch.int | |
%550 = torch.aten.eq.int %549, %int0_140 : !torch.int, !torch.int -> !torch.bool | |
%551 = torch.aten.Int.bool %550 : !torch.bool -> !torch.int | |
%552 = torch.prim.ListConstruct %523, %533, %543, %545, %549 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%553 = torch.aten.reshape %512, %552 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%554 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_149 = torch.constant.int 0 | |
%int1_150 = torch.constant.int 1 | |
%int0_151 = torch.constant.int 0 | |
%555 = torch.aten.select.int %554, %int0_149, %int0_151 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%556 = torch.aten.item %555 : !torch.vtensor<[1],si64> -> !torch.int | |
%557 = torch.aten.lt.int %556, %int0_149 : !torch.int, !torch.int -> !torch.bool | |
%558 = torch.aten.Int.bool %557 : !torch.bool -> !torch.int | |
%559 = torch.aten.mul.int %558, %int1_150 : !torch.int, !torch.int -> !torch.int | |
%560 = torch.aten.add.int %556, %559 : !torch.int, !torch.int -> !torch.int | |
%int1_152 = torch.constant.int 1 | |
%561 = torch.aten.select.int %554, %int0_149, %int1_152 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%562 = torch.aten.item %561 : !torch.vtensor<[1],si64> -> !torch.int | |
%563 = torch.aten.lt.int %562, %int0_149 : !torch.int, !torch.int -> !torch.bool | |
%564 = torch.aten.Int.bool %563 : !torch.bool -> !torch.int | |
%565 = torch.aten.mul.int %564, %int1_150 : !torch.int, !torch.int -> !torch.int | |
%566 = torch.aten.add.int %562, %565 : !torch.int, !torch.int -> !torch.int | |
%int2_153 = torch.constant.int 2 | |
%567 = torch.aten.select.int %554, %int0_149, %int2_153 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%568 = torch.aten.item %567 : !torch.vtensor<[1],si64> -> !torch.int | |
%569 = torch.aten.lt.int %568, %int0_149 : !torch.int, !torch.int -> !torch.bool | |
%570 = torch.aten.Int.bool %569 : !torch.bool -> !torch.int | |
%571 = torch.aten.mul.int %570, %int1_150 : !torch.int, !torch.int -> !torch.int | |
%572 = torch.aten.add.int %568, %571 : !torch.int, !torch.int -> !torch.int | |
%573 = torch.prim.ListConstruct %560, %566, %572 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_154 = torch.constant.bool false | |
%none_155 = torch.constant.none | |
%574 = torch.aten.tensor %573, %none_155, %none_155, %false_154 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_156, %indices_157 = torch.aten.sort %574, %int0_149, %false_154 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_158 = torch.constant.int 0 | |
%575 = torch.aten.select.int %values_156, %int0_149, %int0_158 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%576 = torch.aten.item %575 : !torch.vtensor<[1],si64> -> !torch.int | |
%577 = torch.aten.unsqueeze %464, %576 : !torch.vtensor<[32],f32>, !torch.int -> !torch.vtensor | |
%int1_159 = torch.constant.int 1 | |
%578 = torch.aten.select.int %values_156, %int0_149, %int1_159 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%579 = torch.aten.item %578 : !torch.vtensor<[1],si64> -> !torch.int | |
%580 = torch.aten.unsqueeze %577, %579 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_160 = torch.constant.int 2 | |
%581 = torch.aten.select.int %values_156, %int0_149, %int2_160 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%582 = torch.aten.item %581 : !torch.vtensor<[1],si64> -> !torch.int | |
%583 = torch.aten.unsqueeze %580, %582 : !torch.vtensor, !torch.int -> !torch.vtensor<[32,1,1,1],f32> | |
%584 = torch.aten.mul.Tensor %553, %583 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[32,1,1,1],f32> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%585 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_161 = torch.constant.int 0 | |
%int1_162 = torch.constant.int 1 | |
%int0_163 = torch.constant.int 0 | |
%586 = torch.aten.select.int %585, %int0_161, %int0_163 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%587 = torch.aten.item %586 : !torch.vtensor<[1],si64> -> !torch.int | |
%588 = torch.aten.lt.int %587, %int0_161 : !torch.int, !torch.int -> !torch.bool | |
%589 = torch.aten.Int.bool %588 : !torch.bool -> !torch.int | |
%590 = torch.aten.mul.int %589, %int1_162 : !torch.int, !torch.int -> !torch.int | |
%591 = torch.aten.add.int %587, %590 : !torch.int, !torch.int -> !torch.int | |
%int1_164 = torch.constant.int 1 | |
%592 = torch.aten.select.int %585, %int0_161, %int1_164 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%593 = torch.aten.item %592 : !torch.vtensor<[1],si64> -> !torch.int | |
%594 = torch.aten.lt.int %593, %int0_161 : !torch.int, !torch.int -> !torch.bool | |
%595 = torch.aten.Int.bool %594 : !torch.bool -> !torch.int | |
%596 = torch.aten.mul.int %595, %int1_162 : !torch.int, !torch.int -> !torch.int | |
%597 = torch.aten.add.int %593, %596 : !torch.int, !torch.int -> !torch.int | |
%int2_165 = torch.constant.int 2 | |
%598 = torch.aten.select.int %585, %int0_161, %int2_165 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%599 = torch.aten.item %598 : !torch.vtensor<[1],si64> -> !torch.int | |
%600 = torch.aten.lt.int %599, %int0_161 : !torch.int, !torch.int -> !torch.bool | |
%601 = torch.aten.Int.bool %600 : !torch.bool -> !torch.int | |
%602 = torch.aten.mul.int %601, %int1_162 : !torch.int, !torch.int -> !torch.int | |
%603 = torch.aten.add.int %599, %602 : !torch.int, !torch.int -> !torch.int | |
%604 = torch.prim.ListConstruct %591, %597, %603 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_166 = torch.constant.bool false | |
%none_167 = torch.constant.none | |
%605 = torch.aten.tensor %604, %none_167, %none_167, %false_166 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_168, %indices_169 = torch.aten.sort %605, %int0_161, %false_166 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_170 = torch.constant.int 0 | |
%606 = torch.aten.select.int %values_168, %int0_161, %int0_170 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%607 = torch.aten.item %606 : !torch.vtensor<[1],si64> -> !torch.int | |
%608 = torch.aten.unsqueeze %476, %607 : !torch.vtensor<[32],f32>, !torch.int -> !torch.vtensor | |
%int1_171 = torch.constant.int 1 | |
%609 = torch.aten.select.int %values_168, %int0_161, %int1_171 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%610 = torch.aten.item %609 : !torch.vtensor<[1],si64> -> !torch.int | |
%611 = torch.aten.unsqueeze %608, %610 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_172 = torch.constant.int 2 | |
%612 = torch.aten.select.int %values_168, %int0_161, %int2_172 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%613 = torch.aten.item %612 : !torch.vtensor<[1],si64> -> !torch.int | |
%614 = torch.aten.unsqueeze %611, %613 : !torch.vtensor, !torch.int -> !torch.vtensor<[32,1,1,1],f32> | |
%int1_173 = torch.constant.int 1 | |
%615 = torch.aten.add.Tensor %584, %614, %int1_173 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.vtensor<[32,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,32,?,?,?],f32> | |
%616 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%617 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_174 = torch.constant.int 12 | |
%618 = torch.aten.item %616 : !torch.vtensor<[],f32> -> !torch.float | |
%619 = torch.aten.item %617 : !torch.vtensor<[],si8> -> !torch.int | |
%620 = torch.aten.quantize_per_tensor %615, %618, %619, %int12_174 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%621 = torch.aten.int_repr %620 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],si8> | |
%622 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%623 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%624 = torch.aten.item %622 : !torch.vtensor<[],f32> -> !torch.float | |
%625 = torch.aten.item %623 : !torch.vtensor<[],si8> -> !torch.int | |
%626 = torch.aten._make_per_tensor_quantized_tensor %621, %624, %625 : !torch.vtensor<[?,32,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%627 = torch.aten.dequantize.self %626 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%628 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%629 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_175 = torch.constant.int 12 | |
%630 = torch.aten.item %628 : !torch.vtensor<[],f32> -> !torch.float | |
%631 = torch.aten.item %629 : !torch.vtensor<[],si8> -> !torch.int | |
%632 = torch.aten.quantize_per_tensor %8, %630, %631, %int12_175 : !torch.vtensor<[32,32,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32,32,3,3,3],!torch.qint8> | |
%633 = torch.aten.int_repr %632 : !torch.vtensor<[32,32,3,3,3],!torch.qint8> -> !torch.vtensor<[32,32,3,3,3],si8> | |
%634 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%635 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%636 = torch.aten.item %634 : !torch.vtensor<[],f32> -> !torch.float | |
%637 = torch.aten.item %635 : !torch.vtensor<[],si8> -> !torch.int | |
%638 = torch.aten._make_per_tensor_quantized_tensor %633, %636, %637 : !torch.vtensor<[32,32,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[32,32,3,3,3],!torch.qint8> | |
%639 = torch.aten.dequantize.self %638 : !torch.vtensor<[32,32,3,3,3],!torch.qint8> -> !torch.vtensor<[32,32,3,3,3],f32> | |
%int1_176 = torch.constant.int 1 | |
%int1_177 = torch.constant.int 1 | |
%int1_178 = torch.constant.int 1 | |
%int1_179 = torch.constant.int 1 | |
%int1_180 = torch.constant.int 1 | |
%int1_181 = torch.constant.int 1 | |
%int1_182 = torch.constant.int 1 | |
%int1_183 = torch.constant.int 1 | |
%int1_184 = torch.constant.int 1 | |
%int0_185 = torch.constant.int 0 | |
%640 = torch.prim.ListConstruct %int1_176, %int1_177, %int1_178 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%641 = torch.prim.ListConstruct %int1_179, %int1_180, %int1_181 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%642 = torch.prim.ListConstruct %int1_182, %int1_183, %int1_184 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%643 = torch.prim.ListConstruct %int0_185, %int0_185 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_186 = torch.constant.bool false | |
%none_187 = torch.constant.none | |
%int1_188 = torch.constant.int 1 | |
%644 = torch.aten.convolution %627, %639, %none_187, %642, %640, %641, %false_186, %643, %int1_188 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.vtensor<[32,32,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,32,?,?,?],f32> | |
%645 = torch.aten.relu %644 : !torch.vtensor<[?,32,?,?,?],f32> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%646 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%647 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_189 = torch.constant.int 12 | |
%648 = torch.aten.item %646 : !torch.vtensor<[],f32> -> !torch.float | |
%649 = torch.aten.item %647 : !torch.vtensor<[],si8> -> !torch.int | |
%650 = torch.aten.quantize_per_tensor %645, %648, %649, %int12_189 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%651 = torch.aten.int_repr %650 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],si8> | |
%652 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%653 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%654 = torch.aten.item %652 : !torch.vtensor<[],f32> -> !torch.float | |
%655 = torch.aten.item %653 : !torch.vtensor<[],si8> -> !torch.int | |
%656 = torch.aten._make_per_tensor_quantized_tensor %651, %654, %655 : !torch.vtensor<[?,32,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%657 = torch.aten.dequantize.self %656 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%658 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%659 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_190 = torch.constant.int 12 | |
%660 = torch.aten.item %658 : !torch.vtensor<[],f32> -> !torch.float | |
%661 = torch.aten.item %659 : !torch.vtensor<[],si8> -> !torch.int | |
%662 = torch.aten.quantize_per_tensor %9, %660, %661, %int12_190 : !torch.vtensor<[32],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%663 = torch.aten.int_repr %662 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],si8> | |
%664 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%665 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%666 = torch.aten.item %664 : !torch.vtensor<[],f32> -> !torch.float | |
%667 = torch.aten.item %665 : !torch.vtensor<[],si8> -> !torch.int | |
%668 = torch.aten._make_per_tensor_quantized_tensor %663, %666, %667 : !torch.vtensor<[32],si8>, !torch.float, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%669 = torch.aten.dequantize.self %668 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],f32> | |
%670 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%671 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_191 = torch.constant.int 12 | |
%672 = torch.aten.item %670 : !torch.vtensor<[],f32> -> !torch.float | |
%673 = torch.aten.item %671 : !torch.vtensor<[],si8> -> !torch.int | |
%674 = torch.aten.quantize_per_tensor %10, %672, %673, %int12_191 : !torch.vtensor<[32],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%675 = torch.aten.int_repr %674 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],si8> | |
%676 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%677 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%678 = torch.aten.item %676 : !torch.vtensor<[],f32> -> !torch.float | |
%679 = torch.aten.item %677 : !torch.vtensor<[],si8> -> !torch.int | |
%680 = torch.aten._make_per_tensor_quantized_tensor %675, %678, %679 : !torch.vtensor<[32],si8>, !torch.float, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%681 = torch.aten.dequantize.self %680 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],f32> | |
%682 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_192 = torch.constant.int 0 | |
%int0_193 = torch.constant.int 0 | |
%683 = torch.aten.select.int %682, %int0_192, %int0_193 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%684 = torch.aten.item %683 : !torch.vtensor<[1],si64> -> !torch.int | |
%685 = torch.aten.eq.int %684, %int0_192 : !torch.int, !torch.int -> !torch.bool | |
%686 = torch.aten.Int.bool %685 : !torch.bool -> !torch.int | |
%int0_194 = torch.constant.int 0 | |
%687 = torch.aten.size.int %657, %int0_194 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%688 = torch.prim.NumToTensor.Scalar %686 : !torch.int -> !torch.vtensor<[],i1> | |
%689 = torch.prim.NumToTensor.Scalar %687 : !torch.int -> !torch.vtensor<[],si64> | |
%690 = torch.prim.NumToTensor.Scalar %684 : !torch.int -> !torch.vtensor<[],si64> | |
%691 = torch.aten.where.self %688, %689, %690 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%692 = torch.aten.item %691 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_195 = torch.constant.int 1 | |
%693 = torch.aten.select.int %682, %int0_192, %int1_195 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%694 = torch.aten.item %693 : !torch.vtensor<[1],si64> -> !torch.int | |
%695 = torch.aten.eq.int %694, %int0_192 : !torch.int, !torch.int -> !torch.bool | |
%696 = torch.aten.Int.bool %695 : !torch.bool -> !torch.int | |
%int1_196 = torch.constant.int 1 | |
%697 = torch.aten.size.int %657, %int1_196 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%698 = torch.prim.NumToTensor.Scalar %696 : !torch.int -> !torch.vtensor<[],i1> | |
%699 = torch.prim.NumToTensor.Scalar %697 : !torch.int -> !torch.vtensor<[],si64> | |
%700 = torch.prim.NumToTensor.Scalar %694 : !torch.int -> !torch.vtensor<[],si64> | |
%701 = torch.aten.where.self %698, %699, %700 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%702 = torch.aten.item %701 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_197 = torch.constant.int 2 | |
%703 = torch.aten.select.int %682, %int0_192, %int2_197 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%704 = torch.aten.item %703 : !torch.vtensor<[1],si64> -> !torch.int | |
%705 = torch.aten.eq.int %704, %int0_192 : !torch.int, !torch.int -> !torch.bool | |
%706 = torch.aten.Int.bool %705 : !torch.bool -> !torch.int | |
%int2_198 = torch.constant.int 2 | |
%707 = torch.aten.size.int %657, %int2_198 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%708 = torch.prim.NumToTensor.Scalar %706 : !torch.int -> !torch.vtensor<[],i1> | |
%709 = torch.prim.NumToTensor.Scalar %707 : !torch.int -> !torch.vtensor<[],si64> | |
%710 = torch.prim.NumToTensor.Scalar %704 : !torch.int -> !torch.vtensor<[],si64> | |
%711 = torch.aten.where.self %708, %709, %710 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%712 = torch.aten.item %711 : !torch.vtensor<[],si64> -> !torch.int | |
%713 = torch.prim.ListConstruct %692, %702, %712 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%714 = torch.aten.reshape %657, %713 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%715 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%716 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_199 = torch.constant.none | |
%true_200 = torch.constant.bool true | |
%false_201 = torch.constant.bool false | |
%float9.999990e-06_202 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_203 = torch.constant.float 0.000000e+00 | |
%717 = torch.aten.instance_norm %714, %715, %716, %none_199, %none_199, %true_200, %float0.000000e00_203, %float9.999990e-06_202, %false_201 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%718 = torch.aten._shape_as_tensor %657 : !torch.vtensor<[?,32,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_204 = torch.constant.int 0 | |
%int0_205 = torch.constant.int 0 | |
%719 = torch.aten.select.int %718, %int0_204, %int0_205 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%720 = torch.aten.item %719 : !torch.vtensor<[1],si64> -> !torch.int | |
%721 = torch.aten.eq.int %720, %int0_204 : !torch.int, !torch.int -> !torch.bool | |
%722 = torch.aten.Int.bool %721 : !torch.bool -> !torch.int | |
%int0_206 = torch.constant.int 0 | |
%723 = torch.aten.size.int %717, %int0_206 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%724 = torch.prim.NumToTensor.Scalar %722 : !torch.int -> !torch.vtensor<[],i1> | |
%725 = torch.prim.NumToTensor.Scalar %723 : !torch.int -> !torch.vtensor<[],si64> | |
%726 = torch.prim.NumToTensor.Scalar %720 : !torch.int -> !torch.vtensor<[],si64> | |
%727 = torch.aten.where.self %724, %725, %726 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%728 = torch.aten.item %727 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_207 = torch.constant.int 1 | |
%729 = torch.aten.select.int %718, %int0_204, %int1_207 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%730 = torch.aten.item %729 : !torch.vtensor<[1],si64> -> !torch.int | |
%731 = torch.aten.eq.int %730, %int0_204 : !torch.int, !torch.int -> !torch.bool | |
%732 = torch.aten.Int.bool %731 : !torch.bool -> !torch.int | |
%int1_208 = torch.constant.int 1 | |
%733 = torch.aten.size.int %717, %int1_208 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%734 = torch.prim.NumToTensor.Scalar %732 : !torch.int -> !torch.vtensor<[],i1> | |
%735 = torch.prim.NumToTensor.Scalar %733 : !torch.int -> !torch.vtensor<[],si64> | |
%736 = torch.prim.NumToTensor.Scalar %730 : !torch.int -> !torch.vtensor<[],si64> | |
%737 = torch.aten.where.self %734, %735, %736 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%738 = torch.aten.item %737 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_209 = torch.constant.int 2 | |
%739 = torch.aten.select.int %718, %int0_204, %int2_209 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%740 = torch.aten.item %739 : !torch.vtensor<[1],si64> -> !torch.int | |
%741 = torch.aten.eq.int %740, %int0_204 : !torch.int, !torch.int -> !torch.bool | |
%742 = torch.aten.Int.bool %741 : !torch.bool -> !torch.int | |
%int2_210 = torch.constant.int 2 | |
%743 = torch.aten.size.int %717, %int2_210 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%744 = torch.prim.NumToTensor.Scalar %742 : !torch.int -> !torch.vtensor<[],i1> | |
%745 = torch.prim.NumToTensor.Scalar %743 : !torch.int -> !torch.vtensor<[],si64> | |
%746 = torch.prim.NumToTensor.Scalar %740 : !torch.int -> !torch.vtensor<[],si64> | |
%747 = torch.aten.where.self %744, %745, %746 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%748 = torch.aten.item %747 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_211 = torch.constant.int 3 | |
%749 = torch.aten.select.int %718, %int0_204, %int3_211 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%750 = torch.aten.item %749 : !torch.vtensor<[1],si64> -> !torch.int | |
%751 = torch.aten.eq.int %750, %int0_204 : !torch.int, !torch.int -> !torch.bool | |
%752 = torch.aten.Int.bool %751 : !torch.bool -> !torch.int | |
%int4_212 = torch.constant.int 4 | |
%753 = torch.aten.select.int %718, %int0_204, %int4_212 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%754 = torch.aten.item %753 : !torch.vtensor<[1],si64> -> !torch.int | |
%755 = torch.aten.eq.int %754, %int0_204 : !torch.int, !torch.int -> !torch.bool | |
%756 = torch.aten.Int.bool %755 : !torch.bool -> !torch.int | |
%757 = torch.prim.ListConstruct %728, %738, %748, %750, %754 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%758 = torch.aten.reshape %717, %757 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%759 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_213 = torch.constant.int 0 | |
%int1_214 = torch.constant.int 1 | |
%int0_215 = torch.constant.int 0 | |
%760 = torch.aten.select.int %759, %int0_213, %int0_215 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%761 = torch.aten.item %760 : !torch.vtensor<[1],si64> -> !torch.int | |
%762 = torch.aten.lt.int %761, %int0_213 : !torch.int, !torch.int -> !torch.bool | |
%763 = torch.aten.Int.bool %762 : !torch.bool -> !torch.int | |
%764 = torch.aten.mul.int %763, %int1_214 : !torch.int, !torch.int -> !torch.int | |
%765 = torch.aten.add.int %761, %764 : !torch.int, !torch.int -> !torch.int | |
%int1_216 = torch.constant.int 1 | |
%766 = torch.aten.select.int %759, %int0_213, %int1_216 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%767 = torch.aten.item %766 : !torch.vtensor<[1],si64> -> !torch.int | |
%768 = torch.aten.lt.int %767, %int0_213 : !torch.int, !torch.int -> !torch.bool | |
%769 = torch.aten.Int.bool %768 : !torch.bool -> !torch.int | |
%770 = torch.aten.mul.int %769, %int1_214 : !torch.int, !torch.int -> !torch.int | |
%771 = torch.aten.add.int %767, %770 : !torch.int, !torch.int -> !torch.int | |
%int2_217 = torch.constant.int 2 | |
%772 = torch.aten.select.int %759, %int0_213, %int2_217 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%773 = torch.aten.item %772 : !torch.vtensor<[1],si64> -> !torch.int | |
%774 = torch.aten.lt.int %773, %int0_213 : !torch.int, !torch.int -> !torch.bool | |
%775 = torch.aten.Int.bool %774 : !torch.bool -> !torch.int | |
%776 = torch.aten.mul.int %775, %int1_214 : !torch.int, !torch.int -> !torch.int | |
%777 = torch.aten.add.int %773, %776 : !torch.int, !torch.int -> !torch.int | |
%778 = torch.prim.ListConstruct %765, %771, %777 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_218 = torch.constant.bool false | |
%none_219 = torch.constant.none | |
%779 = torch.aten.tensor %778, %none_219, %none_219, %false_218 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_220, %indices_221 = torch.aten.sort %779, %int0_213, %false_218 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_222 = torch.constant.int 0 | |
%780 = torch.aten.select.int %values_220, %int0_213, %int0_222 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%781 = torch.aten.item %780 : !torch.vtensor<[1],si64> -> !torch.int | |
%782 = torch.aten.unsqueeze %669, %781 : !torch.vtensor<[32],f32>, !torch.int -> !torch.vtensor | |
%int1_223 = torch.constant.int 1 | |
%783 = torch.aten.select.int %values_220, %int0_213, %int1_223 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%784 = torch.aten.item %783 : !torch.vtensor<[1],si64> -> !torch.int | |
%785 = torch.aten.unsqueeze %782, %784 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_224 = torch.constant.int 2 | |
%786 = torch.aten.select.int %values_220, %int0_213, %int2_224 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%787 = torch.aten.item %786 : !torch.vtensor<[1],si64> -> !torch.int | |
%788 = torch.aten.unsqueeze %785, %787 : !torch.vtensor, !torch.int -> !torch.vtensor<[32,1,1,1],f32> | |
%789 = torch.aten.mul.Tensor %758, %788 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[32,1,1,1],f32> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%790 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_225 = torch.constant.int 0 | |
%int1_226 = torch.constant.int 1 | |
%int0_227 = torch.constant.int 0 | |
%791 = torch.aten.select.int %790, %int0_225, %int0_227 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%792 = torch.aten.item %791 : !torch.vtensor<[1],si64> -> !torch.int | |
%793 = torch.aten.lt.int %792, %int0_225 : !torch.int, !torch.int -> !torch.bool | |
%794 = torch.aten.Int.bool %793 : !torch.bool -> !torch.int | |
%795 = torch.aten.mul.int %794, %int1_226 : !torch.int, !torch.int -> !torch.int | |
%796 = torch.aten.add.int %792, %795 : !torch.int, !torch.int -> !torch.int | |
%int1_228 = torch.constant.int 1 | |
%797 = torch.aten.select.int %790, %int0_225, %int1_228 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%798 = torch.aten.item %797 : !torch.vtensor<[1],si64> -> !torch.int | |
%799 = torch.aten.lt.int %798, %int0_225 : !torch.int, !torch.int -> !torch.bool | |
%800 = torch.aten.Int.bool %799 : !torch.bool -> !torch.int | |
%801 = torch.aten.mul.int %800, %int1_226 : !torch.int, !torch.int -> !torch.int | |
%802 = torch.aten.add.int %798, %801 : !torch.int, !torch.int -> !torch.int | |
%int2_229 = torch.constant.int 2 | |
%803 = torch.aten.select.int %790, %int0_225, %int2_229 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%804 = torch.aten.item %803 : !torch.vtensor<[1],si64> -> !torch.int | |
%805 = torch.aten.lt.int %804, %int0_225 : !torch.int, !torch.int -> !torch.bool | |
%806 = torch.aten.Int.bool %805 : !torch.bool -> !torch.int | |
%807 = torch.aten.mul.int %806, %int1_226 : !torch.int, !torch.int -> !torch.int | |
%808 = torch.aten.add.int %804, %807 : !torch.int, !torch.int -> !torch.int | |
%809 = torch.prim.ListConstruct %796, %802, %808 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_230 = torch.constant.bool false | |
%none_231 = torch.constant.none | |
%810 = torch.aten.tensor %809, %none_231, %none_231, %false_230 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_232, %indices_233 = torch.aten.sort %810, %int0_225, %false_230 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_234 = torch.constant.int 0 | |
%811 = torch.aten.select.int %values_232, %int0_225, %int0_234 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%812 = torch.aten.item %811 : !torch.vtensor<[1],si64> -> !torch.int | |
%813 = torch.aten.unsqueeze %681, %812 : !torch.vtensor<[32],f32>, !torch.int -> !torch.vtensor | |
%int1_235 = torch.constant.int 1 | |
%814 = torch.aten.select.int %values_232, %int0_225, %int1_235 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%815 = torch.aten.item %814 : !torch.vtensor<[1],si64> -> !torch.int | |
%816 = torch.aten.unsqueeze %813, %815 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_236 = torch.constant.int 2 | |
%817 = torch.aten.select.int %values_232, %int0_225, %int2_236 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%818 = torch.aten.item %817 : !torch.vtensor<[1],si64> -> !torch.int | |
%819 = torch.aten.unsqueeze %816, %818 : !torch.vtensor, !torch.int -> !torch.vtensor<[32,1,1,1],f32> | |
%int1_237 = torch.constant.int 1 | |
%820 = torch.aten.add.Tensor %789, %819, %int1_237 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.vtensor<[32,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,32,?,?,?],f32> | |
%821 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%822 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_238 = torch.constant.int 12 | |
%823 = torch.aten.item %821 : !torch.vtensor<[],f32> -> !torch.float | |
%824 = torch.aten.item %822 : !torch.vtensor<[],si8> -> !torch.int | |
%825 = torch.aten.quantize_per_tensor %820, %823, %824, %int12_238 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%826 = torch.aten.int_repr %825 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],si8> | |
%827 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%828 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%829 = torch.aten.item %827 : !torch.vtensor<[],f32> -> !torch.float | |
%830 = torch.aten.item %828 : !torch.vtensor<[],si8> -> !torch.int | |
%831 = torch.aten._make_per_tensor_quantized_tensor %826, %829, %830 : !torch.vtensor<[?,32,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%832 = torch.aten.dequantize.self %831 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%833 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%834 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_239 = torch.constant.int 12 | |
%835 = torch.aten.item %833 : !torch.vtensor<[],f32> -> !torch.float | |
%836 = torch.aten.item %834 : !torch.vtensor<[],si8> -> !torch.int | |
%837 = torch.aten.quantize_per_tensor %11, %835, %836, %int12_239 : !torch.vtensor<[64,32,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64,32,3,3,3],!torch.qint8> | |
%838 = torch.aten.int_repr %837 : !torch.vtensor<[64,32,3,3,3],!torch.qint8> -> !torch.vtensor<[64,32,3,3,3],si8> | |
%839 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%840 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%841 = torch.aten.item %839 : !torch.vtensor<[],f32> -> !torch.float | |
%842 = torch.aten.item %840 : !torch.vtensor<[],si8> -> !torch.int | |
%843 = torch.aten._make_per_tensor_quantized_tensor %838, %841, %842 : !torch.vtensor<[64,32,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[64,32,3,3,3],!torch.qint8> | |
%844 = torch.aten.dequantize.self %843 : !torch.vtensor<[64,32,3,3,3],!torch.qint8> -> !torch.vtensor<[64,32,3,3,3],f32> | |
%int1_240 = torch.constant.int 1 | |
%int1_241 = torch.constant.int 1 | |
%int1_242 = torch.constant.int 1 | |
%int1_243 = torch.constant.int 1 | |
%int1_244 = torch.constant.int 1 | |
%int1_245 = torch.constant.int 1 | |
%int1_246 = torch.constant.int 1 | |
%int1_247 = torch.constant.int 1 | |
%int1_248 = torch.constant.int 1 | |
%int0_249 = torch.constant.int 0 | |
%845 = torch.prim.ListConstruct %int1_240, %int1_241, %int1_242 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%846 = torch.prim.ListConstruct %int1_243, %int1_244, %int1_245 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%847 = torch.prim.ListConstruct %int1_246, %int1_247, %int1_248 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%848 = torch.prim.ListConstruct %int0_249, %int0_249 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_250 = torch.constant.bool false | |
%none_251 = torch.constant.none | |
%int1_252 = torch.constant.int 1 | |
%849 = torch.aten.convolution %832, %844, %none_251, %847, %845, %846, %false_250, %848, %int1_252 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.vtensor<[64,32,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,64,?,?,?],f32> | |
%850 = torch.aten.relu %849 : !torch.vtensor<[?,64,?,?,?],f32> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%851 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%852 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_253 = torch.constant.int 12 | |
%853 = torch.aten.item %851 : !torch.vtensor<[],f32> -> !torch.float | |
%854 = torch.aten.item %852 : !torch.vtensor<[],si8> -> !torch.int | |
%855 = torch.aten.quantize_per_tensor %850, %853, %854, %int12_253 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%856 = torch.aten.int_repr %855 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],si8> | |
%857 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%858 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%859 = torch.aten.item %857 : !torch.vtensor<[],f32> -> !torch.float | |
%860 = torch.aten.item %858 : !torch.vtensor<[],si8> -> !torch.int | |
%861 = torch.aten._make_per_tensor_quantized_tensor %856, %859, %860 : !torch.vtensor<[?,64,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%862 = torch.aten.dequantize.self %861 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%int2_254 = torch.constant.int 2 | |
%int2_255 = torch.constant.int 2 | |
%int2_256 = torch.constant.int 2 | |
%863 = torch.prim.ListConstruct %int2_254, %int2_255, %int2_256 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int0_257 = torch.constant.int 0 | |
%int0_258 = torch.constant.int 0 | |
%int0_259 = torch.constant.int 0 | |
%864 = torch.prim.ListConstruct %int0_257, %int0_258, %int0_259 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int2_260 = torch.constant.int 2 | |
%int2_261 = torch.constant.int 2 | |
%int2_262 = torch.constant.int 2 | |
%865 = torch.prim.ListConstruct %int2_260, %int2_261, %int2_262 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int1_263 = torch.constant.int 1 | |
%int1_264 = torch.constant.int 1 | |
%int1_265 = torch.constant.int 1 | |
%866 = torch.prim.ListConstruct %int1_263, %int1_264, %int1_265 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_266 = torch.constant.bool false | |
%867 = torch.aten.max_pool3d %862, %863, %865, %864, %866, %false_266 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.vtensor<[?,64,?,?,?],f32> | |
%868 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%869 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_267 = torch.constant.int 12 | |
%870 = torch.aten.item %868 : !torch.vtensor<[],f32> -> !torch.float | |
%871 = torch.aten.item %869 : !torch.vtensor<[],si8> -> !torch.int | |
%872 = torch.aten.quantize_per_tensor %867, %870, %871, %int12_267 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%873 = torch.aten.int_repr %872 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],si8> | |
%874 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%875 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%876 = torch.aten.item %874 : !torch.vtensor<[],f32> -> !torch.float | |
%877 = torch.aten.item %875 : !torch.vtensor<[],si8> -> !torch.int | |
%878 = torch.aten._make_per_tensor_quantized_tensor %873, %876, %877 : !torch.vtensor<[?,64,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%879 = torch.aten.dequantize.self %878 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%880 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%881 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_268 = torch.constant.int 12 | |
%882 = torch.aten.item %880 : !torch.vtensor<[],f32> -> !torch.float | |
%883 = torch.aten.item %881 : !torch.vtensor<[],si8> -> !torch.int | |
%884 = torch.aten.quantize_per_tensor %12, %882, %883, %int12_268 : !torch.vtensor<[64],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%885 = torch.aten.int_repr %884 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],si8> | |
%886 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%887 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%888 = torch.aten.item %886 : !torch.vtensor<[],f32> -> !torch.float | |
%889 = torch.aten.item %887 : !torch.vtensor<[],si8> -> !torch.int | |
%890 = torch.aten._make_per_tensor_quantized_tensor %885, %888, %889 : !torch.vtensor<[64],si8>, !torch.float, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%891 = torch.aten.dequantize.self %890 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],f32> | |
%892 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%893 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_269 = torch.constant.int 12 | |
%894 = torch.aten.item %892 : !torch.vtensor<[],f32> -> !torch.float | |
%895 = torch.aten.item %893 : !torch.vtensor<[],si8> -> !torch.int | |
%896 = torch.aten.quantize_per_tensor %13, %894, %895, %int12_269 : !torch.vtensor<[64],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%897 = torch.aten.int_repr %896 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],si8> | |
%898 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%899 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%900 = torch.aten.item %898 : !torch.vtensor<[],f32> -> !torch.float | |
%901 = torch.aten.item %899 : !torch.vtensor<[],si8> -> !torch.int | |
%902 = torch.aten._make_per_tensor_quantized_tensor %897, %900, %901 : !torch.vtensor<[64],si8>, !torch.float, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%903 = torch.aten.dequantize.self %902 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],f32> | |
%904 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_270 = torch.constant.int 0 | |
%int0_271 = torch.constant.int 0 | |
%905 = torch.aten.select.int %904, %int0_270, %int0_271 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%906 = torch.aten.item %905 : !torch.vtensor<[1],si64> -> !torch.int | |
%907 = torch.aten.eq.int %906, %int0_270 : !torch.int, !torch.int -> !torch.bool | |
%908 = torch.aten.Int.bool %907 : !torch.bool -> !torch.int | |
%int0_272 = torch.constant.int 0 | |
%909 = torch.aten.size.int %879, %int0_272 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%910 = torch.prim.NumToTensor.Scalar %908 : !torch.int -> !torch.vtensor<[],i1> | |
%911 = torch.prim.NumToTensor.Scalar %909 : !torch.int -> !torch.vtensor<[],si64> | |
%912 = torch.prim.NumToTensor.Scalar %906 : !torch.int -> !torch.vtensor<[],si64> | |
%913 = torch.aten.where.self %910, %911, %912 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%914 = torch.aten.item %913 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_273 = torch.constant.int 1 | |
%915 = torch.aten.select.int %904, %int0_270, %int1_273 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%916 = torch.aten.item %915 : !torch.vtensor<[1],si64> -> !torch.int | |
%917 = torch.aten.eq.int %916, %int0_270 : !torch.int, !torch.int -> !torch.bool | |
%918 = torch.aten.Int.bool %917 : !torch.bool -> !torch.int | |
%int1_274 = torch.constant.int 1 | |
%919 = torch.aten.size.int %879, %int1_274 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%920 = torch.prim.NumToTensor.Scalar %918 : !torch.int -> !torch.vtensor<[],i1> | |
%921 = torch.prim.NumToTensor.Scalar %919 : !torch.int -> !torch.vtensor<[],si64> | |
%922 = torch.prim.NumToTensor.Scalar %916 : !torch.int -> !torch.vtensor<[],si64> | |
%923 = torch.aten.where.self %920, %921, %922 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%924 = torch.aten.item %923 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_275 = torch.constant.int 2 | |
%925 = torch.aten.select.int %904, %int0_270, %int2_275 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%926 = torch.aten.item %925 : !torch.vtensor<[1],si64> -> !torch.int | |
%927 = torch.aten.eq.int %926, %int0_270 : !torch.int, !torch.int -> !torch.bool | |
%928 = torch.aten.Int.bool %927 : !torch.bool -> !torch.int | |
%int2_276 = torch.constant.int 2 | |
%929 = torch.aten.size.int %879, %int2_276 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%930 = torch.prim.NumToTensor.Scalar %928 : !torch.int -> !torch.vtensor<[],i1> | |
%931 = torch.prim.NumToTensor.Scalar %929 : !torch.int -> !torch.vtensor<[],si64> | |
%932 = torch.prim.NumToTensor.Scalar %926 : !torch.int -> !torch.vtensor<[],si64> | |
%933 = torch.aten.where.self %930, %931, %932 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%934 = torch.aten.item %933 : !torch.vtensor<[],si64> -> !torch.int | |
%935 = torch.prim.ListConstruct %914, %924, %934 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%936 = torch.aten.reshape %879, %935 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%937 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%938 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_277 = torch.constant.none | |
%true_278 = torch.constant.bool true | |
%false_279 = torch.constant.bool false | |
%float9.999990e-06_280 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_281 = torch.constant.float 0.000000e+00 | |
%939 = torch.aten.instance_norm %936, %937, %938, %none_277, %none_277, %true_278, %float0.000000e00_281, %float9.999990e-06_280, %false_279 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%940 = torch.aten._shape_as_tensor %879 : !torch.vtensor<[?,64,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_282 = torch.constant.int 0 | |
%int0_283 = torch.constant.int 0 | |
%941 = torch.aten.select.int %940, %int0_282, %int0_283 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%942 = torch.aten.item %941 : !torch.vtensor<[1],si64> -> !torch.int | |
%943 = torch.aten.eq.int %942, %int0_282 : !torch.int, !torch.int -> !torch.bool | |
%944 = torch.aten.Int.bool %943 : !torch.bool -> !torch.int | |
%int0_284 = torch.constant.int 0 | |
%945 = torch.aten.size.int %939, %int0_284 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%946 = torch.prim.NumToTensor.Scalar %944 : !torch.int -> !torch.vtensor<[],i1> | |
%947 = torch.prim.NumToTensor.Scalar %945 : !torch.int -> !torch.vtensor<[],si64> | |
%948 = torch.prim.NumToTensor.Scalar %942 : !torch.int -> !torch.vtensor<[],si64> | |
%949 = torch.aten.where.self %946, %947, %948 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%950 = torch.aten.item %949 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_285 = torch.constant.int 1 | |
%951 = torch.aten.select.int %940, %int0_282, %int1_285 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%952 = torch.aten.item %951 : !torch.vtensor<[1],si64> -> !torch.int | |
%953 = torch.aten.eq.int %952, %int0_282 : !torch.int, !torch.int -> !torch.bool | |
%954 = torch.aten.Int.bool %953 : !torch.bool -> !torch.int | |
%int1_286 = torch.constant.int 1 | |
%955 = torch.aten.size.int %939, %int1_286 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%956 = torch.prim.NumToTensor.Scalar %954 : !torch.int -> !torch.vtensor<[],i1> | |
%957 = torch.prim.NumToTensor.Scalar %955 : !torch.int -> !torch.vtensor<[],si64> | |
%958 = torch.prim.NumToTensor.Scalar %952 : !torch.int -> !torch.vtensor<[],si64> | |
%959 = torch.aten.where.self %956, %957, %958 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%960 = torch.aten.item %959 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_287 = torch.constant.int 2 | |
%961 = torch.aten.select.int %940, %int0_282, %int2_287 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%962 = torch.aten.item %961 : !torch.vtensor<[1],si64> -> !torch.int | |
%963 = torch.aten.eq.int %962, %int0_282 : !torch.int, !torch.int -> !torch.bool | |
%964 = torch.aten.Int.bool %963 : !torch.bool -> !torch.int | |
%int2_288 = torch.constant.int 2 | |
%965 = torch.aten.size.int %939, %int2_288 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%966 = torch.prim.NumToTensor.Scalar %964 : !torch.int -> !torch.vtensor<[],i1> | |
%967 = torch.prim.NumToTensor.Scalar %965 : !torch.int -> !torch.vtensor<[],si64> | |
%968 = torch.prim.NumToTensor.Scalar %962 : !torch.int -> !torch.vtensor<[],si64> | |
%969 = torch.aten.where.self %966, %967, %968 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%970 = torch.aten.item %969 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_289 = torch.constant.int 3 | |
%971 = torch.aten.select.int %940, %int0_282, %int3_289 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%972 = torch.aten.item %971 : !torch.vtensor<[1],si64> -> !torch.int | |
%973 = torch.aten.eq.int %972, %int0_282 : !torch.int, !torch.int -> !torch.bool | |
%974 = torch.aten.Int.bool %973 : !torch.bool -> !torch.int | |
%int4_290 = torch.constant.int 4 | |
%975 = torch.aten.select.int %940, %int0_282, %int4_290 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%976 = torch.aten.item %975 : !torch.vtensor<[1],si64> -> !torch.int | |
%977 = torch.aten.eq.int %976, %int0_282 : !torch.int, !torch.int -> !torch.bool | |
%978 = torch.aten.Int.bool %977 : !torch.bool -> !torch.int | |
%979 = torch.prim.ListConstruct %950, %960, %970, %972, %976 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%980 = torch.aten.reshape %939, %979 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%981 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_291 = torch.constant.int 0 | |
%int1_292 = torch.constant.int 1 | |
%int0_293 = torch.constant.int 0 | |
%982 = torch.aten.select.int %981, %int0_291, %int0_293 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%983 = torch.aten.item %982 : !torch.vtensor<[1],si64> -> !torch.int | |
%984 = torch.aten.lt.int %983, %int0_291 : !torch.int, !torch.int -> !torch.bool | |
%985 = torch.aten.Int.bool %984 : !torch.bool -> !torch.int | |
%986 = torch.aten.mul.int %985, %int1_292 : !torch.int, !torch.int -> !torch.int | |
%987 = torch.aten.add.int %983, %986 : !torch.int, !torch.int -> !torch.int | |
%int1_294 = torch.constant.int 1 | |
%988 = torch.aten.select.int %981, %int0_291, %int1_294 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%989 = torch.aten.item %988 : !torch.vtensor<[1],si64> -> !torch.int | |
%990 = torch.aten.lt.int %989, %int0_291 : !torch.int, !torch.int -> !torch.bool | |
%991 = torch.aten.Int.bool %990 : !torch.bool -> !torch.int | |
%992 = torch.aten.mul.int %991, %int1_292 : !torch.int, !torch.int -> !torch.int | |
%993 = torch.aten.add.int %989, %992 : !torch.int, !torch.int -> !torch.int | |
%int2_295 = torch.constant.int 2 | |
%994 = torch.aten.select.int %981, %int0_291, %int2_295 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%995 = torch.aten.item %994 : !torch.vtensor<[1],si64> -> !torch.int | |
%996 = torch.aten.lt.int %995, %int0_291 : !torch.int, !torch.int -> !torch.bool | |
%997 = torch.aten.Int.bool %996 : !torch.bool -> !torch.int | |
%998 = torch.aten.mul.int %997, %int1_292 : !torch.int, !torch.int -> !torch.int | |
%999 = torch.aten.add.int %995, %998 : !torch.int, !torch.int -> !torch.int | |
%1000 = torch.prim.ListConstruct %987, %993, %999 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_296 = torch.constant.bool false | |
%none_297 = torch.constant.none | |
%1001 = torch.aten.tensor %1000, %none_297, %none_297, %false_296 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_298, %indices_299 = torch.aten.sort %1001, %int0_291, %false_296 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_300 = torch.constant.int 0 | |
%1002 = torch.aten.select.int %values_298, %int0_291, %int0_300 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1003 = torch.aten.item %1002 : !torch.vtensor<[1],si64> -> !torch.int | |
%1004 = torch.aten.unsqueeze %891, %1003 : !torch.vtensor<[64],f32>, !torch.int -> !torch.vtensor | |
%int1_301 = torch.constant.int 1 | |
%1005 = torch.aten.select.int %values_298, %int0_291, %int1_301 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1006 = torch.aten.item %1005 : !torch.vtensor<[1],si64> -> !torch.int | |
%1007 = torch.aten.unsqueeze %1004, %1006 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_302 = torch.constant.int 2 | |
%1008 = torch.aten.select.int %values_298, %int0_291, %int2_302 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1009 = torch.aten.item %1008 : !torch.vtensor<[1],si64> -> !torch.int | |
%1010 = torch.aten.unsqueeze %1007, %1009 : !torch.vtensor, !torch.int -> !torch.vtensor<[64,1,1,1],f32> | |
%1011 = torch.aten.mul.Tensor %980, %1010 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[64,1,1,1],f32> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1012 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_303 = torch.constant.int 0 | |
%int1_304 = torch.constant.int 1 | |
%int0_305 = torch.constant.int 0 | |
%1013 = torch.aten.select.int %1012, %int0_303, %int0_305 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1014 = torch.aten.item %1013 : !torch.vtensor<[1],si64> -> !torch.int | |
%1015 = torch.aten.lt.int %1014, %int0_303 : !torch.int, !torch.int -> !torch.bool | |
%1016 = torch.aten.Int.bool %1015 : !torch.bool -> !torch.int | |
%1017 = torch.aten.mul.int %1016, %int1_304 : !torch.int, !torch.int -> !torch.int | |
%1018 = torch.aten.add.int %1014, %1017 : !torch.int, !torch.int -> !torch.int | |
%int1_306 = torch.constant.int 1 | |
%1019 = torch.aten.select.int %1012, %int0_303, %int1_306 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1020 = torch.aten.item %1019 : !torch.vtensor<[1],si64> -> !torch.int | |
%1021 = torch.aten.lt.int %1020, %int0_303 : !torch.int, !torch.int -> !torch.bool | |
%1022 = torch.aten.Int.bool %1021 : !torch.bool -> !torch.int | |
%1023 = torch.aten.mul.int %1022, %int1_304 : !torch.int, !torch.int -> !torch.int | |
%1024 = torch.aten.add.int %1020, %1023 : !torch.int, !torch.int -> !torch.int | |
%int2_307 = torch.constant.int 2 | |
%1025 = torch.aten.select.int %1012, %int0_303, %int2_307 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1026 = torch.aten.item %1025 : !torch.vtensor<[1],si64> -> !torch.int | |
%1027 = torch.aten.lt.int %1026, %int0_303 : !torch.int, !torch.int -> !torch.bool | |
%1028 = torch.aten.Int.bool %1027 : !torch.bool -> !torch.int | |
%1029 = torch.aten.mul.int %1028, %int1_304 : !torch.int, !torch.int -> !torch.int | |
%1030 = torch.aten.add.int %1026, %1029 : !torch.int, !torch.int -> !torch.int | |
%1031 = torch.prim.ListConstruct %1018, %1024, %1030 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_308 = torch.constant.bool false | |
%none_309 = torch.constant.none | |
%1032 = torch.aten.tensor %1031, %none_309, %none_309, %false_308 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_310, %indices_311 = torch.aten.sort %1032, %int0_303, %false_308 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_312 = torch.constant.int 0 | |
%1033 = torch.aten.select.int %values_310, %int0_303, %int0_312 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1034 = torch.aten.item %1033 : !torch.vtensor<[1],si64> -> !torch.int | |
%1035 = torch.aten.unsqueeze %903, %1034 : !torch.vtensor<[64],f32>, !torch.int -> !torch.vtensor | |
%int1_313 = torch.constant.int 1 | |
%1036 = torch.aten.select.int %values_310, %int0_303, %int1_313 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1037 = torch.aten.item %1036 : !torch.vtensor<[1],si64> -> !torch.int | |
%1038 = torch.aten.unsqueeze %1035, %1037 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_314 = torch.constant.int 2 | |
%1039 = torch.aten.select.int %values_310, %int0_303, %int2_314 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1040 = torch.aten.item %1039 : !torch.vtensor<[1],si64> -> !torch.int | |
%1041 = torch.aten.unsqueeze %1038, %1040 : !torch.vtensor, !torch.int -> !torch.vtensor<[64,1,1,1],f32> | |
%int1_315 = torch.constant.int 1 | |
%1042 = torch.aten.add.Tensor %1011, %1041, %int1_315 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.vtensor<[64,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1043 = torch.vtensor.literal(dense<3.125000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1044 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_316 = torch.constant.int 12 | |
%1045 = torch.aten.item %1043 : !torch.vtensor<[],f32> -> !torch.float | |
%1046 = torch.aten.item %1044 : !torch.vtensor<[],si8> -> !torch.int | |
%1047 = torch.aten.quantize_per_tensor %1042, %1045, %1046, %int12_316 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%1048 = torch.aten.int_repr %1047 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],si8> | |
%1049 = torch.vtensor.literal(dense<3.125000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1050 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1051 = torch.aten.item %1049 : !torch.vtensor<[],f32> -> !torch.float | |
%1052 = torch.aten.item %1050 : !torch.vtensor<[],si8> -> !torch.int | |
%1053 = torch.aten._make_per_tensor_quantized_tensor %1048, %1051, %1052 : !torch.vtensor<[?,64,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%1054 = torch.aten.dequantize.self %1053 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1055 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1056 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_317 = torch.constant.int 12 | |
%1057 = torch.aten.item %1055 : !torch.vtensor<[],f32> -> !torch.float | |
%1058 = torch.aten.item %1056 : !torch.vtensor<[],si8> -> !torch.int | |
%1059 = torch.aten.quantize_per_tensor %14, %1057, %1058, %int12_317 : !torch.vtensor<[64,64,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64,64,3,3,3],!torch.qint8> | |
%1060 = torch.aten.int_repr %1059 : !torch.vtensor<[64,64,3,3,3],!torch.qint8> -> !torch.vtensor<[64,64,3,3,3],si8> | |
%1061 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1062 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1063 = torch.aten.item %1061 : !torch.vtensor<[],f32> -> !torch.float | |
%1064 = torch.aten.item %1062 : !torch.vtensor<[],si8> -> !torch.int | |
%1065 = torch.aten._make_per_tensor_quantized_tensor %1060, %1063, %1064 : !torch.vtensor<[64,64,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[64,64,3,3,3],!torch.qint8> | |
%1066 = torch.aten.dequantize.self %1065 : !torch.vtensor<[64,64,3,3,3],!torch.qint8> -> !torch.vtensor<[64,64,3,3,3],f32> | |
%int1_318 = torch.constant.int 1 | |
%int1_319 = torch.constant.int 1 | |
%int1_320 = torch.constant.int 1 | |
%int1_321 = torch.constant.int 1 | |
%int1_322 = torch.constant.int 1 | |
%int1_323 = torch.constant.int 1 | |
%int1_324 = torch.constant.int 1 | |
%int1_325 = torch.constant.int 1 | |
%int1_326 = torch.constant.int 1 | |
%int0_327 = torch.constant.int 0 | |
%1067 = torch.prim.ListConstruct %int1_318, %int1_319, %int1_320 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1068 = torch.prim.ListConstruct %int1_321, %int1_322, %int1_323 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1069 = torch.prim.ListConstruct %int1_324, %int1_325, %int1_326 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1070 = torch.prim.ListConstruct %int0_327, %int0_327 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_328 = torch.constant.bool false | |
%none_329 = torch.constant.none | |
%int1_330 = torch.constant.int 1 | |
%1071 = torch.aten.convolution %1054, %1066, %none_329, %1069, %1067, %1068, %false_328, %1070, %int1_330 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.vtensor<[64,64,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1072 = torch.aten.relu %1071 : !torch.vtensor<[?,64,?,?,?],f32> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1073 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1074 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_331 = torch.constant.int 12 | |
%1075 = torch.aten.item %1073 : !torch.vtensor<[],f32> -> !torch.float | |
%1076 = torch.aten.item %1074 : !torch.vtensor<[],si8> -> !torch.int | |
%1077 = torch.aten.quantize_per_tensor %1072, %1075, %1076, %int12_331 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%1078 = torch.aten.int_repr %1077 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],si8> | |
%1079 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1080 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1081 = torch.aten.item %1079 : !torch.vtensor<[],f32> -> !torch.float | |
%1082 = torch.aten.item %1080 : !torch.vtensor<[],si8> -> !torch.int | |
%1083 = torch.aten._make_per_tensor_quantized_tensor %1078, %1081, %1082 : !torch.vtensor<[?,64,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%1084 = torch.aten.dequantize.self %1083 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1085 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1086 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_332 = torch.constant.int 12 | |
%1087 = torch.aten.item %1085 : !torch.vtensor<[],f32> -> !torch.float | |
%1088 = torch.aten.item %1086 : !torch.vtensor<[],si8> -> !torch.int | |
%1089 = torch.aten.quantize_per_tensor %15, %1087, %1088, %int12_332 : !torch.vtensor<[64],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%1090 = torch.aten.int_repr %1089 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],si8> | |
%1091 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1092 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1093 = torch.aten.item %1091 : !torch.vtensor<[],f32> -> !torch.float | |
%1094 = torch.aten.item %1092 : !torch.vtensor<[],si8> -> !torch.int | |
%1095 = torch.aten._make_per_tensor_quantized_tensor %1090, %1093, %1094 : !torch.vtensor<[64],si8>, !torch.float, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%1096 = torch.aten.dequantize.self %1095 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],f32> | |
%1097 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1098 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_333 = torch.constant.int 12 | |
%1099 = torch.aten.item %1097 : !torch.vtensor<[],f32> -> !torch.float | |
%1100 = torch.aten.item %1098 : !torch.vtensor<[],si8> -> !torch.int | |
%1101 = torch.aten.quantize_per_tensor %16, %1099, %1100, %int12_333 : !torch.vtensor<[64],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%1102 = torch.aten.int_repr %1101 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],si8> | |
%1103 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1104 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1105 = torch.aten.item %1103 : !torch.vtensor<[],f32> -> !torch.float | |
%1106 = torch.aten.item %1104 : !torch.vtensor<[],si8> -> !torch.int | |
%1107 = torch.aten._make_per_tensor_quantized_tensor %1102, %1105, %1106 : !torch.vtensor<[64],si8>, !torch.float, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%1108 = torch.aten.dequantize.self %1107 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],f32> | |
%1109 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_334 = torch.constant.int 0 | |
%int0_335 = torch.constant.int 0 | |
%1110 = torch.aten.select.int %1109, %int0_334, %int0_335 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1111 = torch.aten.item %1110 : !torch.vtensor<[1],si64> -> !torch.int | |
%1112 = torch.aten.eq.int %1111, %int0_334 : !torch.int, !torch.int -> !torch.bool | |
%1113 = torch.aten.Int.bool %1112 : !torch.bool -> !torch.int | |
%int0_336 = torch.constant.int 0 | |
%1114 = torch.aten.size.int %1084, %int0_336 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%1115 = torch.prim.NumToTensor.Scalar %1113 : !torch.int -> !torch.vtensor<[],i1> | |
%1116 = torch.prim.NumToTensor.Scalar %1114 : !torch.int -> !torch.vtensor<[],si64> | |
%1117 = torch.prim.NumToTensor.Scalar %1111 : !torch.int -> !torch.vtensor<[],si64> | |
%1118 = torch.aten.where.self %1115, %1116, %1117 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1119 = torch.aten.item %1118 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_337 = torch.constant.int 1 | |
%1120 = torch.aten.select.int %1109, %int0_334, %int1_337 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1121 = torch.aten.item %1120 : !torch.vtensor<[1],si64> -> !torch.int | |
%1122 = torch.aten.eq.int %1121, %int0_334 : !torch.int, !torch.int -> !torch.bool | |
%1123 = torch.aten.Int.bool %1122 : !torch.bool -> !torch.int | |
%int1_338 = torch.constant.int 1 | |
%1124 = torch.aten.size.int %1084, %int1_338 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%1125 = torch.prim.NumToTensor.Scalar %1123 : !torch.int -> !torch.vtensor<[],i1> | |
%1126 = torch.prim.NumToTensor.Scalar %1124 : !torch.int -> !torch.vtensor<[],si64> | |
%1127 = torch.prim.NumToTensor.Scalar %1121 : !torch.int -> !torch.vtensor<[],si64> | |
%1128 = torch.aten.where.self %1125, %1126, %1127 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1129 = torch.aten.item %1128 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_339 = torch.constant.int 2 | |
%1130 = torch.aten.select.int %1109, %int0_334, %int2_339 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1131 = torch.aten.item %1130 : !torch.vtensor<[1],si64> -> !torch.int | |
%1132 = torch.aten.eq.int %1131, %int0_334 : !torch.int, !torch.int -> !torch.bool | |
%1133 = torch.aten.Int.bool %1132 : !torch.bool -> !torch.int | |
%int2_340 = torch.constant.int 2 | |
%1134 = torch.aten.size.int %1084, %int2_340 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%1135 = torch.prim.NumToTensor.Scalar %1133 : !torch.int -> !torch.vtensor<[],i1> | |
%1136 = torch.prim.NumToTensor.Scalar %1134 : !torch.int -> !torch.vtensor<[],si64> | |
%1137 = torch.prim.NumToTensor.Scalar %1131 : !torch.int -> !torch.vtensor<[],si64> | |
%1138 = torch.aten.where.self %1135, %1136, %1137 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1139 = torch.aten.item %1138 : !torch.vtensor<[],si64> -> !torch.int | |
%1140 = torch.prim.ListConstruct %1119, %1129, %1139 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1141 = torch.aten.reshape %1084, %1140 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%1142 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%1143 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_341 = torch.constant.none | |
%true_342 = torch.constant.bool true | |
%false_343 = torch.constant.bool false | |
%float9.999990e-06_344 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_345 = torch.constant.float 0.000000e+00 | |
%1144 = torch.aten.instance_norm %1141, %1142, %1143, %none_341, %none_341, %true_342, %float0.000000e00_345, %float9.999990e-06_344, %false_343 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%1145 = torch.aten._shape_as_tensor %1084 : !torch.vtensor<[?,64,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_346 = torch.constant.int 0 | |
%int0_347 = torch.constant.int 0 | |
%1146 = torch.aten.select.int %1145, %int0_346, %int0_347 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1147 = torch.aten.item %1146 : !torch.vtensor<[1],si64> -> !torch.int | |
%1148 = torch.aten.eq.int %1147, %int0_346 : !torch.int, !torch.int -> !torch.bool | |
%1149 = torch.aten.Int.bool %1148 : !torch.bool -> !torch.int | |
%int0_348 = torch.constant.int 0 | |
%1150 = torch.aten.size.int %1144, %int0_348 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1151 = torch.prim.NumToTensor.Scalar %1149 : !torch.int -> !torch.vtensor<[],i1> | |
%1152 = torch.prim.NumToTensor.Scalar %1150 : !torch.int -> !torch.vtensor<[],si64> | |
%1153 = torch.prim.NumToTensor.Scalar %1147 : !torch.int -> !torch.vtensor<[],si64> | |
%1154 = torch.aten.where.self %1151, %1152, %1153 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1155 = torch.aten.item %1154 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_349 = torch.constant.int 1 | |
%1156 = torch.aten.select.int %1145, %int0_346, %int1_349 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1157 = torch.aten.item %1156 : !torch.vtensor<[1],si64> -> !torch.int | |
%1158 = torch.aten.eq.int %1157, %int0_346 : !torch.int, !torch.int -> !torch.bool | |
%1159 = torch.aten.Int.bool %1158 : !torch.bool -> !torch.int | |
%int1_350 = torch.constant.int 1 | |
%1160 = torch.aten.size.int %1144, %int1_350 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1161 = torch.prim.NumToTensor.Scalar %1159 : !torch.int -> !torch.vtensor<[],i1> | |
%1162 = torch.prim.NumToTensor.Scalar %1160 : !torch.int -> !torch.vtensor<[],si64> | |
%1163 = torch.prim.NumToTensor.Scalar %1157 : !torch.int -> !torch.vtensor<[],si64> | |
%1164 = torch.aten.where.self %1161, %1162, %1163 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1165 = torch.aten.item %1164 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_351 = torch.constant.int 2 | |
%1166 = torch.aten.select.int %1145, %int0_346, %int2_351 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1167 = torch.aten.item %1166 : !torch.vtensor<[1],si64> -> !torch.int | |
%1168 = torch.aten.eq.int %1167, %int0_346 : !torch.int, !torch.int -> !torch.bool | |
%1169 = torch.aten.Int.bool %1168 : !torch.bool -> !torch.int | |
%int2_352 = torch.constant.int 2 | |
%1170 = torch.aten.size.int %1144, %int2_352 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1171 = torch.prim.NumToTensor.Scalar %1169 : !torch.int -> !torch.vtensor<[],i1> | |
%1172 = torch.prim.NumToTensor.Scalar %1170 : !torch.int -> !torch.vtensor<[],si64> | |
%1173 = torch.prim.NumToTensor.Scalar %1167 : !torch.int -> !torch.vtensor<[],si64> | |
%1174 = torch.aten.where.self %1171, %1172, %1173 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1175 = torch.aten.item %1174 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_353 = torch.constant.int 3 | |
%1176 = torch.aten.select.int %1145, %int0_346, %int3_353 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1177 = torch.aten.item %1176 : !torch.vtensor<[1],si64> -> !torch.int | |
%1178 = torch.aten.eq.int %1177, %int0_346 : !torch.int, !torch.int -> !torch.bool | |
%1179 = torch.aten.Int.bool %1178 : !torch.bool -> !torch.int | |
%int4_354 = torch.constant.int 4 | |
%1180 = torch.aten.select.int %1145, %int0_346, %int4_354 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1181 = torch.aten.item %1180 : !torch.vtensor<[1],si64> -> !torch.int | |
%1182 = torch.aten.eq.int %1181, %int0_346 : !torch.int, !torch.int -> !torch.bool | |
%1183 = torch.aten.Int.bool %1182 : !torch.bool -> !torch.int | |
%1184 = torch.prim.ListConstruct %1155, %1165, %1175, %1177, %1181 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1185 = torch.aten.reshape %1144, %1184 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%1186 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_355 = torch.constant.int 0 | |
%int1_356 = torch.constant.int 1 | |
%int0_357 = torch.constant.int 0 | |
%1187 = torch.aten.select.int %1186, %int0_355, %int0_357 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1188 = torch.aten.item %1187 : !torch.vtensor<[1],si64> -> !torch.int | |
%1189 = torch.aten.lt.int %1188, %int0_355 : !torch.int, !torch.int -> !torch.bool | |
%1190 = torch.aten.Int.bool %1189 : !torch.bool -> !torch.int | |
%1191 = torch.aten.mul.int %1190, %int1_356 : !torch.int, !torch.int -> !torch.int | |
%1192 = torch.aten.add.int %1188, %1191 : !torch.int, !torch.int -> !torch.int | |
%int1_358 = torch.constant.int 1 | |
%1193 = torch.aten.select.int %1186, %int0_355, %int1_358 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1194 = torch.aten.item %1193 : !torch.vtensor<[1],si64> -> !torch.int | |
%1195 = torch.aten.lt.int %1194, %int0_355 : !torch.int, !torch.int -> !torch.bool | |
%1196 = torch.aten.Int.bool %1195 : !torch.bool -> !torch.int | |
%1197 = torch.aten.mul.int %1196, %int1_356 : !torch.int, !torch.int -> !torch.int | |
%1198 = torch.aten.add.int %1194, %1197 : !torch.int, !torch.int -> !torch.int | |
%int2_359 = torch.constant.int 2 | |
%1199 = torch.aten.select.int %1186, %int0_355, %int2_359 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1200 = torch.aten.item %1199 : !torch.vtensor<[1],si64> -> !torch.int | |
%1201 = torch.aten.lt.int %1200, %int0_355 : !torch.int, !torch.int -> !torch.bool | |
%1202 = torch.aten.Int.bool %1201 : !torch.bool -> !torch.int | |
%1203 = torch.aten.mul.int %1202, %int1_356 : !torch.int, !torch.int -> !torch.int | |
%1204 = torch.aten.add.int %1200, %1203 : !torch.int, !torch.int -> !torch.int | |
%1205 = torch.prim.ListConstruct %1192, %1198, %1204 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_360 = torch.constant.bool false | |
%none_361 = torch.constant.none | |
%1206 = torch.aten.tensor %1205, %none_361, %none_361, %false_360 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_362, %indices_363 = torch.aten.sort %1206, %int0_355, %false_360 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_364 = torch.constant.int 0 | |
%1207 = torch.aten.select.int %values_362, %int0_355, %int0_364 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1208 = torch.aten.item %1207 : !torch.vtensor<[1],si64> -> !torch.int | |
%1209 = torch.aten.unsqueeze %1096, %1208 : !torch.vtensor<[64],f32>, !torch.int -> !torch.vtensor | |
%int1_365 = torch.constant.int 1 | |
%1210 = torch.aten.select.int %values_362, %int0_355, %int1_365 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1211 = torch.aten.item %1210 : !torch.vtensor<[1],si64> -> !torch.int | |
%1212 = torch.aten.unsqueeze %1209, %1211 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_366 = torch.constant.int 2 | |
%1213 = torch.aten.select.int %values_362, %int0_355, %int2_366 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1214 = torch.aten.item %1213 : !torch.vtensor<[1],si64> -> !torch.int | |
%1215 = torch.aten.unsqueeze %1212, %1214 : !torch.vtensor, !torch.int -> !torch.vtensor<[64,1,1,1],f32> | |
%1216 = torch.aten.mul.Tensor %1185, %1215 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[64,1,1,1],f32> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1217 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_367 = torch.constant.int 0 | |
%int1_368 = torch.constant.int 1 | |
%int0_369 = torch.constant.int 0 | |
%1218 = torch.aten.select.int %1217, %int0_367, %int0_369 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1219 = torch.aten.item %1218 : !torch.vtensor<[1],si64> -> !torch.int | |
%1220 = torch.aten.lt.int %1219, %int0_367 : !torch.int, !torch.int -> !torch.bool | |
%1221 = torch.aten.Int.bool %1220 : !torch.bool -> !torch.int | |
%1222 = torch.aten.mul.int %1221, %int1_368 : !torch.int, !torch.int -> !torch.int | |
%1223 = torch.aten.add.int %1219, %1222 : !torch.int, !torch.int -> !torch.int | |
%int1_370 = torch.constant.int 1 | |
%1224 = torch.aten.select.int %1217, %int0_367, %int1_370 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1225 = torch.aten.item %1224 : !torch.vtensor<[1],si64> -> !torch.int | |
%1226 = torch.aten.lt.int %1225, %int0_367 : !torch.int, !torch.int -> !torch.bool | |
%1227 = torch.aten.Int.bool %1226 : !torch.bool -> !torch.int | |
%1228 = torch.aten.mul.int %1227, %int1_368 : !torch.int, !torch.int -> !torch.int | |
%1229 = torch.aten.add.int %1225, %1228 : !torch.int, !torch.int -> !torch.int | |
%int2_371 = torch.constant.int 2 | |
%1230 = torch.aten.select.int %1217, %int0_367, %int2_371 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1231 = torch.aten.item %1230 : !torch.vtensor<[1],si64> -> !torch.int | |
%1232 = torch.aten.lt.int %1231, %int0_367 : !torch.int, !torch.int -> !torch.bool | |
%1233 = torch.aten.Int.bool %1232 : !torch.bool -> !torch.int | |
%1234 = torch.aten.mul.int %1233, %int1_368 : !torch.int, !torch.int -> !torch.int | |
%1235 = torch.aten.add.int %1231, %1234 : !torch.int, !torch.int -> !torch.int | |
%1236 = torch.prim.ListConstruct %1223, %1229, %1235 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_372 = torch.constant.bool false | |
%none_373 = torch.constant.none | |
%1237 = torch.aten.tensor %1236, %none_373, %none_373, %false_372 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_374, %indices_375 = torch.aten.sort %1237, %int0_367, %false_372 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_376 = torch.constant.int 0 | |
%1238 = torch.aten.select.int %values_374, %int0_367, %int0_376 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1239 = torch.aten.item %1238 : !torch.vtensor<[1],si64> -> !torch.int | |
%1240 = torch.aten.unsqueeze %1108, %1239 : !torch.vtensor<[64],f32>, !torch.int -> !torch.vtensor | |
%int1_377 = torch.constant.int 1 | |
%1241 = torch.aten.select.int %values_374, %int0_367, %int1_377 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1242 = torch.aten.item %1241 : !torch.vtensor<[1],si64> -> !torch.int | |
%1243 = torch.aten.unsqueeze %1240, %1242 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_378 = torch.constant.int 2 | |
%1244 = torch.aten.select.int %values_374, %int0_367, %int2_378 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1245 = torch.aten.item %1244 : !torch.vtensor<[1],si64> -> !torch.int | |
%1246 = torch.aten.unsqueeze %1243, %1245 : !torch.vtensor, !torch.int -> !torch.vtensor<[64,1,1,1],f32> | |
%int1_379 = torch.constant.int 1 | |
%1247 = torch.aten.add.Tensor %1216, %1246, %int1_379 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.vtensor<[64,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1248 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1249 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_380 = torch.constant.int 12 | |
%1250 = torch.aten.item %1248 : !torch.vtensor<[],f32> -> !torch.float | |
%1251 = torch.aten.item %1249 : !torch.vtensor<[],si8> -> !torch.int | |
%1252 = torch.aten.quantize_per_tensor %1247, %1250, %1251, %int12_380 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%1253 = torch.aten.int_repr %1252 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],si8> | |
%1254 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1255 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1256 = torch.aten.item %1254 : !torch.vtensor<[],f32> -> !torch.float | |
%1257 = torch.aten.item %1255 : !torch.vtensor<[],si8> -> !torch.int | |
%1258 = torch.aten._make_per_tensor_quantized_tensor %1253, %1256, %1257 : !torch.vtensor<[?,64,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%1259 = torch.aten.dequantize.self %1258 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%1260 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1261 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_381 = torch.constant.int 12 | |
%1262 = torch.aten.item %1260 : !torch.vtensor<[],f32> -> !torch.float | |
%1263 = torch.aten.item %1261 : !torch.vtensor<[],si8> -> !torch.int | |
%1264 = torch.aten.quantize_per_tensor %17, %1262, %1263, %int12_381 : !torch.vtensor<[128,64,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128,64,3,3,3],!torch.qint8> | |
%1265 = torch.aten.int_repr %1264 : !torch.vtensor<[128,64,3,3,3],!torch.qint8> -> !torch.vtensor<[128,64,3,3,3],si8> | |
%1266 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1267 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1268 = torch.aten.item %1266 : !torch.vtensor<[],f32> -> !torch.float | |
%1269 = torch.aten.item %1267 : !torch.vtensor<[],si8> -> !torch.int | |
%1270 = torch.aten._make_per_tensor_quantized_tensor %1265, %1268, %1269 : !torch.vtensor<[128,64,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[128,64,3,3,3],!torch.qint8> | |
%1271 = torch.aten.dequantize.self %1270 : !torch.vtensor<[128,64,3,3,3],!torch.qint8> -> !torch.vtensor<[128,64,3,3,3],f32> | |
%int1_382 = torch.constant.int 1 | |
%int1_383 = torch.constant.int 1 | |
%int1_384 = torch.constant.int 1 | |
%int1_385 = torch.constant.int 1 | |
%int1_386 = torch.constant.int 1 | |
%int1_387 = torch.constant.int 1 | |
%int1_388 = torch.constant.int 1 | |
%int1_389 = torch.constant.int 1 | |
%int1_390 = torch.constant.int 1 | |
%int0_391 = torch.constant.int 0 | |
%1272 = torch.prim.ListConstruct %int1_382, %int1_383, %int1_384 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1273 = torch.prim.ListConstruct %int1_385, %int1_386, %int1_387 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1274 = torch.prim.ListConstruct %int1_388, %int1_389, %int1_390 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1275 = torch.prim.ListConstruct %int0_391, %int0_391 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_392 = torch.constant.bool false | |
%none_393 = torch.constant.none | |
%int1_394 = torch.constant.int 1 | |
%1276 = torch.aten.convolution %1259, %1271, %none_393, %1274, %1272, %1273, %false_392, %1275, %int1_394 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.vtensor<[128,64,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1277 = torch.aten.relu %1276 : !torch.vtensor<[?,128,?,?,?],f32> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1278 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1279 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_395 = torch.constant.int 12 | |
%1280 = torch.aten.item %1278 : !torch.vtensor<[],f32> -> !torch.float | |
%1281 = torch.aten.item %1279 : !torch.vtensor<[],si8> -> !torch.int | |
%1282 = torch.aten.quantize_per_tensor %1277, %1280, %1281, %int12_395 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1283 = torch.aten.int_repr %1282 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],si8> | |
%1284 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1285 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1286 = torch.aten.item %1284 : !torch.vtensor<[],f32> -> !torch.float | |
%1287 = torch.aten.item %1285 : !torch.vtensor<[],si8> -> !torch.int | |
%1288 = torch.aten._make_per_tensor_quantized_tensor %1283, %1286, %1287 : !torch.vtensor<[?,128,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1289 = torch.aten.dequantize.self %1288 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%int2_396 = torch.constant.int 2 | |
%int2_397 = torch.constant.int 2 | |
%int2_398 = torch.constant.int 2 | |
%1290 = torch.prim.ListConstruct %int2_396, %int2_397, %int2_398 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int0_399 = torch.constant.int 0 | |
%int0_400 = torch.constant.int 0 | |
%int0_401 = torch.constant.int 0 | |
%1291 = torch.prim.ListConstruct %int0_399, %int0_400, %int0_401 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int2_402 = torch.constant.int 2 | |
%int2_403 = torch.constant.int 2 | |
%int2_404 = torch.constant.int 2 | |
%1292 = torch.prim.ListConstruct %int2_402, %int2_403, %int2_404 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%int1_405 = torch.constant.int 1 | |
%int1_406 = torch.constant.int 1 | |
%int1_407 = torch.constant.int 1 | |
%1293 = torch.prim.ListConstruct %int1_405, %int1_406, %int1_407 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_408 = torch.constant.bool false | |
%1294 = torch.aten.max_pool3d %1289, %1290, %1292, %1291, %1293, %false_408 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1295 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1296 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_409 = torch.constant.int 12 | |
%1297 = torch.aten.item %1295 : !torch.vtensor<[],f32> -> !torch.float | |
%1298 = torch.aten.item %1296 : !torch.vtensor<[],si8> -> !torch.int | |
%1299 = torch.aten.quantize_per_tensor %1294, %1297, %1298, %int12_409 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1300 = torch.aten.int_repr %1299 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],si8> | |
%1301 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1302 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1303 = torch.aten.item %1301 : !torch.vtensor<[],f32> -> !torch.float | |
%1304 = torch.aten.item %1302 : !torch.vtensor<[],si8> -> !torch.int | |
%1305 = torch.aten._make_per_tensor_quantized_tensor %1300, %1303, %1304 : !torch.vtensor<[?,128,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1306 = torch.aten.dequantize.self %1305 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1307 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1308 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_410 = torch.constant.int 12 | |
%1309 = torch.aten.item %1307 : !torch.vtensor<[],f32> -> !torch.float | |
%1310 = torch.aten.item %1308 : !torch.vtensor<[],si8> -> !torch.int | |
%1311 = torch.aten.quantize_per_tensor %18, %1309, %1310, %int12_410 : !torch.vtensor<[128],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%1312 = torch.aten.int_repr %1311 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],si8> | |
%1313 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1314 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1315 = torch.aten.item %1313 : !torch.vtensor<[],f32> -> !torch.float | |
%1316 = torch.aten.item %1314 : !torch.vtensor<[],si8> -> !torch.int | |
%1317 = torch.aten._make_per_tensor_quantized_tensor %1312, %1315, %1316 : !torch.vtensor<[128],si8>, !torch.float, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%1318 = torch.aten.dequantize.self %1317 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],f32> | |
%1319 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1320 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_411 = torch.constant.int 12 | |
%1321 = torch.aten.item %1319 : !torch.vtensor<[],f32> -> !torch.float | |
%1322 = torch.aten.item %1320 : !torch.vtensor<[],si8> -> !torch.int | |
%1323 = torch.aten.quantize_per_tensor %19, %1321, %1322, %int12_411 : !torch.vtensor<[128],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%1324 = torch.aten.int_repr %1323 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],si8> | |
%1325 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1326 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1327 = torch.aten.item %1325 : !torch.vtensor<[],f32> -> !torch.float | |
%1328 = torch.aten.item %1326 : !torch.vtensor<[],si8> -> !torch.int | |
%1329 = torch.aten._make_per_tensor_quantized_tensor %1324, %1327, %1328 : !torch.vtensor<[128],si8>, !torch.float, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%1330 = torch.aten.dequantize.self %1329 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],f32> | |
%1331 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_412 = torch.constant.int 0 | |
%int0_413 = torch.constant.int 0 | |
%1332 = torch.aten.select.int %1331, %int0_412, %int0_413 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1333 = torch.aten.item %1332 : !torch.vtensor<[1],si64> -> !torch.int | |
%1334 = torch.aten.eq.int %1333, %int0_412 : !torch.int, !torch.int -> !torch.bool | |
%1335 = torch.aten.Int.bool %1334 : !torch.bool -> !torch.int | |
%int0_414 = torch.constant.int 0 | |
%1336 = torch.aten.size.int %1306, %int0_414 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%1337 = torch.prim.NumToTensor.Scalar %1335 : !torch.int -> !torch.vtensor<[],i1> | |
%1338 = torch.prim.NumToTensor.Scalar %1336 : !torch.int -> !torch.vtensor<[],si64> | |
%1339 = torch.prim.NumToTensor.Scalar %1333 : !torch.int -> !torch.vtensor<[],si64> | |
%1340 = torch.aten.where.self %1337, %1338, %1339 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1341 = torch.aten.item %1340 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_415 = torch.constant.int 1 | |
%1342 = torch.aten.select.int %1331, %int0_412, %int1_415 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1343 = torch.aten.item %1342 : !torch.vtensor<[1],si64> -> !torch.int | |
%1344 = torch.aten.eq.int %1343, %int0_412 : !torch.int, !torch.int -> !torch.bool | |
%1345 = torch.aten.Int.bool %1344 : !torch.bool -> !torch.int | |
%int1_416 = torch.constant.int 1 | |
%1346 = torch.aten.size.int %1306, %int1_416 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%1347 = torch.prim.NumToTensor.Scalar %1345 : !torch.int -> !torch.vtensor<[],i1> | |
%1348 = torch.prim.NumToTensor.Scalar %1346 : !torch.int -> !torch.vtensor<[],si64> | |
%1349 = torch.prim.NumToTensor.Scalar %1343 : !torch.int -> !torch.vtensor<[],si64> | |
%1350 = torch.aten.where.self %1347, %1348, %1349 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1351 = torch.aten.item %1350 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_417 = torch.constant.int 2 | |
%1352 = torch.aten.select.int %1331, %int0_412, %int2_417 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1353 = torch.aten.item %1352 : !torch.vtensor<[1],si64> -> !torch.int | |
%1354 = torch.aten.eq.int %1353, %int0_412 : !torch.int, !torch.int -> !torch.bool | |
%1355 = torch.aten.Int.bool %1354 : !torch.bool -> !torch.int | |
%int2_418 = torch.constant.int 2 | |
%1356 = torch.aten.size.int %1306, %int2_418 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%1357 = torch.prim.NumToTensor.Scalar %1355 : !torch.int -> !torch.vtensor<[],i1> | |
%1358 = torch.prim.NumToTensor.Scalar %1356 : !torch.int -> !torch.vtensor<[],si64> | |
%1359 = torch.prim.NumToTensor.Scalar %1353 : !torch.int -> !torch.vtensor<[],si64> | |
%1360 = torch.aten.where.self %1357, %1358, %1359 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1361 = torch.aten.item %1360 : !torch.vtensor<[],si64> -> !torch.int | |
%1362 = torch.prim.ListConstruct %1341, %1351, %1361 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1363 = torch.aten.reshape %1306, %1362 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%1364 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%1365 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_419 = torch.constant.none | |
%true_420 = torch.constant.bool true | |
%false_421 = torch.constant.bool false | |
%float9.999990e-06_422 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_423 = torch.constant.float 0.000000e+00 | |
%1366 = torch.aten.instance_norm %1363, %1364, %1365, %none_419, %none_419, %true_420, %float0.000000e00_423, %float9.999990e-06_422, %false_421 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%1367 = torch.aten._shape_as_tensor %1306 : !torch.vtensor<[?,128,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_424 = torch.constant.int 0 | |
%int0_425 = torch.constant.int 0 | |
%1368 = torch.aten.select.int %1367, %int0_424, %int0_425 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1369 = torch.aten.item %1368 : !torch.vtensor<[1],si64> -> !torch.int | |
%1370 = torch.aten.eq.int %1369, %int0_424 : !torch.int, !torch.int -> !torch.bool | |
%1371 = torch.aten.Int.bool %1370 : !torch.bool -> !torch.int | |
%int0_426 = torch.constant.int 0 | |
%1372 = torch.aten.size.int %1366, %int0_426 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1373 = torch.prim.NumToTensor.Scalar %1371 : !torch.int -> !torch.vtensor<[],i1> | |
%1374 = torch.prim.NumToTensor.Scalar %1372 : !torch.int -> !torch.vtensor<[],si64> | |
%1375 = torch.prim.NumToTensor.Scalar %1369 : !torch.int -> !torch.vtensor<[],si64> | |
%1376 = torch.aten.where.self %1373, %1374, %1375 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1377 = torch.aten.item %1376 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_427 = torch.constant.int 1 | |
%1378 = torch.aten.select.int %1367, %int0_424, %int1_427 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1379 = torch.aten.item %1378 : !torch.vtensor<[1],si64> -> !torch.int | |
%1380 = torch.aten.eq.int %1379, %int0_424 : !torch.int, !torch.int -> !torch.bool | |
%1381 = torch.aten.Int.bool %1380 : !torch.bool -> !torch.int | |
%int1_428 = torch.constant.int 1 | |
%1382 = torch.aten.size.int %1366, %int1_428 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1383 = torch.prim.NumToTensor.Scalar %1381 : !torch.int -> !torch.vtensor<[],i1> | |
%1384 = torch.prim.NumToTensor.Scalar %1382 : !torch.int -> !torch.vtensor<[],si64> | |
%1385 = torch.prim.NumToTensor.Scalar %1379 : !torch.int -> !torch.vtensor<[],si64> | |
%1386 = torch.aten.where.self %1383, %1384, %1385 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1387 = torch.aten.item %1386 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_429 = torch.constant.int 2 | |
%1388 = torch.aten.select.int %1367, %int0_424, %int2_429 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1389 = torch.aten.item %1388 : !torch.vtensor<[1],si64> -> !torch.int | |
%1390 = torch.aten.eq.int %1389, %int0_424 : !torch.int, !torch.int -> !torch.bool | |
%1391 = torch.aten.Int.bool %1390 : !torch.bool -> !torch.int | |
%int2_430 = torch.constant.int 2 | |
%1392 = torch.aten.size.int %1366, %int2_430 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1393 = torch.prim.NumToTensor.Scalar %1391 : !torch.int -> !torch.vtensor<[],i1> | |
%1394 = torch.prim.NumToTensor.Scalar %1392 : !torch.int -> !torch.vtensor<[],si64> | |
%1395 = torch.prim.NumToTensor.Scalar %1389 : !torch.int -> !torch.vtensor<[],si64> | |
%1396 = torch.aten.where.self %1393, %1394, %1395 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1397 = torch.aten.item %1396 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_431 = torch.constant.int 3 | |
%1398 = torch.aten.select.int %1367, %int0_424, %int3_431 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1399 = torch.aten.item %1398 : !torch.vtensor<[1],si64> -> !torch.int | |
%1400 = torch.aten.eq.int %1399, %int0_424 : !torch.int, !torch.int -> !torch.bool | |
%1401 = torch.aten.Int.bool %1400 : !torch.bool -> !torch.int | |
%int4_432 = torch.constant.int 4 | |
%1402 = torch.aten.select.int %1367, %int0_424, %int4_432 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1403 = torch.aten.item %1402 : !torch.vtensor<[1],si64> -> !torch.int | |
%1404 = torch.aten.eq.int %1403, %int0_424 : !torch.int, !torch.int -> !torch.bool | |
%1405 = torch.aten.Int.bool %1404 : !torch.bool -> !torch.int | |
%1406 = torch.prim.ListConstruct %1377, %1387, %1397, %1399, %1403 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1407 = torch.aten.reshape %1366, %1406 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%1408 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_433 = torch.constant.int 0 | |
%int1_434 = torch.constant.int 1 | |
%int0_435 = torch.constant.int 0 | |
%1409 = torch.aten.select.int %1408, %int0_433, %int0_435 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1410 = torch.aten.item %1409 : !torch.vtensor<[1],si64> -> !torch.int | |
%1411 = torch.aten.lt.int %1410, %int0_433 : !torch.int, !torch.int -> !torch.bool | |
%1412 = torch.aten.Int.bool %1411 : !torch.bool -> !torch.int | |
%1413 = torch.aten.mul.int %1412, %int1_434 : !torch.int, !torch.int -> !torch.int | |
%1414 = torch.aten.add.int %1410, %1413 : !torch.int, !torch.int -> !torch.int | |
%int1_436 = torch.constant.int 1 | |
%1415 = torch.aten.select.int %1408, %int0_433, %int1_436 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1416 = torch.aten.item %1415 : !torch.vtensor<[1],si64> -> !torch.int | |
%1417 = torch.aten.lt.int %1416, %int0_433 : !torch.int, !torch.int -> !torch.bool | |
%1418 = torch.aten.Int.bool %1417 : !torch.bool -> !torch.int | |
%1419 = torch.aten.mul.int %1418, %int1_434 : !torch.int, !torch.int -> !torch.int | |
%1420 = torch.aten.add.int %1416, %1419 : !torch.int, !torch.int -> !torch.int | |
%int2_437 = torch.constant.int 2 | |
%1421 = torch.aten.select.int %1408, %int0_433, %int2_437 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1422 = torch.aten.item %1421 : !torch.vtensor<[1],si64> -> !torch.int | |
%1423 = torch.aten.lt.int %1422, %int0_433 : !torch.int, !torch.int -> !torch.bool | |
%1424 = torch.aten.Int.bool %1423 : !torch.bool -> !torch.int | |
%1425 = torch.aten.mul.int %1424, %int1_434 : !torch.int, !torch.int -> !torch.int | |
%1426 = torch.aten.add.int %1422, %1425 : !torch.int, !torch.int -> !torch.int | |
%1427 = torch.prim.ListConstruct %1414, %1420, %1426 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_438 = torch.constant.bool false | |
%none_439 = torch.constant.none | |
%1428 = torch.aten.tensor %1427, %none_439, %none_439, %false_438 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_440, %indices_441 = torch.aten.sort %1428, %int0_433, %false_438 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_442 = torch.constant.int 0 | |
%1429 = torch.aten.select.int %values_440, %int0_433, %int0_442 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1430 = torch.aten.item %1429 : !torch.vtensor<[1],si64> -> !torch.int | |
%1431 = torch.aten.unsqueeze %1318, %1430 : !torch.vtensor<[128],f32>, !torch.int -> !torch.vtensor | |
%int1_443 = torch.constant.int 1 | |
%1432 = torch.aten.select.int %values_440, %int0_433, %int1_443 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1433 = torch.aten.item %1432 : !torch.vtensor<[1],si64> -> !torch.int | |
%1434 = torch.aten.unsqueeze %1431, %1433 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_444 = torch.constant.int 2 | |
%1435 = torch.aten.select.int %values_440, %int0_433, %int2_444 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1436 = torch.aten.item %1435 : !torch.vtensor<[1],si64> -> !torch.int | |
%1437 = torch.aten.unsqueeze %1434, %1436 : !torch.vtensor, !torch.int -> !torch.vtensor<[128,1,1,1],f32> | |
%1438 = torch.aten.mul.Tensor %1407, %1437 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[128,1,1,1],f32> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1439 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_445 = torch.constant.int 0 | |
%int1_446 = torch.constant.int 1 | |
%int0_447 = torch.constant.int 0 | |
%1440 = torch.aten.select.int %1439, %int0_445, %int0_447 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1441 = torch.aten.item %1440 : !torch.vtensor<[1],si64> -> !torch.int | |
%1442 = torch.aten.lt.int %1441, %int0_445 : !torch.int, !torch.int -> !torch.bool | |
%1443 = torch.aten.Int.bool %1442 : !torch.bool -> !torch.int | |
%1444 = torch.aten.mul.int %1443, %int1_446 : !torch.int, !torch.int -> !torch.int | |
%1445 = torch.aten.add.int %1441, %1444 : !torch.int, !torch.int -> !torch.int | |
%int1_448 = torch.constant.int 1 | |
%1446 = torch.aten.select.int %1439, %int0_445, %int1_448 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1447 = torch.aten.item %1446 : !torch.vtensor<[1],si64> -> !torch.int | |
%1448 = torch.aten.lt.int %1447, %int0_445 : !torch.int, !torch.int -> !torch.bool | |
%1449 = torch.aten.Int.bool %1448 : !torch.bool -> !torch.int | |
%1450 = torch.aten.mul.int %1449, %int1_446 : !torch.int, !torch.int -> !torch.int | |
%1451 = torch.aten.add.int %1447, %1450 : !torch.int, !torch.int -> !torch.int | |
%int2_449 = torch.constant.int 2 | |
%1452 = torch.aten.select.int %1439, %int0_445, %int2_449 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1453 = torch.aten.item %1452 : !torch.vtensor<[1],si64> -> !torch.int | |
%1454 = torch.aten.lt.int %1453, %int0_445 : !torch.int, !torch.int -> !torch.bool | |
%1455 = torch.aten.Int.bool %1454 : !torch.bool -> !torch.int | |
%1456 = torch.aten.mul.int %1455, %int1_446 : !torch.int, !torch.int -> !torch.int | |
%1457 = torch.aten.add.int %1453, %1456 : !torch.int, !torch.int -> !torch.int | |
%1458 = torch.prim.ListConstruct %1445, %1451, %1457 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_450 = torch.constant.bool false | |
%none_451 = torch.constant.none | |
%1459 = torch.aten.tensor %1458, %none_451, %none_451, %false_450 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_452, %indices_453 = torch.aten.sort %1459, %int0_445, %false_450 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_454 = torch.constant.int 0 | |
%1460 = torch.aten.select.int %values_452, %int0_445, %int0_454 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1461 = torch.aten.item %1460 : !torch.vtensor<[1],si64> -> !torch.int | |
%1462 = torch.aten.unsqueeze %1330, %1461 : !torch.vtensor<[128],f32>, !torch.int -> !torch.vtensor | |
%int1_455 = torch.constant.int 1 | |
%1463 = torch.aten.select.int %values_452, %int0_445, %int1_455 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1464 = torch.aten.item %1463 : !torch.vtensor<[1],si64> -> !torch.int | |
%1465 = torch.aten.unsqueeze %1462, %1464 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_456 = torch.constant.int 2 | |
%1466 = torch.aten.select.int %values_452, %int0_445, %int2_456 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1467 = torch.aten.item %1466 : !torch.vtensor<[1],si64> -> !torch.int | |
%1468 = torch.aten.unsqueeze %1465, %1467 : !torch.vtensor, !torch.int -> !torch.vtensor<[128,1,1,1],f32> | |
%int1_457 = torch.constant.int 1 | |
%1469 = torch.aten.add.Tensor %1438, %1468, %int1_457 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.vtensor<[128,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1470 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1471 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_458 = torch.constant.int 12 | |
%1472 = torch.aten.item %1470 : !torch.vtensor<[],f32> -> !torch.float | |
%1473 = torch.aten.item %1471 : !torch.vtensor<[],si8> -> !torch.int | |
%1474 = torch.aten.quantize_per_tensor %1469, %1472, %1473, %int12_458 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1475 = torch.aten.int_repr %1474 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],si8> | |
%1476 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1477 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1478 = torch.aten.item %1476 : !torch.vtensor<[],f32> -> !torch.float | |
%1479 = torch.aten.item %1477 : !torch.vtensor<[],si8> -> !torch.int | |
%1480 = torch.aten._make_per_tensor_quantized_tensor %1475, %1478, %1479 : !torch.vtensor<[?,128,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1481 = torch.aten.dequantize.self %1480 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1482 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1483 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_459 = torch.constant.int 12 | |
%1484 = torch.aten.item %1482 : !torch.vtensor<[],f32> -> !torch.float | |
%1485 = torch.aten.item %1483 : !torch.vtensor<[],si8> -> !torch.int | |
%1486 = torch.aten.quantize_per_tensor %20, %1484, %1485, %int12_459 : !torch.vtensor<[128,128,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128,128,3,3,3],!torch.qint8> | |
%1487 = torch.aten.int_repr %1486 : !torch.vtensor<[128,128,3,3,3],!torch.qint8> -> !torch.vtensor<[128,128,3,3,3],si8> | |
%1488 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1489 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1490 = torch.aten.item %1488 : !torch.vtensor<[],f32> -> !torch.float | |
%1491 = torch.aten.item %1489 : !torch.vtensor<[],si8> -> !torch.int | |
%1492 = torch.aten._make_per_tensor_quantized_tensor %1487, %1490, %1491 : !torch.vtensor<[128,128,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[128,128,3,3,3],!torch.qint8> | |
%1493 = torch.aten.dequantize.self %1492 : !torch.vtensor<[128,128,3,3,3],!torch.qint8> -> !torch.vtensor<[128,128,3,3,3],f32> | |
%int1_460 = torch.constant.int 1 | |
%int1_461 = torch.constant.int 1 | |
%int1_462 = torch.constant.int 1 | |
%int1_463 = torch.constant.int 1 | |
%int1_464 = torch.constant.int 1 | |
%int1_465 = torch.constant.int 1 | |
%int1_466 = torch.constant.int 1 | |
%int1_467 = torch.constant.int 1 | |
%int1_468 = torch.constant.int 1 | |
%int0_469 = torch.constant.int 0 | |
%1494 = torch.prim.ListConstruct %int1_460, %int1_461, %int1_462 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1495 = torch.prim.ListConstruct %int1_463, %int1_464, %int1_465 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1496 = torch.prim.ListConstruct %int1_466, %int1_467, %int1_468 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1497 = torch.prim.ListConstruct %int0_469, %int0_469 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_470 = torch.constant.bool false | |
%none_471 = torch.constant.none | |
%int1_472 = torch.constant.int 1 | |
%1498 = torch.aten.convolution %1481, %1493, %none_471, %1496, %1494, %1495, %false_470, %1497, %int1_472 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.vtensor<[128,128,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1499 = torch.aten.relu %1498 : !torch.vtensor<[?,128,?,?,?],f32> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1500 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1501 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_473 = torch.constant.int 12 | |
%1502 = torch.aten.item %1500 : !torch.vtensor<[],f32> -> !torch.float | |
%1503 = torch.aten.item %1501 : !torch.vtensor<[],si8> -> !torch.int | |
%1504 = torch.aten.quantize_per_tensor %1499, %1502, %1503, %int12_473 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1505 = torch.aten.int_repr %1504 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],si8> | |
%1506 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1507 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1508 = torch.aten.item %1506 : !torch.vtensor<[],f32> -> !torch.float | |
%1509 = torch.aten.item %1507 : !torch.vtensor<[],si8> -> !torch.int | |
%1510 = torch.aten._make_per_tensor_quantized_tensor %1505, %1508, %1509 : !torch.vtensor<[?,128,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1511 = torch.aten.dequantize.self %1510 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1512 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1513 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_474 = torch.constant.int 12 | |
%1514 = torch.aten.item %1512 : !torch.vtensor<[],f32> -> !torch.float | |
%1515 = torch.aten.item %1513 : !torch.vtensor<[],si8> -> !torch.int | |
%1516 = torch.aten.quantize_per_tensor %21, %1514, %1515, %int12_474 : !torch.vtensor<[128],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%1517 = torch.aten.int_repr %1516 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],si8> | |
%1518 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1519 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1520 = torch.aten.item %1518 : !torch.vtensor<[],f32> -> !torch.float | |
%1521 = torch.aten.item %1519 : !torch.vtensor<[],si8> -> !torch.int | |
%1522 = torch.aten._make_per_tensor_quantized_tensor %1517, %1520, %1521 : !torch.vtensor<[128],si8>, !torch.float, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%1523 = torch.aten.dequantize.self %1522 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],f32> | |
%1524 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1525 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_475 = torch.constant.int 12 | |
%1526 = torch.aten.item %1524 : !torch.vtensor<[],f32> -> !torch.float | |
%1527 = torch.aten.item %1525 : !torch.vtensor<[],si8> -> !torch.int | |
%1528 = torch.aten.quantize_per_tensor %22, %1526, %1527, %int12_475 : !torch.vtensor<[128],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%1529 = torch.aten.int_repr %1528 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],si8> | |
%1530 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1531 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1532 = torch.aten.item %1530 : !torch.vtensor<[],f32> -> !torch.float | |
%1533 = torch.aten.item %1531 : !torch.vtensor<[],si8> -> !torch.int | |
%1534 = torch.aten._make_per_tensor_quantized_tensor %1529, %1532, %1533 : !torch.vtensor<[128],si8>, !torch.float, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%1535 = torch.aten.dequantize.self %1534 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],f32> | |
%1536 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_476 = torch.constant.int 0 | |
%int0_477 = torch.constant.int 0 | |
%1537 = torch.aten.select.int %1536, %int0_476, %int0_477 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1538 = torch.aten.item %1537 : !torch.vtensor<[1],si64> -> !torch.int | |
%1539 = torch.aten.eq.int %1538, %int0_476 : !torch.int, !torch.int -> !torch.bool | |
%1540 = torch.aten.Int.bool %1539 : !torch.bool -> !torch.int | |
%int0_478 = torch.constant.int 0 | |
%1541 = torch.aten.size.int %1511, %int0_478 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%1542 = torch.prim.NumToTensor.Scalar %1540 : !torch.int -> !torch.vtensor<[],i1> | |
%1543 = torch.prim.NumToTensor.Scalar %1541 : !torch.int -> !torch.vtensor<[],si64> | |
%1544 = torch.prim.NumToTensor.Scalar %1538 : !torch.int -> !torch.vtensor<[],si64> | |
%1545 = torch.aten.where.self %1542, %1543, %1544 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1546 = torch.aten.item %1545 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_479 = torch.constant.int 1 | |
%1547 = torch.aten.select.int %1536, %int0_476, %int1_479 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1548 = torch.aten.item %1547 : !torch.vtensor<[1],si64> -> !torch.int | |
%1549 = torch.aten.eq.int %1548, %int0_476 : !torch.int, !torch.int -> !torch.bool | |
%1550 = torch.aten.Int.bool %1549 : !torch.bool -> !torch.int | |
%int1_480 = torch.constant.int 1 | |
%1551 = torch.aten.size.int %1511, %int1_480 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%1552 = torch.prim.NumToTensor.Scalar %1550 : !torch.int -> !torch.vtensor<[],i1> | |
%1553 = torch.prim.NumToTensor.Scalar %1551 : !torch.int -> !torch.vtensor<[],si64> | |
%1554 = torch.prim.NumToTensor.Scalar %1548 : !torch.int -> !torch.vtensor<[],si64> | |
%1555 = torch.aten.where.self %1552, %1553, %1554 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1556 = torch.aten.item %1555 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_481 = torch.constant.int 2 | |
%1557 = torch.aten.select.int %1536, %int0_476, %int2_481 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1558 = torch.aten.item %1557 : !torch.vtensor<[1],si64> -> !torch.int | |
%1559 = torch.aten.eq.int %1558, %int0_476 : !torch.int, !torch.int -> !torch.bool | |
%1560 = torch.aten.Int.bool %1559 : !torch.bool -> !torch.int | |
%int2_482 = torch.constant.int 2 | |
%1561 = torch.aten.size.int %1511, %int2_482 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%1562 = torch.prim.NumToTensor.Scalar %1560 : !torch.int -> !torch.vtensor<[],i1> | |
%1563 = torch.prim.NumToTensor.Scalar %1561 : !torch.int -> !torch.vtensor<[],si64> | |
%1564 = torch.prim.NumToTensor.Scalar %1558 : !torch.int -> !torch.vtensor<[],si64> | |
%1565 = torch.aten.where.self %1562, %1563, %1564 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1566 = torch.aten.item %1565 : !torch.vtensor<[],si64> -> !torch.int | |
%1567 = torch.prim.ListConstruct %1546, %1556, %1566 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1568 = torch.aten.reshape %1511, %1567 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%1569 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%1570 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_483 = torch.constant.none | |
%true_484 = torch.constant.bool true | |
%false_485 = torch.constant.bool false | |
%float9.999990e-06_486 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_487 = torch.constant.float 0.000000e+00 | |
%1571 = torch.aten.instance_norm %1568, %1569, %1570, %none_483, %none_483, %true_484, %float0.000000e00_487, %float9.999990e-06_486, %false_485 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%1572 = torch.aten._shape_as_tensor %1511 : !torch.vtensor<[?,128,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_488 = torch.constant.int 0 | |
%int0_489 = torch.constant.int 0 | |
%1573 = torch.aten.select.int %1572, %int0_488, %int0_489 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1574 = torch.aten.item %1573 : !torch.vtensor<[1],si64> -> !torch.int | |
%1575 = torch.aten.eq.int %1574, %int0_488 : !torch.int, !torch.int -> !torch.bool | |
%1576 = torch.aten.Int.bool %1575 : !torch.bool -> !torch.int | |
%int0_490 = torch.constant.int 0 | |
%1577 = torch.aten.size.int %1571, %int0_490 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1578 = torch.prim.NumToTensor.Scalar %1576 : !torch.int -> !torch.vtensor<[],i1> | |
%1579 = torch.prim.NumToTensor.Scalar %1577 : !torch.int -> !torch.vtensor<[],si64> | |
%1580 = torch.prim.NumToTensor.Scalar %1574 : !torch.int -> !torch.vtensor<[],si64> | |
%1581 = torch.aten.where.self %1578, %1579, %1580 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1582 = torch.aten.item %1581 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_491 = torch.constant.int 1 | |
%1583 = torch.aten.select.int %1572, %int0_488, %int1_491 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1584 = torch.aten.item %1583 : !torch.vtensor<[1],si64> -> !torch.int | |
%1585 = torch.aten.eq.int %1584, %int0_488 : !torch.int, !torch.int -> !torch.bool | |
%1586 = torch.aten.Int.bool %1585 : !torch.bool -> !torch.int | |
%int1_492 = torch.constant.int 1 | |
%1587 = torch.aten.size.int %1571, %int1_492 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1588 = torch.prim.NumToTensor.Scalar %1586 : !torch.int -> !torch.vtensor<[],i1> | |
%1589 = torch.prim.NumToTensor.Scalar %1587 : !torch.int -> !torch.vtensor<[],si64> | |
%1590 = torch.prim.NumToTensor.Scalar %1584 : !torch.int -> !torch.vtensor<[],si64> | |
%1591 = torch.aten.where.self %1588, %1589, %1590 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1592 = torch.aten.item %1591 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_493 = torch.constant.int 2 | |
%1593 = torch.aten.select.int %1572, %int0_488, %int2_493 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1594 = torch.aten.item %1593 : !torch.vtensor<[1],si64> -> !torch.int | |
%1595 = torch.aten.eq.int %1594, %int0_488 : !torch.int, !torch.int -> !torch.bool | |
%1596 = torch.aten.Int.bool %1595 : !torch.bool -> !torch.int | |
%int2_494 = torch.constant.int 2 | |
%1597 = torch.aten.size.int %1571, %int2_494 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1598 = torch.prim.NumToTensor.Scalar %1596 : !torch.int -> !torch.vtensor<[],i1> | |
%1599 = torch.prim.NumToTensor.Scalar %1597 : !torch.int -> !torch.vtensor<[],si64> | |
%1600 = torch.prim.NumToTensor.Scalar %1594 : !torch.int -> !torch.vtensor<[],si64> | |
%1601 = torch.aten.where.self %1598, %1599, %1600 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1602 = torch.aten.item %1601 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_495 = torch.constant.int 3 | |
%1603 = torch.aten.select.int %1572, %int0_488, %int3_495 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1604 = torch.aten.item %1603 : !torch.vtensor<[1],si64> -> !torch.int | |
%1605 = torch.aten.eq.int %1604, %int0_488 : !torch.int, !torch.int -> !torch.bool | |
%1606 = torch.aten.Int.bool %1605 : !torch.bool -> !torch.int | |
%int4_496 = torch.constant.int 4 | |
%1607 = torch.aten.select.int %1572, %int0_488, %int4_496 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1608 = torch.aten.item %1607 : !torch.vtensor<[1],si64> -> !torch.int | |
%1609 = torch.aten.eq.int %1608, %int0_488 : !torch.int, !torch.int -> !torch.bool | |
%1610 = torch.aten.Int.bool %1609 : !torch.bool -> !torch.int | |
%1611 = torch.prim.ListConstruct %1582, %1592, %1602, %1604, %1608 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1612 = torch.aten.reshape %1571, %1611 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%1613 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_497 = torch.constant.int 0 | |
%int1_498 = torch.constant.int 1 | |
%int0_499 = torch.constant.int 0 | |
%1614 = torch.aten.select.int %1613, %int0_497, %int0_499 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1615 = torch.aten.item %1614 : !torch.vtensor<[1],si64> -> !torch.int | |
%1616 = torch.aten.lt.int %1615, %int0_497 : !torch.int, !torch.int -> !torch.bool | |
%1617 = torch.aten.Int.bool %1616 : !torch.bool -> !torch.int | |
%1618 = torch.aten.mul.int %1617, %int1_498 : !torch.int, !torch.int -> !torch.int | |
%1619 = torch.aten.add.int %1615, %1618 : !torch.int, !torch.int -> !torch.int | |
%int1_500 = torch.constant.int 1 | |
%1620 = torch.aten.select.int %1613, %int0_497, %int1_500 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1621 = torch.aten.item %1620 : !torch.vtensor<[1],si64> -> !torch.int | |
%1622 = torch.aten.lt.int %1621, %int0_497 : !torch.int, !torch.int -> !torch.bool | |
%1623 = torch.aten.Int.bool %1622 : !torch.bool -> !torch.int | |
%1624 = torch.aten.mul.int %1623, %int1_498 : !torch.int, !torch.int -> !torch.int | |
%1625 = torch.aten.add.int %1621, %1624 : !torch.int, !torch.int -> !torch.int | |
%int2_501 = torch.constant.int 2 | |
%1626 = torch.aten.select.int %1613, %int0_497, %int2_501 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1627 = torch.aten.item %1626 : !torch.vtensor<[1],si64> -> !torch.int | |
%1628 = torch.aten.lt.int %1627, %int0_497 : !torch.int, !torch.int -> !torch.bool | |
%1629 = torch.aten.Int.bool %1628 : !torch.bool -> !torch.int | |
%1630 = torch.aten.mul.int %1629, %int1_498 : !torch.int, !torch.int -> !torch.int | |
%1631 = torch.aten.add.int %1627, %1630 : !torch.int, !torch.int -> !torch.int | |
%1632 = torch.prim.ListConstruct %1619, %1625, %1631 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_502 = torch.constant.bool false | |
%none_503 = torch.constant.none | |
%1633 = torch.aten.tensor %1632, %none_503, %none_503, %false_502 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_504, %indices_505 = torch.aten.sort %1633, %int0_497, %false_502 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_506 = torch.constant.int 0 | |
%1634 = torch.aten.select.int %values_504, %int0_497, %int0_506 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1635 = torch.aten.item %1634 : !torch.vtensor<[1],si64> -> !torch.int | |
%1636 = torch.aten.unsqueeze %1523, %1635 : !torch.vtensor<[128],f32>, !torch.int -> !torch.vtensor | |
%int1_507 = torch.constant.int 1 | |
%1637 = torch.aten.select.int %values_504, %int0_497, %int1_507 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1638 = torch.aten.item %1637 : !torch.vtensor<[1],si64> -> !torch.int | |
%1639 = torch.aten.unsqueeze %1636, %1638 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_508 = torch.constant.int 2 | |
%1640 = torch.aten.select.int %values_504, %int0_497, %int2_508 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1641 = torch.aten.item %1640 : !torch.vtensor<[1],si64> -> !torch.int | |
%1642 = torch.aten.unsqueeze %1639, %1641 : !torch.vtensor, !torch.int -> !torch.vtensor<[128,1,1,1],f32> | |
%1643 = torch.aten.mul.Tensor %1612, %1642 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[128,1,1,1],f32> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1644 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_509 = torch.constant.int 0 | |
%int1_510 = torch.constant.int 1 | |
%int0_511 = torch.constant.int 0 | |
%1645 = torch.aten.select.int %1644, %int0_509, %int0_511 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1646 = torch.aten.item %1645 : !torch.vtensor<[1],si64> -> !torch.int | |
%1647 = torch.aten.lt.int %1646, %int0_509 : !torch.int, !torch.int -> !torch.bool | |
%1648 = torch.aten.Int.bool %1647 : !torch.bool -> !torch.int | |
%1649 = torch.aten.mul.int %1648, %int1_510 : !torch.int, !torch.int -> !torch.int | |
%1650 = torch.aten.add.int %1646, %1649 : !torch.int, !torch.int -> !torch.int | |
%int1_512 = torch.constant.int 1 | |
%1651 = torch.aten.select.int %1644, %int0_509, %int1_512 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1652 = torch.aten.item %1651 : !torch.vtensor<[1],si64> -> !torch.int | |
%1653 = torch.aten.lt.int %1652, %int0_509 : !torch.int, !torch.int -> !torch.bool | |
%1654 = torch.aten.Int.bool %1653 : !torch.bool -> !torch.int | |
%1655 = torch.aten.mul.int %1654, %int1_510 : !torch.int, !torch.int -> !torch.int | |
%1656 = torch.aten.add.int %1652, %1655 : !torch.int, !torch.int -> !torch.int | |
%int2_513 = torch.constant.int 2 | |
%1657 = torch.aten.select.int %1644, %int0_509, %int2_513 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1658 = torch.aten.item %1657 : !torch.vtensor<[1],si64> -> !torch.int | |
%1659 = torch.aten.lt.int %1658, %int0_509 : !torch.int, !torch.int -> !torch.bool | |
%1660 = torch.aten.Int.bool %1659 : !torch.bool -> !torch.int | |
%1661 = torch.aten.mul.int %1660, %int1_510 : !torch.int, !torch.int -> !torch.int | |
%1662 = torch.aten.add.int %1658, %1661 : !torch.int, !torch.int -> !torch.int | |
%1663 = torch.prim.ListConstruct %1650, %1656, %1662 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_514 = torch.constant.bool false | |
%none_515 = torch.constant.none | |
%1664 = torch.aten.tensor %1663, %none_515, %none_515, %false_514 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_516, %indices_517 = torch.aten.sort %1664, %int0_509, %false_514 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_518 = torch.constant.int 0 | |
%1665 = torch.aten.select.int %values_516, %int0_509, %int0_518 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1666 = torch.aten.item %1665 : !torch.vtensor<[1],si64> -> !torch.int | |
%1667 = torch.aten.unsqueeze %1535, %1666 : !torch.vtensor<[128],f32>, !torch.int -> !torch.vtensor | |
%int1_519 = torch.constant.int 1 | |
%1668 = torch.aten.select.int %values_516, %int0_509, %int1_519 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1669 = torch.aten.item %1668 : !torch.vtensor<[1],si64> -> !torch.int | |
%1670 = torch.aten.unsqueeze %1667, %1669 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_520 = torch.constant.int 2 | |
%1671 = torch.aten.select.int %values_516, %int0_509, %int2_520 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1672 = torch.aten.item %1671 : !torch.vtensor<[1],si64> -> !torch.int | |
%1673 = torch.aten.unsqueeze %1670, %1672 : !torch.vtensor, !torch.int -> !torch.vtensor<[128,1,1,1],f32> | |
%int1_521 = torch.constant.int 1 | |
%1674 = torch.aten.add.Tensor %1643, %1673, %int1_521 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.vtensor<[128,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1675 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1676 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_522 = torch.constant.int 12 | |
%1677 = torch.aten.item %1675 : !torch.vtensor<[],f32> -> !torch.float | |
%1678 = torch.aten.item %1676 : !torch.vtensor<[],si8> -> !torch.int | |
%1679 = torch.aten.quantize_per_tensor %1674, %1677, %1678, %int12_522 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1680 = torch.aten.int_repr %1679 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],si8> | |
%1681 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1682 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1683 = torch.aten.item %1681 : !torch.vtensor<[],f32> -> !torch.float | |
%1684 = torch.aten.item %1682 : !torch.vtensor<[],si8> -> !torch.int | |
%1685 = torch.aten._make_per_tensor_quantized_tensor %1680, %1683, %1684 : !torch.vtensor<[?,128,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1686 = torch.aten.dequantize.self %1685 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1687 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1688 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_523 = torch.constant.int 12 | |
%1689 = torch.aten.item %1687 : !torch.vtensor<[],f32> -> !torch.float | |
%1690 = torch.aten.item %1688 : !torch.vtensor<[],si8> -> !torch.int | |
%1691 = torch.aten.quantize_per_tensor %23, %1689, %1690, %int12_523 : !torch.vtensor<[256,128,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[256,128,3,3,3],!torch.qint8> | |
%1692 = torch.aten.int_repr %1691 : !torch.vtensor<[256,128,3,3,3],!torch.qint8> -> !torch.vtensor<[256,128,3,3,3],si8> | |
%1693 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1694 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1695 = torch.aten.item %1693 : !torch.vtensor<[],f32> -> !torch.float | |
%1696 = torch.aten.item %1694 : !torch.vtensor<[],si8> -> !torch.int | |
%1697 = torch.aten._make_per_tensor_quantized_tensor %1692, %1695, %1696 : !torch.vtensor<[256,128,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[256,128,3,3,3],!torch.qint8> | |
%1698 = torch.aten.dequantize.self %1697 : !torch.vtensor<[256,128,3,3,3],!torch.qint8> -> !torch.vtensor<[256,128,3,3,3],f32> | |
%int1_524 = torch.constant.int 1 | |
%int1_525 = torch.constant.int 1 | |
%int1_526 = torch.constant.int 1 | |
%int1_527 = torch.constant.int 1 | |
%int1_528 = torch.constant.int 1 | |
%int1_529 = torch.constant.int 1 | |
%int1_530 = torch.constant.int 1 | |
%int1_531 = torch.constant.int 1 | |
%int1_532 = torch.constant.int 1 | |
%int0_533 = torch.constant.int 0 | |
%1699 = torch.prim.ListConstruct %int1_524, %int1_525, %int1_526 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1700 = torch.prim.ListConstruct %int1_527, %int1_528, %int1_529 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1701 = torch.prim.ListConstruct %int1_530, %int1_531, %int1_532 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1702 = torch.prim.ListConstruct %int0_533, %int0_533 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_534 = torch.constant.bool false | |
%none_535 = torch.constant.none | |
%int1_536 = torch.constant.int 1 | |
%1703 = torch.aten.convolution %1686, %1698, %none_535, %1701, %1699, %1700, %false_534, %1702, %int1_536 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.vtensor<[256,128,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,256,?,?,?],f32> | |
%1704 = torch.aten.relu %1703 : !torch.vtensor<[?,256,?,?,?],f32> -> !torch.vtensor<[?,256,?,?,?],f32> | |
%1705 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1706 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_537 = torch.constant.int 12 | |
%1707 = torch.aten.item %1705 : !torch.vtensor<[],f32> -> !torch.float | |
%1708 = torch.aten.item %1706 : !torch.vtensor<[],si8> -> !torch.int | |
%1709 = torch.aten.quantize_per_tensor %1704, %1707, %1708, %int12_537 : !torch.vtensor<[?,256,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,256,?,?,?],!torch.qint8> | |
%1710 = torch.aten.int_repr %1709 : !torch.vtensor<[?,256,?,?,?],!torch.qint8> -> !torch.vtensor<[?,256,?,?,?],si8> | |
%1711 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1712 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1713 = torch.aten.item %1711 : !torch.vtensor<[],f32> -> !torch.float | |
%1714 = torch.aten.item %1712 : !torch.vtensor<[],si8> -> !torch.int | |
%1715 = torch.aten._make_per_tensor_quantized_tensor %1710, %1713, %1714 : !torch.vtensor<[?,256,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,256,?,?,?],!torch.qint8> | |
%1716 = torch.aten.dequantize.self %1715 : !torch.vtensor<[?,256,?,?,?],!torch.qint8> -> !torch.vtensor<[?,256,?,?,?],f32> | |
%1717 = torch.vtensor.literal(dense<16> : tensor<si64>) : !torch.vtensor<[],si64> | |
%1718 = torch.vtensor.literal(dense<32> : tensor<si64>) : !torch.vtensor<[],si64> | |
%1719 = torch.vtensor.literal(dense<32> : tensor<si64>) : !torch.vtensor<[],si64> | |
%1720 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_538 = torch.constant.int 0 | |
%int0_539 = torch.constant.int 0 | |
%int0_540 = torch.constant.int 0 | |
%1721 = torch.aten.select.int %1720, %int0_538, %int0_540 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1722 = torch.aten.item %1721 : !torch.vtensor<[1],si64> -> !torch.int | |
%1723 = torch.aten.lt.int %1722, %int0_538 : !torch.int, !torch.int -> !torch.bool | |
%1724 = torch.aten.Int.bool %1723 : !torch.bool -> !torch.int | |
%1725 = torch.aten.mul.int %1724, %int0_539 : !torch.int, !torch.int -> !torch.int | |
%1726 = torch.aten.add.int %1722, %1725 : !torch.int, !torch.int -> !torch.int | |
%1727 = torch.prim.ListConstruct %1726 : (!torch.int) -> !torch.list<int> | |
%false_541 = torch.constant.bool false | |
%none_542 = torch.constant.none | |
%1728 = torch.aten.tensor %1727, %none_542, %none_542, %false_541 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_543, %indices_544 = torch.aten.sort %1728, %int0_538, %false_541 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_545 = torch.constant.int 0 | |
%1729 = torch.aten.select.int %values_543, %int0_538, %int0_545 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1730 = torch.aten.item %1729 : !torch.vtensor<[1],si64> -> !torch.int | |
%1731 = torch.aten.unsqueeze %1717, %1730 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%1732 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_546 = torch.constant.int 0 | |
%int0_547 = torch.constant.int 0 | |
%int0_548 = torch.constant.int 0 | |
%1733 = torch.aten.select.int %1732, %int0_546, %int0_548 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1734 = torch.aten.item %1733 : !torch.vtensor<[1],si64> -> !torch.int | |
%1735 = torch.aten.lt.int %1734, %int0_546 : !torch.int, !torch.int -> !torch.bool | |
%1736 = torch.aten.Int.bool %1735 : !torch.bool -> !torch.int | |
%1737 = torch.aten.mul.int %1736, %int0_547 : !torch.int, !torch.int -> !torch.int | |
%1738 = torch.aten.add.int %1734, %1737 : !torch.int, !torch.int -> !torch.int | |
%1739 = torch.prim.ListConstruct %1738 : (!torch.int) -> !torch.list<int> | |
%false_549 = torch.constant.bool false | |
%none_550 = torch.constant.none | |
%1740 = torch.aten.tensor %1739, %none_550, %none_550, %false_549 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_551, %indices_552 = torch.aten.sort %1740, %int0_546, %false_549 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_553 = torch.constant.int 0 | |
%1741 = torch.aten.select.int %values_551, %int0_546, %int0_553 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1742 = torch.aten.item %1741 : !torch.vtensor<[1],si64> -> !torch.int | |
%1743 = torch.aten.unsqueeze %1718, %1742 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%1744 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_554 = torch.constant.int 0 | |
%int0_555 = torch.constant.int 0 | |
%int0_556 = torch.constant.int 0 | |
%1745 = torch.aten.select.int %1744, %int0_554, %int0_556 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1746 = torch.aten.item %1745 : !torch.vtensor<[1],si64> -> !torch.int | |
%1747 = torch.aten.lt.int %1746, %int0_554 : !torch.int, !torch.int -> !torch.bool | |
%1748 = torch.aten.Int.bool %1747 : !torch.bool -> !torch.int | |
%1749 = torch.aten.mul.int %1748, %int0_555 : !torch.int, !torch.int -> !torch.int | |
%1750 = torch.aten.add.int %1746, %1749 : !torch.int, !torch.int -> !torch.int | |
%1751 = torch.prim.ListConstruct %1750 : (!torch.int) -> !torch.list<int> | |
%false_557 = torch.constant.bool false | |
%none_558 = torch.constant.none | |
%1752 = torch.aten.tensor %1751, %none_558, %none_558, %false_557 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_559, %indices_560 = torch.aten.sort %1752, %int0_554, %false_557 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_561 = torch.constant.int 0 | |
%1753 = torch.aten.select.int %values_559, %int0_554, %int0_561 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1754 = torch.aten.item %1753 : !torch.vtensor<[1],si64> -> !torch.int | |
%1755 = torch.aten.unsqueeze %1719, %1754 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%1756 = torch.prim.ListConstruct %1731, %1743, %1755 : (!torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>) -> !torch.list<vtensor> | |
%int0_562 = torch.constant.int 0 | |
%1757 = torch.aten.cat %1756, %int0_562 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[3],si64> | |
%1758 = torch.aten._shape_as_tensor %1716 : !torch.vtensor<[?,256,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%1759 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%1760 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%1761 = torch.vtensor.literal(dense<2> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%none_563 = torch.constant.none | |
%int1_564 = torch.constant.int 1 | |
%1762 = torch.prim.ListConstruct %int1_564 : (!torch.int) -> !torch.list<int> | |
%1763 = torch.aten.ones %1762, %none_563, %none_563, %none_563, %none_563 : !torch.list<int>, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[1],si64> | |
%int0_565 = torch.constant.int 0 | |
%int0_566 = torch.constant.int 0 | |
%1764 = torch.prim.NumToTensor.Scalar %int0_566 : !torch.int -> !torch.vtensor<[1],si64> | |
%1765 = torch.aten.index_select %1760, %int0_565, %1764 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%1766 = torch.aten.item %1765 : !torch.vtensor<[1],si64> -> !torch.int | |
%1767 = torch.aten.index_select %1761, %int0_565, %1764 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%1768 = torch.aten.item %1767 : !torch.vtensor<[1],si64> -> !torch.int | |
%1769 = torch.aten.index_select %1759, %int0_565, %1764 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%1770 = torch.aten.item %1769 : !torch.vtensor<[1],si64> -> !torch.int | |
%1771 = torch.aten.index_select %1763, %int0_565, %1764 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%1772 = torch.aten.item %1771 : !torch.vtensor<[1],si64> -> !torch.int | |
%1773 = torch.aten.slice.Tensor %1758, %1770, %1766, %1768, %1772 : !torch.vtensor<[5],si64>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[2],si64> | |
%int4_567 = torch.constant.int 4 | |
%none_568 = torch.constant.none | |
%false_569 = torch.constant.bool false | |
%1774 = torch.aten.to.dtype %1757, %int4_567, %false_569, %false_569, %none_568 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool, !torch.bool, !torch.none -> !torch.vtensor<[3],si64> | |
%1775 = torch.prim.ListConstruct %1773, %1774 : (!torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>) -> !torch.list<vtensor> | |
%int0_570 = torch.constant.int 0 | |
%1776 = torch.aten.cat %1775, %int0_570 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[5],si64> | |
%1777 = torch.operator "onnx.Resize"(%1716, %none, %none, %1776) {torch.onnx.coordinate_transformation_mode = "asymmetric", torch.onnx.cubic_coeff_a = -7.500000e-01 : f32, torch.onnx.mode = "nearest", torch.onnx.nearest_mode = "floor"} : (!torch.vtensor<[?,256,?,?,?],f32>, !torch.none, !torch.none, !torch.vtensor<[5],si64>) -> !torch.vtensor<[?,?,?,?,?],f32> | |
%1778 = torch.prim.ListConstruct %1289, %1777 : (!torch.vtensor<[?,128,?,?,?],f32>, !torch.vtensor<[?,?,?,?,?],f32>) -> !torch.list<vtensor> | |
%int1_571 = torch.constant.int 1 | |
%1779 = torch.aten.cat %1778, %int1_571 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[?,?,?,?,?],f32> | |
%1780 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1781 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_572 = torch.constant.int 12 | |
%1782 = torch.aten.item %1780 : !torch.vtensor<[],f32> -> !torch.float | |
%1783 = torch.aten.item %1781 : !torch.vtensor<[],si8> -> !torch.int | |
%1784 = torch.aten.quantize_per_tensor %1779, %1782, %1783, %int12_572 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,?,?,?,?],!torch.qint8> | |
%1785 = torch.aten.int_repr %1784 : !torch.vtensor<[?,?,?,?,?],!torch.qint8> -> !torch.vtensor<[?,?,?,?,?],si8> | |
%1786 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1787 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1788 = torch.aten.item %1786 : !torch.vtensor<[],f32> -> !torch.float | |
%1789 = torch.aten.item %1787 : !torch.vtensor<[],si8> -> !torch.int | |
%1790 = torch.aten._make_per_tensor_quantized_tensor %1785, %1788, %1789 : !torch.vtensor<[?,?,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,?,?,?,?],!torch.qint8> | |
%1791 = torch.aten.dequantize.self %1790 : !torch.vtensor<[?,?,?,?,?],!torch.qint8> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%1792 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1793 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_573 = torch.constant.int 12 | |
%1794 = torch.aten.item %1792 : !torch.vtensor<[],f32> -> !torch.float | |
%1795 = torch.aten.item %1793 : !torch.vtensor<[],si8> -> !torch.int | |
%1796 = torch.aten.quantize_per_tensor %24, %1794, %1795, %int12_573 : !torch.vtensor<[384],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[384],!torch.qint8> | |
%1797 = torch.aten.int_repr %1796 : !torch.vtensor<[384],!torch.qint8> -> !torch.vtensor<[384],si8> | |
%1798 = torch.vtensor.literal(dense<7.812500e-03> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1799 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1800 = torch.aten.item %1798 : !torch.vtensor<[],f32> -> !torch.float | |
%1801 = torch.aten.item %1799 : !torch.vtensor<[],si8> -> !torch.int | |
%1802 = torch.aten._make_per_tensor_quantized_tensor %1797, %1800, %1801 : !torch.vtensor<[384],si8>, !torch.float, !torch.int -> !torch.vtensor<[384],!torch.qint8> | |
%1803 = torch.aten.dequantize.self %1802 : !torch.vtensor<[384],!torch.qint8> -> !torch.vtensor<[384],f32> | |
%1804 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1805 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_574 = torch.constant.int 12 | |
%1806 = torch.aten.item %1804 : !torch.vtensor<[],f32> -> !torch.float | |
%1807 = torch.aten.item %1805 : !torch.vtensor<[],si8> -> !torch.int | |
%1808 = torch.aten.quantize_per_tensor %25, %1806, %1807, %int12_574 : !torch.vtensor<[384],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[384],!torch.qint8> | |
%1809 = torch.aten.int_repr %1808 : !torch.vtensor<[384],!torch.qint8> -> !torch.vtensor<[384],si8> | |
%1810 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1811 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1812 = torch.aten.item %1810 : !torch.vtensor<[],f32> -> !torch.float | |
%1813 = torch.aten.item %1811 : !torch.vtensor<[],si8> -> !torch.int | |
%1814 = torch.aten._make_per_tensor_quantized_tensor %1809, %1812, %1813 : !torch.vtensor<[384],si8>, !torch.float, !torch.int -> !torch.vtensor<[384],!torch.qint8> | |
%1815 = torch.aten.dequantize.self %1814 : !torch.vtensor<[384],!torch.qint8> -> !torch.vtensor<[384],f32> | |
%1816 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_575 = torch.constant.int 0 | |
%int0_576 = torch.constant.int 0 | |
%1817 = torch.aten.select.int %1816, %int0_575, %int0_576 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1818 = torch.aten.item %1817 : !torch.vtensor<[1],si64> -> !torch.int | |
%1819 = torch.aten.eq.int %1818, %int0_575 : !torch.int, !torch.int -> !torch.bool | |
%1820 = torch.aten.Int.bool %1819 : !torch.bool -> !torch.int | |
%int0_577 = torch.constant.int 0 | |
%1821 = torch.aten.size.int %1791, %int0_577 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%1822 = torch.prim.NumToTensor.Scalar %1820 : !torch.int -> !torch.vtensor<[],i1> | |
%1823 = torch.prim.NumToTensor.Scalar %1821 : !torch.int -> !torch.vtensor<[],si64> | |
%1824 = torch.prim.NumToTensor.Scalar %1818 : !torch.int -> !torch.vtensor<[],si64> | |
%1825 = torch.aten.where.self %1822, %1823, %1824 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1826 = torch.aten.item %1825 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_578 = torch.constant.int 1 | |
%1827 = torch.aten.select.int %1816, %int0_575, %int1_578 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1828 = torch.aten.item %1827 : !torch.vtensor<[1],si64> -> !torch.int | |
%1829 = torch.aten.eq.int %1828, %int0_575 : !torch.int, !torch.int -> !torch.bool | |
%1830 = torch.aten.Int.bool %1829 : !torch.bool -> !torch.int | |
%int1_579 = torch.constant.int 1 | |
%1831 = torch.aten.size.int %1791, %int1_579 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%1832 = torch.prim.NumToTensor.Scalar %1830 : !torch.int -> !torch.vtensor<[],i1> | |
%1833 = torch.prim.NumToTensor.Scalar %1831 : !torch.int -> !torch.vtensor<[],si64> | |
%1834 = torch.prim.NumToTensor.Scalar %1828 : !torch.int -> !torch.vtensor<[],si64> | |
%1835 = torch.aten.where.self %1832, %1833, %1834 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1836 = torch.aten.item %1835 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_580 = torch.constant.int 2 | |
%1837 = torch.aten.select.int %1816, %int0_575, %int2_580 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1838 = torch.aten.item %1837 : !torch.vtensor<[1],si64> -> !torch.int | |
%1839 = torch.aten.eq.int %1838, %int0_575 : !torch.int, !torch.int -> !torch.bool | |
%1840 = torch.aten.Int.bool %1839 : !torch.bool -> !torch.int | |
%int2_581 = torch.constant.int 2 | |
%1841 = torch.aten.size.int %1791, %int2_581 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%1842 = torch.prim.NumToTensor.Scalar %1840 : !torch.int -> !torch.vtensor<[],i1> | |
%1843 = torch.prim.NumToTensor.Scalar %1841 : !torch.int -> !torch.vtensor<[],si64> | |
%1844 = torch.prim.NumToTensor.Scalar %1838 : !torch.int -> !torch.vtensor<[],si64> | |
%1845 = torch.aten.where.self %1842, %1843, %1844 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1846 = torch.aten.item %1845 : !torch.vtensor<[],si64> -> !torch.int | |
%1847 = torch.prim.ListConstruct %1826, %1836, %1846 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1848 = torch.aten.reshape %1791, %1847 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%1849 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%1850 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_582 = torch.constant.none | |
%true_583 = torch.constant.bool true | |
%false_584 = torch.constant.bool false | |
%float9.999990e-06_585 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_586 = torch.constant.float 0.000000e+00 | |
%1851 = torch.aten.instance_norm %1848, %1849, %1850, %none_582, %none_582, %true_583, %float0.000000e00_586, %float9.999990e-06_585, %false_584 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%1852 = torch.aten._shape_as_tensor %1791 : !torch.vtensor<[?,?,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_587 = torch.constant.int 0 | |
%int0_588 = torch.constant.int 0 | |
%1853 = torch.aten.select.int %1852, %int0_587, %int0_588 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1854 = torch.aten.item %1853 : !torch.vtensor<[1],si64> -> !torch.int | |
%1855 = torch.aten.eq.int %1854, %int0_587 : !torch.int, !torch.int -> !torch.bool | |
%1856 = torch.aten.Int.bool %1855 : !torch.bool -> !torch.int | |
%int0_589 = torch.constant.int 0 | |
%1857 = torch.aten.size.int %1851, %int0_589 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1858 = torch.prim.NumToTensor.Scalar %1856 : !torch.int -> !torch.vtensor<[],i1> | |
%1859 = torch.prim.NumToTensor.Scalar %1857 : !torch.int -> !torch.vtensor<[],si64> | |
%1860 = torch.prim.NumToTensor.Scalar %1854 : !torch.int -> !torch.vtensor<[],si64> | |
%1861 = torch.aten.where.self %1858, %1859, %1860 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1862 = torch.aten.item %1861 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_590 = torch.constant.int 1 | |
%1863 = torch.aten.select.int %1852, %int0_587, %int1_590 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1864 = torch.aten.item %1863 : !torch.vtensor<[1],si64> -> !torch.int | |
%1865 = torch.aten.eq.int %1864, %int0_587 : !torch.int, !torch.int -> !torch.bool | |
%1866 = torch.aten.Int.bool %1865 : !torch.bool -> !torch.int | |
%int1_591 = torch.constant.int 1 | |
%1867 = torch.aten.size.int %1851, %int1_591 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1868 = torch.prim.NumToTensor.Scalar %1866 : !torch.int -> !torch.vtensor<[],i1> | |
%1869 = torch.prim.NumToTensor.Scalar %1867 : !torch.int -> !torch.vtensor<[],si64> | |
%1870 = torch.prim.NumToTensor.Scalar %1864 : !torch.int -> !torch.vtensor<[],si64> | |
%1871 = torch.aten.where.self %1868, %1869, %1870 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1872 = torch.aten.item %1871 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_592 = torch.constant.int 2 | |
%1873 = torch.aten.select.int %1852, %int0_587, %int2_592 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1874 = torch.aten.item %1873 : !torch.vtensor<[1],si64> -> !torch.int | |
%1875 = torch.aten.eq.int %1874, %int0_587 : !torch.int, !torch.int -> !torch.bool | |
%1876 = torch.aten.Int.bool %1875 : !torch.bool -> !torch.int | |
%int2_593 = torch.constant.int 2 | |
%1877 = torch.aten.size.int %1851, %int2_593 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%1878 = torch.prim.NumToTensor.Scalar %1876 : !torch.int -> !torch.vtensor<[],i1> | |
%1879 = torch.prim.NumToTensor.Scalar %1877 : !torch.int -> !torch.vtensor<[],si64> | |
%1880 = torch.prim.NumToTensor.Scalar %1874 : !torch.int -> !torch.vtensor<[],si64> | |
%1881 = torch.aten.where.self %1878, %1879, %1880 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%1882 = torch.aten.item %1881 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_594 = torch.constant.int 3 | |
%1883 = torch.aten.select.int %1852, %int0_587, %int3_594 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1884 = torch.aten.item %1883 : !torch.vtensor<[1],si64> -> !torch.int | |
%1885 = torch.aten.eq.int %1884, %int0_587 : !torch.int, !torch.int -> !torch.bool | |
%1886 = torch.aten.Int.bool %1885 : !torch.bool -> !torch.int | |
%int4_595 = torch.constant.int 4 | |
%1887 = torch.aten.select.int %1852, %int0_587, %int4_595 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1888 = torch.aten.item %1887 : !torch.vtensor<[1],si64> -> !torch.int | |
%1889 = torch.aten.eq.int %1888, %int0_587 : !torch.int, !torch.int -> !torch.bool | |
%1890 = torch.aten.Int.bool %1889 : !torch.bool -> !torch.int | |
%1891 = torch.prim.ListConstruct %1862, %1872, %1882, %1884, %1888 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1892 = torch.aten.reshape %1851, %1891 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%1893 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_596 = torch.constant.int 0 | |
%int1_597 = torch.constant.int 1 | |
%int0_598 = torch.constant.int 0 | |
%1894 = torch.aten.select.int %1893, %int0_596, %int0_598 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1895 = torch.aten.item %1894 : !torch.vtensor<[1],si64> -> !torch.int | |
%1896 = torch.aten.lt.int %1895, %int0_596 : !torch.int, !torch.int -> !torch.bool | |
%1897 = torch.aten.Int.bool %1896 : !torch.bool -> !torch.int | |
%1898 = torch.aten.mul.int %1897, %int1_597 : !torch.int, !torch.int -> !torch.int | |
%1899 = torch.aten.add.int %1895, %1898 : !torch.int, !torch.int -> !torch.int | |
%int1_599 = torch.constant.int 1 | |
%1900 = torch.aten.select.int %1893, %int0_596, %int1_599 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1901 = torch.aten.item %1900 : !torch.vtensor<[1],si64> -> !torch.int | |
%1902 = torch.aten.lt.int %1901, %int0_596 : !torch.int, !torch.int -> !torch.bool | |
%1903 = torch.aten.Int.bool %1902 : !torch.bool -> !torch.int | |
%1904 = torch.aten.mul.int %1903, %int1_597 : !torch.int, !torch.int -> !torch.int | |
%1905 = torch.aten.add.int %1901, %1904 : !torch.int, !torch.int -> !torch.int | |
%int2_600 = torch.constant.int 2 | |
%1906 = torch.aten.select.int %1893, %int0_596, %int2_600 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1907 = torch.aten.item %1906 : !torch.vtensor<[1],si64> -> !torch.int | |
%1908 = torch.aten.lt.int %1907, %int0_596 : !torch.int, !torch.int -> !torch.bool | |
%1909 = torch.aten.Int.bool %1908 : !torch.bool -> !torch.int | |
%1910 = torch.aten.mul.int %1909, %int1_597 : !torch.int, !torch.int -> !torch.int | |
%1911 = torch.aten.add.int %1907, %1910 : !torch.int, !torch.int -> !torch.int | |
%1912 = torch.prim.ListConstruct %1899, %1905, %1911 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_601 = torch.constant.bool false | |
%none_602 = torch.constant.none | |
%1913 = torch.aten.tensor %1912, %none_602, %none_602, %false_601 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_603, %indices_604 = torch.aten.sort %1913, %int0_596, %false_601 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_605 = torch.constant.int 0 | |
%1914 = torch.aten.select.int %values_603, %int0_596, %int0_605 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1915 = torch.aten.item %1914 : !torch.vtensor<[1],si64> -> !torch.int | |
%1916 = torch.aten.unsqueeze %1803, %1915 : !torch.vtensor<[384],f32>, !torch.int -> !torch.vtensor | |
%int1_606 = torch.constant.int 1 | |
%1917 = torch.aten.select.int %values_603, %int0_596, %int1_606 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1918 = torch.aten.item %1917 : !torch.vtensor<[1],si64> -> !torch.int | |
%1919 = torch.aten.unsqueeze %1916, %1918 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_607 = torch.constant.int 2 | |
%1920 = torch.aten.select.int %values_603, %int0_596, %int2_607 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1921 = torch.aten.item %1920 : !torch.vtensor<[1],si64> -> !torch.int | |
%1922 = torch.aten.unsqueeze %1919, %1921 : !torch.vtensor, !torch.int -> !torch.vtensor<[384,1,1,1],f32> | |
%1923 = torch.aten.mul.Tensor %1892, %1922 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[384,1,1,1],f32> -> !torch.vtensor<[?,384,?,?,?],f32> | |
%1924 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_608 = torch.constant.int 0 | |
%int1_609 = torch.constant.int 1 | |
%int0_610 = torch.constant.int 0 | |
%1925 = torch.aten.select.int %1924, %int0_608, %int0_610 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1926 = torch.aten.item %1925 : !torch.vtensor<[1],si64> -> !torch.int | |
%1927 = torch.aten.lt.int %1926, %int0_608 : !torch.int, !torch.int -> !torch.bool | |
%1928 = torch.aten.Int.bool %1927 : !torch.bool -> !torch.int | |
%1929 = torch.aten.mul.int %1928, %int1_609 : !torch.int, !torch.int -> !torch.int | |
%1930 = torch.aten.add.int %1926, %1929 : !torch.int, !torch.int -> !torch.int | |
%int1_611 = torch.constant.int 1 | |
%1931 = torch.aten.select.int %1924, %int0_608, %int1_611 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1932 = torch.aten.item %1931 : !torch.vtensor<[1],si64> -> !torch.int | |
%1933 = torch.aten.lt.int %1932, %int0_608 : !torch.int, !torch.int -> !torch.bool | |
%1934 = torch.aten.Int.bool %1933 : !torch.bool -> !torch.int | |
%1935 = torch.aten.mul.int %1934, %int1_609 : !torch.int, !torch.int -> !torch.int | |
%1936 = torch.aten.add.int %1932, %1935 : !torch.int, !torch.int -> !torch.int | |
%int2_612 = torch.constant.int 2 | |
%1937 = torch.aten.select.int %1924, %int0_608, %int2_612 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1938 = torch.aten.item %1937 : !torch.vtensor<[1],si64> -> !torch.int | |
%1939 = torch.aten.lt.int %1938, %int0_608 : !torch.int, !torch.int -> !torch.bool | |
%1940 = torch.aten.Int.bool %1939 : !torch.bool -> !torch.int | |
%1941 = torch.aten.mul.int %1940, %int1_609 : !torch.int, !torch.int -> !torch.int | |
%1942 = torch.aten.add.int %1938, %1941 : !torch.int, !torch.int -> !torch.int | |
%1943 = torch.prim.ListConstruct %1930, %1936, %1942 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_613 = torch.constant.bool false | |
%none_614 = torch.constant.none | |
%1944 = torch.aten.tensor %1943, %none_614, %none_614, %false_613 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_615, %indices_616 = torch.aten.sort %1944, %int0_608, %false_613 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_617 = torch.constant.int 0 | |
%1945 = torch.aten.select.int %values_615, %int0_608, %int0_617 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1946 = torch.aten.item %1945 : !torch.vtensor<[1],si64> -> !torch.int | |
%1947 = torch.aten.unsqueeze %1815, %1946 : !torch.vtensor<[384],f32>, !torch.int -> !torch.vtensor | |
%int1_618 = torch.constant.int 1 | |
%1948 = torch.aten.select.int %values_615, %int0_608, %int1_618 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1949 = torch.aten.item %1948 : !torch.vtensor<[1],si64> -> !torch.int | |
%1950 = torch.aten.unsqueeze %1947, %1949 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_619 = torch.constant.int 2 | |
%1951 = torch.aten.select.int %values_615, %int0_608, %int2_619 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%1952 = torch.aten.item %1951 : !torch.vtensor<[1],si64> -> !torch.int | |
%1953 = torch.aten.unsqueeze %1950, %1952 : !torch.vtensor, !torch.int -> !torch.vtensor<[384,1,1,1],f32> | |
%int1_620 = torch.constant.int 1 | |
%1954 = torch.aten.add.Tensor %1923, %1953, %int1_620 : !torch.vtensor<[?,384,?,?,?],f32>, !torch.vtensor<[384,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,384,?,?,?],f32> | |
%1955 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1956 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_621 = torch.constant.int 12 | |
%1957 = torch.aten.item %1955 : !torch.vtensor<[],f32> -> !torch.float | |
%1958 = torch.aten.item %1956 : !torch.vtensor<[],si8> -> !torch.int | |
%1959 = torch.aten.quantize_per_tensor %1954, %1957, %1958, %int12_621 : !torch.vtensor<[?,384,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,384,?,?,?],!torch.qint8> | |
%1960 = torch.aten.int_repr %1959 : !torch.vtensor<[?,384,?,?,?],!torch.qint8> -> !torch.vtensor<[?,384,?,?,?],si8> | |
%1961 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1962 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1963 = torch.aten.item %1961 : !torch.vtensor<[],f32> -> !torch.float | |
%1964 = torch.aten.item %1962 : !torch.vtensor<[],si8> -> !torch.int | |
%1965 = torch.aten._make_per_tensor_quantized_tensor %1960, %1963, %1964 : !torch.vtensor<[?,384,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,384,?,?,?],!torch.qint8> | |
%1966 = torch.aten.dequantize.self %1965 : !torch.vtensor<[?,384,?,?,?],!torch.qint8> -> !torch.vtensor<[?,384,?,?,?],f32> | |
%1967 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1968 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_622 = torch.constant.int 12 | |
%1969 = torch.aten.item %1967 : !torch.vtensor<[],f32> -> !torch.float | |
%1970 = torch.aten.item %1968 : !torch.vtensor<[],si8> -> !torch.int | |
%1971 = torch.aten.quantize_per_tensor %26, %1969, %1970, %int12_622 : !torch.vtensor<[128,384,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128,384,3,3,3],!torch.qint8> | |
%1972 = torch.aten.int_repr %1971 : !torch.vtensor<[128,384,3,3,3],!torch.qint8> -> !torch.vtensor<[128,384,3,3,3],si8> | |
%1973 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1974 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1975 = torch.aten.item %1973 : !torch.vtensor<[],f32> -> !torch.float | |
%1976 = torch.aten.item %1974 : !torch.vtensor<[],si8> -> !torch.int | |
%1977 = torch.aten._make_per_tensor_quantized_tensor %1972, %1975, %1976 : !torch.vtensor<[128,384,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[128,384,3,3,3],!torch.qint8> | |
%1978 = torch.aten.dequantize.self %1977 : !torch.vtensor<[128,384,3,3,3],!torch.qint8> -> !torch.vtensor<[128,384,3,3,3],f32> | |
%int1_623 = torch.constant.int 1 | |
%int1_624 = torch.constant.int 1 | |
%int1_625 = torch.constant.int 1 | |
%int1_626 = torch.constant.int 1 | |
%int1_627 = torch.constant.int 1 | |
%int1_628 = torch.constant.int 1 | |
%int1_629 = torch.constant.int 1 | |
%int1_630 = torch.constant.int 1 | |
%int1_631 = torch.constant.int 1 | |
%int0_632 = torch.constant.int 0 | |
%1979 = torch.prim.ListConstruct %int1_623, %int1_624, %int1_625 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1980 = torch.prim.ListConstruct %int1_626, %int1_627, %int1_628 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1981 = torch.prim.ListConstruct %int1_629, %int1_630, %int1_631 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%1982 = torch.prim.ListConstruct %int0_632, %int0_632 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_633 = torch.constant.bool false | |
%none_634 = torch.constant.none | |
%int1_635 = torch.constant.int 1 | |
%1983 = torch.aten.convolution %1966, %1978, %none_634, %1981, %1979, %1980, %false_633, %1982, %int1_635 : !torch.vtensor<[?,384,?,?,?],f32>, !torch.vtensor<[128,384,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1984 = torch.aten.relu %1983 : !torch.vtensor<[?,128,?,?,?],f32> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1985 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1986 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_636 = torch.constant.int 12 | |
%1987 = torch.aten.item %1985 : !torch.vtensor<[],f32> -> !torch.float | |
%1988 = torch.aten.item %1986 : !torch.vtensor<[],si8> -> !torch.int | |
%1989 = torch.aten.quantize_per_tensor %1984, %1987, %1988, %int12_636 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1990 = torch.aten.int_repr %1989 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],si8> | |
%1991 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1992 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%1993 = torch.aten.item %1991 : !torch.vtensor<[],f32> -> !torch.float | |
%1994 = torch.aten.item %1992 : !torch.vtensor<[],si8> -> !torch.int | |
%1995 = torch.aten._make_per_tensor_quantized_tensor %1990, %1993, %1994 : !torch.vtensor<[?,128,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%1996 = torch.aten.dequantize.self %1995 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%1997 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%1998 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_637 = torch.constant.int 12 | |
%1999 = torch.aten.item %1997 : !torch.vtensor<[],f32> -> !torch.float | |
%2000 = torch.aten.item %1998 : !torch.vtensor<[],si8> -> !torch.int | |
%2001 = torch.aten.quantize_per_tensor %27, %1999, %2000, %int12_637 : !torch.vtensor<[128],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%2002 = torch.aten.int_repr %2001 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],si8> | |
%2003 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2004 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2005 = torch.aten.item %2003 : !torch.vtensor<[],f32> -> !torch.float | |
%2006 = torch.aten.item %2004 : !torch.vtensor<[],si8> -> !torch.int | |
%2007 = torch.aten._make_per_tensor_quantized_tensor %2002, %2005, %2006 : !torch.vtensor<[128],si8>, !torch.float, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%2008 = torch.aten.dequantize.self %2007 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],f32> | |
%2009 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2010 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_638 = torch.constant.int 12 | |
%2011 = torch.aten.item %2009 : !torch.vtensor<[],f32> -> !torch.float | |
%2012 = torch.aten.item %2010 : !torch.vtensor<[],si8> -> !torch.int | |
%2013 = torch.aten.quantize_per_tensor %28, %2011, %2012, %int12_638 : !torch.vtensor<[128],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%2014 = torch.aten.int_repr %2013 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],si8> | |
%2015 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2016 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2017 = torch.aten.item %2015 : !torch.vtensor<[],f32> -> !torch.float | |
%2018 = torch.aten.item %2016 : !torch.vtensor<[],si8> -> !torch.int | |
%2019 = torch.aten._make_per_tensor_quantized_tensor %2014, %2017, %2018 : !torch.vtensor<[128],si8>, !torch.float, !torch.int -> !torch.vtensor<[128],!torch.qint8> | |
%2020 = torch.aten.dequantize.self %2019 : !torch.vtensor<[128],!torch.qint8> -> !torch.vtensor<[128],f32> | |
%2021 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_639 = torch.constant.int 0 | |
%int0_640 = torch.constant.int 0 | |
%2022 = torch.aten.select.int %2021, %int0_639, %int0_640 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2023 = torch.aten.item %2022 : !torch.vtensor<[1],si64> -> !torch.int | |
%2024 = torch.aten.eq.int %2023, %int0_639 : !torch.int, !torch.int -> !torch.bool | |
%2025 = torch.aten.Int.bool %2024 : !torch.bool -> !torch.int | |
%int0_641 = torch.constant.int 0 | |
%2026 = torch.aten.size.int %1996, %int0_641 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%2027 = torch.prim.NumToTensor.Scalar %2025 : !torch.int -> !torch.vtensor<[],i1> | |
%2028 = torch.prim.NumToTensor.Scalar %2026 : !torch.int -> !torch.vtensor<[],si64> | |
%2029 = torch.prim.NumToTensor.Scalar %2023 : !torch.int -> !torch.vtensor<[],si64> | |
%2030 = torch.aten.where.self %2027, %2028, %2029 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2031 = torch.aten.item %2030 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_642 = torch.constant.int 1 | |
%2032 = torch.aten.select.int %2021, %int0_639, %int1_642 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2033 = torch.aten.item %2032 : !torch.vtensor<[1],si64> -> !torch.int | |
%2034 = torch.aten.eq.int %2033, %int0_639 : !torch.int, !torch.int -> !torch.bool | |
%2035 = torch.aten.Int.bool %2034 : !torch.bool -> !torch.int | |
%int1_643 = torch.constant.int 1 | |
%2036 = torch.aten.size.int %1996, %int1_643 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%2037 = torch.prim.NumToTensor.Scalar %2035 : !torch.int -> !torch.vtensor<[],i1> | |
%2038 = torch.prim.NumToTensor.Scalar %2036 : !torch.int -> !torch.vtensor<[],si64> | |
%2039 = torch.prim.NumToTensor.Scalar %2033 : !torch.int -> !torch.vtensor<[],si64> | |
%2040 = torch.aten.where.self %2037, %2038, %2039 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2041 = torch.aten.item %2040 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_644 = torch.constant.int 2 | |
%2042 = torch.aten.select.int %2021, %int0_639, %int2_644 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2043 = torch.aten.item %2042 : !torch.vtensor<[1],si64> -> !torch.int | |
%2044 = torch.aten.eq.int %2043, %int0_639 : !torch.int, !torch.int -> !torch.bool | |
%2045 = torch.aten.Int.bool %2044 : !torch.bool -> !torch.int | |
%int2_645 = torch.constant.int 2 | |
%2046 = torch.aten.size.int %1996, %int2_645 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.int -> !torch.int | |
%2047 = torch.prim.NumToTensor.Scalar %2045 : !torch.int -> !torch.vtensor<[],i1> | |
%2048 = torch.prim.NumToTensor.Scalar %2046 : !torch.int -> !torch.vtensor<[],si64> | |
%2049 = torch.prim.NumToTensor.Scalar %2043 : !torch.int -> !torch.vtensor<[],si64> | |
%2050 = torch.aten.where.self %2047, %2048, %2049 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2051 = torch.aten.item %2050 : !torch.vtensor<[],si64> -> !torch.int | |
%2052 = torch.prim.ListConstruct %2031, %2041, %2051 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2053 = torch.aten.reshape %1996, %2052 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%2054 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%2055 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_646 = torch.constant.none | |
%true_647 = torch.constant.bool true | |
%false_648 = torch.constant.bool false | |
%float9.999990e-06_649 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_650 = torch.constant.float 0.000000e+00 | |
%2056 = torch.aten.instance_norm %2053, %2054, %2055, %none_646, %none_646, %true_647, %float0.000000e00_650, %float9.999990e-06_649, %false_648 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%2057 = torch.aten._shape_as_tensor %1996 : !torch.vtensor<[?,128,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_651 = torch.constant.int 0 | |
%int0_652 = torch.constant.int 0 | |
%2058 = torch.aten.select.int %2057, %int0_651, %int0_652 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2059 = torch.aten.item %2058 : !torch.vtensor<[1],si64> -> !torch.int | |
%2060 = torch.aten.eq.int %2059, %int0_651 : !torch.int, !torch.int -> !torch.bool | |
%2061 = torch.aten.Int.bool %2060 : !torch.bool -> !torch.int | |
%int0_653 = torch.constant.int 0 | |
%2062 = torch.aten.size.int %2056, %int0_653 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2063 = torch.prim.NumToTensor.Scalar %2061 : !torch.int -> !torch.vtensor<[],i1> | |
%2064 = torch.prim.NumToTensor.Scalar %2062 : !torch.int -> !torch.vtensor<[],si64> | |
%2065 = torch.prim.NumToTensor.Scalar %2059 : !torch.int -> !torch.vtensor<[],si64> | |
%2066 = torch.aten.where.self %2063, %2064, %2065 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2067 = torch.aten.item %2066 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_654 = torch.constant.int 1 | |
%2068 = torch.aten.select.int %2057, %int0_651, %int1_654 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2069 = torch.aten.item %2068 : !torch.vtensor<[1],si64> -> !torch.int | |
%2070 = torch.aten.eq.int %2069, %int0_651 : !torch.int, !torch.int -> !torch.bool | |
%2071 = torch.aten.Int.bool %2070 : !torch.bool -> !torch.int | |
%int1_655 = torch.constant.int 1 | |
%2072 = torch.aten.size.int %2056, %int1_655 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2073 = torch.prim.NumToTensor.Scalar %2071 : !torch.int -> !torch.vtensor<[],i1> | |
%2074 = torch.prim.NumToTensor.Scalar %2072 : !torch.int -> !torch.vtensor<[],si64> | |
%2075 = torch.prim.NumToTensor.Scalar %2069 : !torch.int -> !torch.vtensor<[],si64> | |
%2076 = torch.aten.where.self %2073, %2074, %2075 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2077 = torch.aten.item %2076 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_656 = torch.constant.int 2 | |
%2078 = torch.aten.select.int %2057, %int0_651, %int2_656 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2079 = torch.aten.item %2078 : !torch.vtensor<[1],si64> -> !torch.int | |
%2080 = torch.aten.eq.int %2079, %int0_651 : !torch.int, !torch.int -> !torch.bool | |
%2081 = torch.aten.Int.bool %2080 : !torch.bool -> !torch.int | |
%int2_657 = torch.constant.int 2 | |
%2082 = torch.aten.size.int %2056, %int2_657 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2083 = torch.prim.NumToTensor.Scalar %2081 : !torch.int -> !torch.vtensor<[],i1> | |
%2084 = torch.prim.NumToTensor.Scalar %2082 : !torch.int -> !torch.vtensor<[],si64> | |
%2085 = torch.prim.NumToTensor.Scalar %2079 : !torch.int -> !torch.vtensor<[],si64> | |
%2086 = torch.aten.where.self %2083, %2084, %2085 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2087 = torch.aten.item %2086 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_658 = torch.constant.int 3 | |
%2088 = torch.aten.select.int %2057, %int0_651, %int3_658 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2089 = torch.aten.item %2088 : !torch.vtensor<[1],si64> -> !torch.int | |
%2090 = torch.aten.eq.int %2089, %int0_651 : !torch.int, !torch.int -> !torch.bool | |
%2091 = torch.aten.Int.bool %2090 : !torch.bool -> !torch.int | |
%int4_659 = torch.constant.int 4 | |
%2092 = torch.aten.select.int %2057, %int0_651, %int4_659 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2093 = torch.aten.item %2092 : !torch.vtensor<[1],si64> -> !torch.int | |
%2094 = torch.aten.eq.int %2093, %int0_651 : !torch.int, !torch.int -> !torch.bool | |
%2095 = torch.aten.Int.bool %2094 : !torch.bool -> !torch.int | |
%2096 = torch.prim.ListConstruct %2067, %2077, %2087, %2089, %2093 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2097 = torch.aten.reshape %2056, %2096 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2098 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_660 = torch.constant.int 0 | |
%int1_661 = torch.constant.int 1 | |
%int0_662 = torch.constant.int 0 | |
%2099 = torch.aten.select.int %2098, %int0_660, %int0_662 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2100 = torch.aten.item %2099 : !torch.vtensor<[1],si64> -> !torch.int | |
%2101 = torch.aten.lt.int %2100, %int0_660 : !torch.int, !torch.int -> !torch.bool | |
%2102 = torch.aten.Int.bool %2101 : !torch.bool -> !torch.int | |
%2103 = torch.aten.mul.int %2102, %int1_661 : !torch.int, !torch.int -> !torch.int | |
%2104 = torch.aten.add.int %2100, %2103 : !torch.int, !torch.int -> !torch.int | |
%int1_663 = torch.constant.int 1 | |
%2105 = torch.aten.select.int %2098, %int0_660, %int1_663 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2106 = torch.aten.item %2105 : !torch.vtensor<[1],si64> -> !torch.int | |
%2107 = torch.aten.lt.int %2106, %int0_660 : !torch.int, !torch.int -> !torch.bool | |
%2108 = torch.aten.Int.bool %2107 : !torch.bool -> !torch.int | |
%2109 = torch.aten.mul.int %2108, %int1_661 : !torch.int, !torch.int -> !torch.int | |
%2110 = torch.aten.add.int %2106, %2109 : !torch.int, !torch.int -> !torch.int | |
%int2_664 = torch.constant.int 2 | |
%2111 = torch.aten.select.int %2098, %int0_660, %int2_664 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2112 = torch.aten.item %2111 : !torch.vtensor<[1],si64> -> !torch.int | |
%2113 = torch.aten.lt.int %2112, %int0_660 : !torch.int, !torch.int -> !torch.bool | |
%2114 = torch.aten.Int.bool %2113 : !torch.bool -> !torch.int | |
%2115 = torch.aten.mul.int %2114, %int1_661 : !torch.int, !torch.int -> !torch.int | |
%2116 = torch.aten.add.int %2112, %2115 : !torch.int, !torch.int -> !torch.int | |
%2117 = torch.prim.ListConstruct %2104, %2110, %2116 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_665 = torch.constant.bool false | |
%none_666 = torch.constant.none | |
%2118 = torch.aten.tensor %2117, %none_666, %none_666, %false_665 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_667, %indices_668 = torch.aten.sort %2118, %int0_660, %false_665 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_669 = torch.constant.int 0 | |
%2119 = torch.aten.select.int %values_667, %int0_660, %int0_669 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2120 = torch.aten.item %2119 : !torch.vtensor<[1],si64> -> !torch.int | |
%2121 = torch.aten.unsqueeze %2008, %2120 : !torch.vtensor<[128],f32>, !torch.int -> !torch.vtensor | |
%int1_670 = torch.constant.int 1 | |
%2122 = torch.aten.select.int %values_667, %int0_660, %int1_670 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2123 = torch.aten.item %2122 : !torch.vtensor<[1],si64> -> !torch.int | |
%2124 = torch.aten.unsqueeze %2121, %2123 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_671 = torch.constant.int 2 | |
%2125 = torch.aten.select.int %values_667, %int0_660, %int2_671 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2126 = torch.aten.item %2125 : !torch.vtensor<[1],si64> -> !torch.int | |
%2127 = torch.aten.unsqueeze %2124, %2126 : !torch.vtensor, !torch.int -> !torch.vtensor<[128,1,1,1],f32> | |
%2128 = torch.aten.mul.Tensor %2097, %2127 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[128,1,1,1],f32> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%2129 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_672 = torch.constant.int 0 | |
%int1_673 = torch.constant.int 1 | |
%int0_674 = torch.constant.int 0 | |
%2130 = torch.aten.select.int %2129, %int0_672, %int0_674 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2131 = torch.aten.item %2130 : !torch.vtensor<[1],si64> -> !torch.int | |
%2132 = torch.aten.lt.int %2131, %int0_672 : !torch.int, !torch.int -> !torch.bool | |
%2133 = torch.aten.Int.bool %2132 : !torch.bool -> !torch.int | |
%2134 = torch.aten.mul.int %2133, %int1_673 : !torch.int, !torch.int -> !torch.int | |
%2135 = torch.aten.add.int %2131, %2134 : !torch.int, !torch.int -> !torch.int | |
%int1_675 = torch.constant.int 1 | |
%2136 = torch.aten.select.int %2129, %int0_672, %int1_675 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2137 = torch.aten.item %2136 : !torch.vtensor<[1],si64> -> !torch.int | |
%2138 = torch.aten.lt.int %2137, %int0_672 : !torch.int, !torch.int -> !torch.bool | |
%2139 = torch.aten.Int.bool %2138 : !torch.bool -> !torch.int | |
%2140 = torch.aten.mul.int %2139, %int1_673 : !torch.int, !torch.int -> !torch.int | |
%2141 = torch.aten.add.int %2137, %2140 : !torch.int, !torch.int -> !torch.int | |
%int2_676 = torch.constant.int 2 | |
%2142 = torch.aten.select.int %2129, %int0_672, %int2_676 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2143 = torch.aten.item %2142 : !torch.vtensor<[1],si64> -> !torch.int | |
%2144 = torch.aten.lt.int %2143, %int0_672 : !torch.int, !torch.int -> !torch.bool | |
%2145 = torch.aten.Int.bool %2144 : !torch.bool -> !torch.int | |
%2146 = torch.aten.mul.int %2145, %int1_673 : !torch.int, !torch.int -> !torch.int | |
%2147 = torch.aten.add.int %2143, %2146 : !torch.int, !torch.int -> !torch.int | |
%2148 = torch.prim.ListConstruct %2135, %2141, %2147 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_677 = torch.constant.bool false | |
%none_678 = torch.constant.none | |
%2149 = torch.aten.tensor %2148, %none_678, %none_678, %false_677 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_679, %indices_680 = torch.aten.sort %2149, %int0_672, %false_677 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_681 = torch.constant.int 0 | |
%2150 = torch.aten.select.int %values_679, %int0_672, %int0_681 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2151 = torch.aten.item %2150 : !torch.vtensor<[1],si64> -> !torch.int | |
%2152 = torch.aten.unsqueeze %2020, %2151 : !torch.vtensor<[128],f32>, !torch.int -> !torch.vtensor | |
%int1_682 = torch.constant.int 1 | |
%2153 = torch.aten.select.int %values_679, %int0_672, %int1_682 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2154 = torch.aten.item %2153 : !torch.vtensor<[1],si64> -> !torch.int | |
%2155 = torch.aten.unsqueeze %2152, %2154 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_683 = torch.constant.int 2 | |
%2156 = torch.aten.select.int %values_679, %int0_672, %int2_683 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2157 = torch.aten.item %2156 : !torch.vtensor<[1],si64> -> !torch.int | |
%2158 = torch.aten.unsqueeze %2155, %2157 : !torch.vtensor, !torch.int -> !torch.vtensor<[128,1,1,1],f32> | |
%int1_684 = torch.constant.int 1 | |
%2159 = torch.aten.add.Tensor %2128, %2158, %int1_684 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.vtensor<[128,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,128,?,?,?],f32> | |
%2160 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2161 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_685 = torch.constant.int 12 | |
%2162 = torch.aten.item %2160 : !torch.vtensor<[],f32> -> !torch.float | |
%2163 = torch.aten.item %2161 : !torch.vtensor<[],si8> -> !torch.int | |
%2164 = torch.aten.quantize_per_tensor %2159, %2162, %2163, %int12_685 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%2165 = torch.aten.int_repr %2164 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],si8> | |
%2166 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2167 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2168 = torch.aten.item %2166 : !torch.vtensor<[],f32> -> !torch.float | |
%2169 = torch.aten.item %2167 : !torch.vtensor<[],si8> -> !torch.int | |
%2170 = torch.aten._make_per_tensor_quantized_tensor %2165, %2168, %2169 : !torch.vtensor<[?,128,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%2171 = torch.aten.dequantize.self %2170 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%2172 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2173 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_686 = torch.constant.int 12 | |
%2174 = torch.aten.item %2172 : !torch.vtensor<[],f32> -> !torch.float | |
%2175 = torch.aten.item %2173 : !torch.vtensor<[],si8> -> !torch.int | |
%2176 = torch.aten.quantize_per_tensor %29, %2174, %2175, %int12_686 : !torch.vtensor<[128,128,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[128,128,3,3,3],!torch.qint8> | |
%2177 = torch.aten.int_repr %2176 : !torch.vtensor<[128,128,3,3,3],!torch.qint8> -> !torch.vtensor<[128,128,3,3,3],si8> | |
%2178 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2179 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2180 = torch.aten.item %2178 : !torch.vtensor<[],f32> -> !torch.float | |
%2181 = torch.aten.item %2179 : !torch.vtensor<[],si8> -> !torch.int | |
%2182 = torch.aten._make_per_tensor_quantized_tensor %2177, %2180, %2181 : !torch.vtensor<[128,128,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[128,128,3,3,3],!torch.qint8> | |
%2183 = torch.aten.dequantize.self %2182 : !torch.vtensor<[128,128,3,3,3],!torch.qint8> -> !torch.vtensor<[128,128,3,3,3],f32> | |
%int1_687 = torch.constant.int 1 | |
%int1_688 = torch.constant.int 1 | |
%int1_689 = torch.constant.int 1 | |
%int1_690 = torch.constant.int 1 | |
%int1_691 = torch.constant.int 1 | |
%int1_692 = torch.constant.int 1 | |
%int1_693 = torch.constant.int 1 | |
%int1_694 = torch.constant.int 1 | |
%int1_695 = torch.constant.int 1 | |
%int0_696 = torch.constant.int 0 | |
%2184 = torch.prim.ListConstruct %int1_687, %int1_688, %int1_689 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2185 = torch.prim.ListConstruct %int1_690, %int1_691, %int1_692 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2186 = torch.prim.ListConstruct %int1_693, %int1_694, %int1_695 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2187 = torch.prim.ListConstruct %int0_696, %int0_696 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_697 = torch.constant.bool false | |
%none_698 = torch.constant.none | |
%int1_699 = torch.constant.int 1 | |
%2188 = torch.aten.convolution %2171, %2183, %none_698, %2186, %2184, %2185, %false_697, %2187, %int1_699 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.vtensor<[128,128,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,128,?,?,?],f32> | |
%2189 = torch.aten.relu %2188 : !torch.vtensor<[?,128,?,?,?],f32> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%2190 = torch.vtensor.literal(dense<5.000000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2191 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_700 = torch.constant.int 12 | |
%2192 = torch.aten.item %2190 : !torch.vtensor<[],f32> -> !torch.float | |
%2193 = torch.aten.item %2191 : !torch.vtensor<[],si8> -> !torch.int | |
%2194 = torch.aten.quantize_per_tensor %2189, %2192, %2193, %int12_700 : !torch.vtensor<[?,128,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%2195 = torch.aten.int_repr %2194 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],si8> | |
%2196 = torch.vtensor.literal(dense<5.000000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2197 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2198 = torch.aten.item %2196 : !torch.vtensor<[],f32> -> !torch.float | |
%2199 = torch.aten.item %2197 : !torch.vtensor<[],si8> -> !torch.int | |
%2200 = torch.aten._make_per_tensor_quantized_tensor %2195, %2198, %2199 : !torch.vtensor<[?,128,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,128,?,?,?],!torch.qint8> | |
%2201 = torch.aten.dequantize.self %2200 : !torch.vtensor<[?,128,?,?,?],!torch.qint8> -> !torch.vtensor<[?,128,?,?,?],f32> | |
%2202 = torch.vtensor.literal(dense<32> : tensor<si64>) : !torch.vtensor<[],si64> | |
%2203 = torch.vtensor.literal(dense<64> : tensor<si64>) : !torch.vtensor<[],si64> | |
%2204 = torch.vtensor.literal(dense<64> : tensor<si64>) : !torch.vtensor<[],si64> | |
%2205 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_701 = torch.constant.int 0 | |
%int0_702 = torch.constant.int 0 | |
%int0_703 = torch.constant.int 0 | |
%2206 = torch.aten.select.int %2205, %int0_701, %int0_703 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2207 = torch.aten.item %2206 : !torch.vtensor<[1],si64> -> !torch.int | |
%2208 = torch.aten.lt.int %2207, %int0_701 : !torch.int, !torch.int -> !torch.bool | |
%2209 = torch.aten.Int.bool %2208 : !torch.bool -> !torch.int | |
%2210 = torch.aten.mul.int %2209, %int0_702 : !torch.int, !torch.int -> !torch.int | |
%2211 = torch.aten.add.int %2207, %2210 : !torch.int, !torch.int -> !torch.int | |
%2212 = torch.prim.ListConstruct %2211 : (!torch.int) -> !torch.list<int> | |
%false_704 = torch.constant.bool false | |
%none_705 = torch.constant.none | |
%2213 = torch.aten.tensor %2212, %none_705, %none_705, %false_704 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_706, %indices_707 = torch.aten.sort %2213, %int0_701, %false_704 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_708 = torch.constant.int 0 | |
%2214 = torch.aten.select.int %values_706, %int0_701, %int0_708 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2215 = torch.aten.item %2214 : !torch.vtensor<[1],si64> -> !torch.int | |
%2216 = torch.aten.unsqueeze %2202, %2215 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%2217 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_709 = torch.constant.int 0 | |
%int0_710 = torch.constant.int 0 | |
%int0_711 = torch.constant.int 0 | |
%2218 = torch.aten.select.int %2217, %int0_709, %int0_711 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2219 = torch.aten.item %2218 : !torch.vtensor<[1],si64> -> !torch.int | |
%2220 = torch.aten.lt.int %2219, %int0_709 : !torch.int, !torch.int -> !torch.bool | |
%2221 = torch.aten.Int.bool %2220 : !torch.bool -> !torch.int | |
%2222 = torch.aten.mul.int %2221, %int0_710 : !torch.int, !torch.int -> !torch.int | |
%2223 = torch.aten.add.int %2219, %2222 : !torch.int, !torch.int -> !torch.int | |
%2224 = torch.prim.ListConstruct %2223 : (!torch.int) -> !torch.list<int> | |
%false_712 = torch.constant.bool false | |
%none_713 = torch.constant.none | |
%2225 = torch.aten.tensor %2224, %none_713, %none_713, %false_712 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_714, %indices_715 = torch.aten.sort %2225, %int0_709, %false_712 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_716 = torch.constant.int 0 | |
%2226 = torch.aten.select.int %values_714, %int0_709, %int0_716 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2227 = torch.aten.item %2226 : !torch.vtensor<[1],si64> -> !torch.int | |
%2228 = torch.aten.unsqueeze %2203, %2227 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%2229 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_717 = torch.constant.int 0 | |
%int0_718 = torch.constant.int 0 | |
%int0_719 = torch.constant.int 0 | |
%2230 = torch.aten.select.int %2229, %int0_717, %int0_719 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2231 = torch.aten.item %2230 : !torch.vtensor<[1],si64> -> !torch.int | |
%2232 = torch.aten.lt.int %2231, %int0_717 : !torch.int, !torch.int -> !torch.bool | |
%2233 = torch.aten.Int.bool %2232 : !torch.bool -> !torch.int | |
%2234 = torch.aten.mul.int %2233, %int0_718 : !torch.int, !torch.int -> !torch.int | |
%2235 = torch.aten.add.int %2231, %2234 : !torch.int, !torch.int -> !torch.int | |
%2236 = torch.prim.ListConstruct %2235 : (!torch.int) -> !torch.list<int> | |
%false_720 = torch.constant.bool false | |
%none_721 = torch.constant.none | |
%2237 = torch.aten.tensor %2236, %none_721, %none_721, %false_720 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_722, %indices_723 = torch.aten.sort %2237, %int0_717, %false_720 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_724 = torch.constant.int 0 | |
%2238 = torch.aten.select.int %values_722, %int0_717, %int0_724 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2239 = torch.aten.item %2238 : !torch.vtensor<[1],si64> -> !torch.int | |
%2240 = torch.aten.unsqueeze %2204, %2239 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%2241 = torch.prim.ListConstruct %2216, %2228, %2240 : (!torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>) -> !torch.list<vtensor> | |
%int0_725 = torch.constant.int 0 | |
%2242 = torch.aten.cat %2241, %int0_725 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[3],si64> | |
%2243 = torch.aten._shape_as_tensor %2201 : !torch.vtensor<[?,128,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%2244 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%2245 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%2246 = torch.vtensor.literal(dense<2> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%none_726 = torch.constant.none | |
%int1_727 = torch.constant.int 1 | |
%2247 = torch.prim.ListConstruct %int1_727 : (!torch.int) -> !torch.list<int> | |
%2248 = torch.aten.ones %2247, %none_726, %none_726, %none_726, %none_726 : !torch.list<int>, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[1],si64> | |
%int0_728 = torch.constant.int 0 | |
%int0_729 = torch.constant.int 0 | |
%2249 = torch.prim.NumToTensor.Scalar %int0_729 : !torch.int -> !torch.vtensor<[1],si64> | |
%2250 = torch.aten.index_select %2245, %int0_728, %2249 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%2251 = torch.aten.item %2250 : !torch.vtensor<[1],si64> -> !torch.int | |
%2252 = torch.aten.index_select %2246, %int0_728, %2249 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%2253 = torch.aten.item %2252 : !torch.vtensor<[1],si64> -> !torch.int | |
%2254 = torch.aten.index_select %2244, %int0_728, %2249 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%2255 = torch.aten.item %2254 : !torch.vtensor<[1],si64> -> !torch.int | |
%2256 = torch.aten.index_select %2248, %int0_728, %2249 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%2257 = torch.aten.item %2256 : !torch.vtensor<[1],si64> -> !torch.int | |
%2258 = torch.aten.slice.Tensor %2243, %2255, %2251, %2253, %2257 : !torch.vtensor<[5],si64>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[2],si64> | |
%int4_730 = torch.constant.int 4 | |
%none_731 = torch.constant.none | |
%false_732 = torch.constant.bool false | |
%2259 = torch.aten.to.dtype %2242, %int4_730, %false_732, %false_732, %none_731 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool, !torch.bool, !torch.none -> !torch.vtensor<[3],si64> | |
%2260 = torch.prim.ListConstruct %2258, %2259 : (!torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>) -> !torch.list<vtensor> | |
%int0_733 = torch.constant.int 0 | |
%2261 = torch.aten.cat %2260, %int0_733 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[5],si64> | |
%2262 = torch.operator "onnx.Resize"(%2201, %none, %none, %2261) {torch.onnx.coordinate_transformation_mode = "asymmetric", torch.onnx.cubic_coeff_a = -7.500000e-01 : f32, torch.onnx.mode = "nearest", torch.onnx.nearest_mode = "floor"} : (!torch.vtensor<[?,128,?,?,?],f32>, !torch.none, !torch.none, !torch.vtensor<[5],si64>) -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2263 = torch.prim.ListConstruct %862, %2262 : (!torch.vtensor<[?,64,?,?,?],f32>, !torch.vtensor<[?,?,?,?,?],f32>) -> !torch.list<vtensor> | |
%int1_734 = torch.constant.int 1 | |
%2264 = torch.aten.cat %2263, %int1_734 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2265 = torch.vtensor.literal(dense<5.000000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2266 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_735 = torch.constant.int 12 | |
%2267 = torch.aten.item %2265 : !torch.vtensor<[],f32> -> !torch.float | |
%2268 = torch.aten.item %2266 : !torch.vtensor<[],si8> -> !torch.int | |
%2269 = torch.aten.quantize_per_tensor %2264, %2267, %2268, %int12_735 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,?,?,?,?],!torch.qint8> | |
%2270 = torch.aten.int_repr %2269 : !torch.vtensor<[?,?,?,?,?],!torch.qint8> -> !torch.vtensor<[?,?,?,?,?],si8> | |
%2271 = torch.vtensor.literal(dense<5.000000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2272 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2273 = torch.aten.item %2271 : !torch.vtensor<[],f32> -> !torch.float | |
%2274 = torch.aten.item %2272 : !torch.vtensor<[],si8> -> !torch.int | |
%2275 = torch.aten._make_per_tensor_quantized_tensor %2270, %2273, %2274 : !torch.vtensor<[?,?,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,?,?,?,?],!torch.qint8> | |
%2276 = torch.aten.dequantize.self %2275 : !torch.vtensor<[?,?,?,?,?],!torch.qint8> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2277 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2278 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_736 = torch.constant.int 12 | |
%2279 = torch.aten.item %2277 : !torch.vtensor<[],f32> -> !torch.float | |
%2280 = torch.aten.item %2278 : !torch.vtensor<[],si8> -> !torch.int | |
%2281 = torch.aten.quantize_per_tensor %30, %2279, %2280, %int12_736 : !torch.vtensor<[192],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[192],!torch.qint8> | |
%2282 = torch.aten.int_repr %2281 : !torch.vtensor<[192],!torch.qint8> -> !torch.vtensor<[192],si8> | |
%2283 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2284 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2285 = torch.aten.item %2283 : !torch.vtensor<[],f32> -> !torch.float | |
%2286 = torch.aten.item %2284 : !torch.vtensor<[],si8> -> !torch.int | |
%2287 = torch.aten._make_per_tensor_quantized_tensor %2282, %2285, %2286 : !torch.vtensor<[192],si8>, !torch.float, !torch.int -> !torch.vtensor<[192],!torch.qint8> | |
%2288 = torch.aten.dequantize.self %2287 : !torch.vtensor<[192],!torch.qint8> -> !torch.vtensor<[192],f32> | |
%2289 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2290 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_737 = torch.constant.int 12 | |
%2291 = torch.aten.item %2289 : !torch.vtensor<[],f32> -> !torch.float | |
%2292 = torch.aten.item %2290 : !torch.vtensor<[],si8> -> !torch.int | |
%2293 = torch.aten.quantize_per_tensor %31, %2291, %2292, %int12_737 : !torch.vtensor<[192],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[192],!torch.qint8> | |
%2294 = torch.aten.int_repr %2293 : !torch.vtensor<[192],!torch.qint8> -> !torch.vtensor<[192],si8> | |
%2295 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2296 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2297 = torch.aten.item %2295 : !torch.vtensor<[],f32> -> !torch.float | |
%2298 = torch.aten.item %2296 : !torch.vtensor<[],si8> -> !torch.int | |
%2299 = torch.aten._make_per_tensor_quantized_tensor %2294, %2297, %2298 : !torch.vtensor<[192],si8>, !torch.float, !torch.int -> !torch.vtensor<[192],!torch.qint8> | |
%2300 = torch.aten.dequantize.self %2299 : !torch.vtensor<[192],!torch.qint8> -> !torch.vtensor<[192],f32> | |
%2301 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_738 = torch.constant.int 0 | |
%int0_739 = torch.constant.int 0 | |
%2302 = torch.aten.select.int %2301, %int0_738, %int0_739 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2303 = torch.aten.item %2302 : !torch.vtensor<[1],si64> -> !torch.int | |
%2304 = torch.aten.eq.int %2303, %int0_738 : !torch.int, !torch.int -> !torch.bool | |
%2305 = torch.aten.Int.bool %2304 : !torch.bool -> !torch.int | |
%int0_740 = torch.constant.int 0 | |
%2306 = torch.aten.size.int %2276, %int0_740 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%2307 = torch.prim.NumToTensor.Scalar %2305 : !torch.int -> !torch.vtensor<[],i1> | |
%2308 = torch.prim.NumToTensor.Scalar %2306 : !torch.int -> !torch.vtensor<[],si64> | |
%2309 = torch.prim.NumToTensor.Scalar %2303 : !torch.int -> !torch.vtensor<[],si64> | |
%2310 = torch.aten.where.self %2307, %2308, %2309 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2311 = torch.aten.item %2310 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_741 = torch.constant.int 1 | |
%2312 = torch.aten.select.int %2301, %int0_738, %int1_741 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2313 = torch.aten.item %2312 : !torch.vtensor<[1],si64> -> !torch.int | |
%2314 = torch.aten.eq.int %2313, %int0_738 : !torch.int, !torch.int -> !torch.bool | |
%2315 = torch.aten.Int.bool %2314 : !torch.bool -> !torch.int | |
%int1_742 = torch.constant.int 1 | |
%2316 = torch.aten.size.int %2276, %int1_742 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%2317 = torch.prim.NumToTensor.Scalar %2315 : !torch.int -> !torch.vtensor<[],i1> | |
%2318 = torch.prim.NumToTensor.Scalar %2316 : !torch.int -> !torch.vtensor<[],si64> | |
%2319 = torch.prim.NumToTensor.Scalar %2313 : !torch.int -> !torch.vtensor<[],si64> | |
%2320 = torch.aten.where.self %2317, %2318, %2319 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2321 = torch.aten.item %2320 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_743 = torch.constant.int 2 | |
%2322 = torch.aten.select.int %2301, %int0_738, %int2_743 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2323 = torch.aten.item %2322 : !torch.vtensor<[1],si64> -> !torch.int | |
%2324 = torch.aten.eq.int %2323, %int0_738 : !torch.int, !torch.int -> !torch.bool | |
%2325 = torch.aten.Int.bool %2324 : !torch.bool -> !torch.int | |
%int2_744 = torch.constant.int 2 | |
%2326 = torch.aten.size.int %2276, %int2_744 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%2327 = torch.prim.NumToTensor.Scalar %2325 : !torch.int -> !torch.vtensor<[],i1> | |
%2328 = torch.prim.NumToTensor.Scalar %2326 : !torch.int -> !torch.vtensor<[],si64> | |
%2329 = torch.prim.NumToTensor.Scalar %2323 : !torch.int -> !torch.vtensor<[],si64> | |
%2330 = torch.aten.where.self %2327, %2328, %2329 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2331 = torch.aten.item %2330 : !torch.vtensor<[],si64> -> !torch.int | |
%2332 = torch.prim.ListConstruct %2311, %2321, %2331 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2333 = torch.aten.reshape %2276, %2332 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%2334 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%2335 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_745 = torch.constant.none | |
%true_746 = torch.constant.bool true | |
%false_747 = torch.constant.bool false | |
%float9.999990e-06_748 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_749 = torch.constant.float 0.000000e+00 | |
%2336 = torch.aten.instance_norm %2333, %2334, %2335, %none_745, %none_745, %true_746, %float0.000000e00_749, %float9.999990e-06_748, %false_747 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%2337 = torch.aten._shape_as_tensor %2276 : !torch.vtensor<[?,?,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_750 = torch.constant.int 0 | |
%int0_751 = torch.constant.int 0 | |
%2338 = torch.aten.select.int %2337, %int0_750, %int0_751 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2339 = torch.aten.item %2338 : !torch.vtensor<[1],si64> -> !torch.int | |
%2340 = torch.aten.eq.int %2339, %int0_750 : !torch.int, !torch.int -> !torch.bool | |
%2341 = torch.aten.Int.bool %2340 : !torch.bool -> !torch.int | |
%int0_752 = torch.constant.int 0 | |
%2342 = torch.aten.size.int %2336, %int0_752 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2343 = torch.prim.NumToTensor.Scalar %2341 : !torch.int -> !torch.vtensor<[],i1> | |
%2344 = torch.prim.NumToTensor.Scalar %2342 : !torch.int -> !torch.vtensor<[],si64> | |
%2345 = torch.prim.NumToTensor.Scalar %2339 : !torch.int -> !torch.vtensor<[],si64> | |
%2346 = torch.aten.where.self %2343, %2344, %2345 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2347 = torch.aten.item %2346 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_753 = torch.constant.int 1 | |
%2348 = torch.aten.select.int %2337, %int0_750, %int1_753 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2349 = torch.aten.item %2348 : !torch.vtensor<[1],si64> -> !torch.int | |
%2350 = torch.aten.eq.int %2349, %int0_750 : !torch.int, !torch.int -> !torch.bool | |
%2351 = torch.aten.Int.bool %2350 : !torch.bool -> !torch.int | |
%int1_754 = torch.constant.int 1 | |
%2352 = torch.aten.size.int %2336, %int1_754 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2353 = torch.prim.NumToTensor.Scalar %2351 : !torch.int -> !torch.vtensor<[],i1> | |
%2354 = torch.prim.NumToTensor.Scalar %2352 : !torch.int -> !torch.vtensor<[],si64> | |
%2355 = torch.prim.NumToTensor.Scalar %2349 : !torch.int -> !torch.vtensor<[],si64> | |
%2356 = torch.aten.where.self %2353, %2354, %2355 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2357 = torch.aten.item %2356 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_755 = torch.constant.int 2 | |
%2358 = torch.aten.select.int %2337, %int0_750, %int2_755 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2359 = torch.aten.item %2358 : !torch.vtensor<[1],si64> -> !torch.int | |
%2360 = torch.aten.eq.int %2359, %int0_750 : !torch.int, !torch.int -> !torch.bool | |
%2361 = torch.aten.Int.bool %2360 : !torch.bool -> !torch.int | |
%int2_756 = torch.constant.int 2 | |
%2362 = torch.aten.size.int %2336, %int2_756 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2363 = torch.prim.NumToTensor.Scalar %2361 : !torch.int -> !torch.vtensor<[],i1> | |
%2364 = torch.prim.NumToTensor.Scalar %2362 : !torch.int -> !torch.vtensor<[],si64> | |
%2365 = torch.prim.NumToTensor.Scalar %2359 : !torch.int -> !torch.vtensor<[],si64> | |
%2366 = torch.aten.where.self %2363, %2364, %2365 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2367 = torch.aten.item %2366 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_757 = torch.constant.int 3 | |
%2368 = torch.aten.select.int %2337, %int0_750, %int3_757 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2369 = torch.aten.item %2368 : !torch.vtensor<[1],si64> -> !torch.int | |
%2370 = torch.aten.eq.int %2369, %int0_750 : !torch.int, !torch.int -> !torch.bool | |
%2371 = torch.aten.Int.bool %2370 : !torch.bool -> !torch.int | |
%int4_758 = torch.constant.int 4 | |
%2372 = torch.aten.select.int %2337, %int0_750, %int4_758 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2373 = torch.aten.item %2372 : !torch.vtensor<[1],si64> -> !torch.int | |
%2374 = torch.aten.eq.int %2373, %int0_750 : !torch.int, !torch.int -> !torch.bool | |
%2375 = torch.aten.Int.bool %2374 : !torch.bool -> !torch.int | |
%2376 = torch.prim.ListConstruct %2347, %2357, %2367, %2369, %2373 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2377 = torch.aten.reshape %2336, %2376 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2378 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_759 = torch.constant.int 0 | |
%int1_760 = torch.constant.int 1 | |
%int0_761 = torch.constant.int 0 | |
%2379 = torch.aten.select.int %2378, %int0_759, %int0_761 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2380 = torch.aten.item %2379 : !torch.vtensor<[1],si64> -> !torch.int | |
%2381 = torch.aten.lt.int %2380, %int0_759 : !torch.int, !torch.int -> !torch.bool | |
%2382 = torch.aten.Int.bool %2381 : !torch.bool -> !torch.int | |
%2383 = torch.aten.mul.int %2382, %int1_760 : !torch.int, !torch.int -> !torch.int | |
%2384 = torch.aten.add.int %2380, %2383 : !torch.int, !torch.int -> !torch.int | |
%int1_762 = torch.constant.int 1 | |
%2385 = torch.aten.select.int %2378, %int0_759, %int1_762 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2386 = torch.aten.item %2385 : !torch.vtensor<[1],si64> -> !torch.int | |
%2387 = torch.aten.lt.int %2386, %int0_759 : !torch.int, !torch.int -> !torch.bool | |
%2388 = torch.aten.Int.bool %2387 : !torch.bool -> !torch.int | |
%2389 = torch.aten.mul.int %2388, %int1_760 : !torch.int, !torch.int -> !torch.int | |
%2390 = torch.aten.add.int %2386, %2389 : !torch.int, !torch.int -> !torch.int | |
%int2_763 = torch.constant.int 2 | |
%2391 = torch.aten.select.int %2378, %int0_759, %int2_763 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2392 = torch.aten.item %2391 : !torch.vtensor<[1],si64> -> !torch.int | |
%2393 = torch.aten.lt.int %2392, %int0_759 : !torch.int, !torch.int -> !torch.bool | |
%2394 = torch.aten.Int.bool %2393 : !torch.bool -> !torch.int | |
%2395 = torch.aten.mul.int %2394, %int1_760 : !torch.int, !torch.int -> !torch.int | |
%2396 = torch.aten.add.int %2392, %2395 : !torch.int, !torch.int -> !torch.int | |
%2397 = torch.prim.ListConstruct %2384, %2390, %2396 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_764 = torch.constant.bool false | |
%none_765 = torch.constant.none | |
%2398 = torch.aten.tensor %2397, %none_765, %none_765, %false_764 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_766, %indices_767 = torch.aten.sort %2398, %int0_759, %false_764 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_768 = torch.constant.int 0 | |
%2399 = torch.aten.select.int %values_766, %int0_759, %int0_768 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2400 = torch.aten.item %2399 : !torch.vtensor<[1],si64> -> !torch.int | |
%2401 = torch.aten.unsqueeze %2288, %2400 : !torch.vtensor<[192],f32>, !torch.int -> !torch.vtensor | |
%int1_769 = torch.constant.int 1 | |
%2402 = torch.aten.select.int %values_766, %int0_759, %int1_769 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2403 = torch.aten.item %2402 : !torch.vtensor<[1],si64> -> !torch.int | |
%2404 = torch.aten.unsqueeze %2401, %2403 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_770 = torch.constant.int 2 | |
%2405 = torch.aten.select.int %values_766, %int0_759, %int2_770 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2406 = torch.aten.item %2405 : !torch.vtensor<[1],si64> -> !torch.int | |
%2407 = torch.aten.unsqueeze %2404, %2406 : !torch.vtensor, !torch.int -> !torch.vtensor<[192,1,1,1],f32> | |
%2408 = torch.aten.mul.Tensor %2377, %2407 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[192,1,1,1],f32> -> !torch.vtensor<[?,192,?,?,?],f32> | |
%2409 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_771 = torch.constant.int 0 | |
%int1_772 = torch.constant.int 1 | |
%int0_773 = torch.constant.int 0 | |
%2410 = torch.aten.select.int %2409, %int0_771, %int0_773 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2411 = torch.aten.item %2410 : !torch.vtensor<[1],si64> -> !torch.int | |
%2412 = torch.aten.lt.int %2411, %int0_771 : !torch.int, !torch.int -> !torch.bool | |
%2413 = torch.aten.Int.bool %2412 : !torch.bool -> !torch.int | |
%2414 = torch.aten.mul.int %2413, %int1_772 : !torch.int, !torch.int -> !torch.int | |
%2415 = torch.aten.add.int %2411, %2414 : !torch.int, !torch.int -> !torch.int | |
%int1_774 = torch.constant.int 1 | |
%2416 = torch.aten.select.int %2409, %int0_771, %int1_774 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2417 = torch.aten.item %2416 : !torch.vtensor<[1],si64> -> !torch.int | |
%2418 = torch.aten.lt.int %2417, %int0_771 : !torch.int, !torch.int -> !torch.bool | |
%2419 = torch.aten.Int.bool %2418 : !torch.bool -> !torch.int | |
%2420 = torch.aten.mul.int %2419, %int1_772 : !torch.int, !torch.int -> !torch.int | |
%2421 = torch.aten.add.int %2417, %2420 : !torch.int, !torch.int -> !torch.int | |
%int2_775 = torch.constant.int 2 | |
%2422 = torch.aten.select.int %2409, %int0_771, %int2_775 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2423 = torch.aten.item %2422 : !torch.vtensor<[1],si64> -> !torch.int | |
%2424 = torch.aten.lt.int %2423, %int0_771 : !torch.int, !torch.int -> !torch.bool | |
%2425 = torch.aten.Int.bool %2424 : !torch.bool -> !torch.int | |
%2426 = torch.aten.mul.int %2425, %int1_772 : !torch.int, !torch.int -> !torch.int | |
%2427 = torch.aten.add.int %2423, %2426 : !torch.int, !torch.int -> !torch.int | |
%2428 = torch.prim.ListConstruct %2415, %2421, %2427 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_776 = torch.constant.bool false | |
%none_777 = torch.constant.none | |
%2429 = torch.aten.tensor %2428, %none_777, %none_777, %false_776 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_778, %indices_779 = torch.aten.sort %2429, %int0_771, %false_776 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_780 = torch.constant.int 0 | |
%2430 = torch.aten.select.int %values_778, %int0_771, %int0_780 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2431 = torch.aten.item %2430 : !torch.vtensor<[1],si64> -> !torch.int | |
%2432 = torch.aten.unsqueeze %2300, %2431 : !torch.vtensor<[192],f32>, !torch.int -> !torch.vtensor | |
%int1_781 = torch.constant.int 1 | |
%2433 = torch.aten.select.int %values_778, %int0_771, %int1_781 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2434 = torch.aten.item %2433 : !torch.vtensor<[1],si64> -> !torch.int | |
%2435 = torch.aten.unsqueeze %2432, %2434 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_782 = torch.constant.int 2 | |
%2436 = torch.aten.select.int %values_778, %int0_771, %int2_782 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2437 = torch.aten.item %2436 : !torch.vtensor<[1],si64> -> !torch.int | |
%2438 = torch.aten.unsqueeze %2435, %2437 : !torch.vtensor, !torch.int -> !torch.vtensor<[192,1,1,1],f32> | |
%int1_783 = torch.constant.int 1 | |
%2439 = torch.aten.add.Tensor %2408, %2438, %int1_783 : !torch.vtensor<[?,192,?,?,?],f32>, !torch.vtensor<[192,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,192,?,?,?],f32> | |
%2440 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2441 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_784 = torch.constant.int 12 | |
%2442 = torch.aten.item %2440 : !torch.vtensor<[],f32> -> !torch.float | |
%2443 = torch.aten.item %2441 : !torch.vtensor<[],si8> -> !torch.int | |
%2444 = torch.aten.quantize_per_tensor %2439, %2442, %2443, %int12_784 : !torch.vtensor<[?,192,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,192,?,?,?],!torch.qint8> | |
%2445 = torch.aten.int_repr %2444 : !torch.vtensor<[?,192,?,?,?],!torch.qint8> -> !torch.vtensor<[?,192,?,?,?],si8> | |
%2446 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2447 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2448 = torch.aten.item %2446 : !torch.vtensor<[],f32> -> !torch.float | |
%2449 = torch.aten.item %2447 : !torch.vtensor<[],si8> -> !torch.int | |
%2450 = torch.aten._make_per_tensor_quantized_tensor %2445, %2448, %2449 : !torch.vtensor<[?,192,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,192,?,?,?],!torch.qint8> | |
%2451 = torch.aten.dequantize.self %2450 : !torch.vtensor<[?,192,?,?,?],!torch.qint8> -> !torch.vtensor<[?,192,?,?,?],f32> | |
%2452 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2453 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_785 = torch.constant.int 12 | |
%2454 = torch.aten.item %2452 : !torch.vtensor<[],f32> -> !torch.float | |
%2455 = torch.aten.item %2453 : !torch.vtensor<[],si8> -> !torch.int | |
%2456 = torch.aten.quantize_per_tensor %32, %2454, %2455, %int12_785 : !torch.vtensor<[64,192,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64,192,3,3,3],!torch.qint8> | |
%2457 = torch.aten.int_repr %2456 : !torch.vtensor<[64,192,3,3,3],!torch.qint8> -> !torch.vtensor<[64,192,3,3,3],si8> | |
%2458 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2459 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2460 = torch.aten.item %2458 : !torch.vtensor<[],f32> -> !torch.float | |
%2461 = torch.aten.item %2459 : !torch.vtensor<[],si8> -> !torch.int | |
%2462 = torch.aten._make_per_tensor_quantized_tensor %2457, %2460, %2461 : !torch.vtensor<[64,192,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[64,192,3,3,3],!torch.qint8> | |
%2463 = torch.aten.dequantize.self %2462 : !torch.vtensor<[64,192,3,3,3],!torch.qint8> -> !torch.vtensor<[64,192,3,3,3],f32> | |
%int1_786 = torch.constant.int 1 | |
%int1_787 = torch.constant.int 1 | |
%int1_788 = torch.constant.int 1 | |
%int1_789 = torch.constant.int 1 | |
%int1_790 = torch.constant.int 1 | |
%int1_791 = torch.constant.int 1 | |
%int1_792 = torch.constant.int 1 | |
%int1_793 = torch.constant.int 1 | |
%int1_794 = torch.constant.int 1 | |
%int0_795 = torch.constant.int 0 | |
%2464 = torch.prim.ListConstruct %int1_786, %int1_787, %int1_788 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2465 = torch.prim.ListConstruct %int1_789, %int1_790, %int1_791 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2466 = torch.prim.ListConstruct %int1_792, %int1_793, %int1_794 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2467 = torch.prim.ListConstruct %int0_795, %int0_795 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_796 = torch.constant.bool false | |
%none_797 = torch.constant.none | |
%int1_798 = torch.constant.int 1 | |
%2468 = torch.aten.convolution %2451, %2463, %none_797, %2466, %2464, %2465, %false_796, %2467, %int1_798 : !torch.vtensor<[?,192,?,?,?],f32>, !torch.vtensor<[64,192,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2469 = torch.aten.relu %2468 : !torch.vtensor<[?,64,?,?,?],f32> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2470 = torch.vtensor.literal(dense<1.000000e+00> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2471 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_799 = torch.constant.int 12 | |
%2472 = torch.aten.item %2470 : !torch.vtensor<[],f32> -> !torch.float | |
%2473 = torch.aten.item %2471 : !torch.vtensor<[],si8> -> !torch.int | |
%2474 = torch.aten.quantize_per_tensor %2469, %2472, %2473, %int12_799 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%2475 = torch.aten.int_repr %2474 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],si8> | |
%2476 = torch.vtensor.literal(dense<1.000000e+00> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2477 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2478 = torch.aten.item %2476 : !torch.vtensor<[],f32> -> !torch.float | |
%2479 = torch.aten.item %2477 : !torch.vtensor<[],si8> -> !torch.int | |
%2480 = torch.aten._make_per_tensor_quantized_tensor %2475, %2478, %2479 : !torch.vtensor<[?,64,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%2481 = torch.aten.dequantize.self %2480 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2482 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2483 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_800 = torch.constant.int 12 | |
%2484 = torch.aten.item %2482 : !torch.vtensor<[],f32> -> !torch.float | |
%2485 = torch.aten.item %2483 : !torch.vtensor<[],si8> -> !torch.int | |
%2486 = torch.aten.quantize_per_tensor %33, %2484, %2485, %int12_800 : !torch.vtensor<[64],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%2487 = torch.aten.int_repr %2486 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],si8> | |
%2488 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2489 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2490 = torch.aten.item %2488 : !torch.vtensor<[],f32> -> !torch.float | |
%2491 = torch.aten.item %2489 : !torch.vtensor<[],si8> -> !torch.int | |
%2492 = torch.aten._make_per_tensor_quantized_tensor %2487, %2490, %2491 : !torch.vtensor<[64],si8>, !torch.float, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%2493 = torch.aten.dequantize.self %2492 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],f32> | |
%2494 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2495 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_801 = torch.constant.int 12 | |
%2496 = torch.aten.item %2494 : !torch.vtensor<[],f32> -> !torch.float | |
%2497 = torch.aten.item %2495 : !torch.vtensor<[],si8> -> !torch.int | |
%2498 = torch.aten.quantize_per_tensor %34, %2496, %2497, %int12_801 : !torch.vtensor<[64],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%2499 = torch.aten.int_repr %2498 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],si8> | |
%2500 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2501 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2502 = torch.aten.item %2500 : !torch.vtensor<[],f32> -> !torch.float | |
%2503 = torch.aten.item %2501 : !torch.vtensor<[],si8> -> !torch.int | |
%2504 = torch.aten._make_per_tensor_quantized_tensor %2499, %2502, %2503 : !torch.vtensor<[64],si8>, !torch.float, !torch.int -> !torch.vtensor<[64],!torch.qint8> | |
%2505 = torch.aten.dequantize.self %2504 : !torch.vtensor<[64],!torch.qint8> -> !torch.vtensor<[64],f32> | |
%2506 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_802 = torch.constant.int 0 | |
%int0_803 = torch.constant.int 0 | |
%2507 = torch.aten.select.int %2506, %int0_802, %int0_803 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2508 = torch.aten.item %2507 : !torch.vtensor<[1],si64> -> !torch.int | |
%2509 = torch.aten.eq.int %2508, %int0_802 : !torch.int, !torch.int -> !torch.bool | |
%2510 = torch.aten.Int.bool %2509 : !torch.bool -> !torch.int | |
%int0_804 = torch.constant.int 0 | |
%2511 = torch.aten.size.int %2481, %int0_804 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%2512 = torch.prim.NumToTensor.Scalar %2510 : !torch.int -> !torch.vtensor<[],i1> | |
%2513 = torch.prim.NumToTensor.Scalar %2511 : !torch.int -> !torch.vtensor<[],si64> | |
%2514 = torch.prim.NumToTensor.Scalar %2508 : !torch.int -> !torch.vtensor<[],si64> | |
%2515 = torch.aten.where.self %2512, %2513, %2514 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2516 = torch.aten.item %2515 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_805 = torch.constant.int 1 | |
%2517 = torch.aten.select.int %2506, %int0_802, %int1_805 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2518 = torch.aten.item %2517 : !torch.vtensor<[1],si64> -> !torch.int | |
%2519 = torch.aten.eq.int %2518, %int0_802 : !torch.int, !torch.int -> !torch.bool | |
%2520 = torch.aten.Int.bool %2519 : !torch.bool -> !torch.int | |
%int1_806 = torch.constant.int 1 | |
%2521 = torch.aten.size.int %2481, %int1_806 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%2522 = torch.prim.NumToTensor.Scalar %2520 : !torch.int -> !torch.vtensor<[],i1> | |
%2523 = torch.prim.NumToTensor.Scalar %2521 : !torch.int -> !torch.vtensor<[],si64> | |
%2524 = torch.prim.NumToTensor.Scalar %2518 : !torch.int -> !torch.vtensor<[],si64> | |
%2525 = torch.aten.where.self %2522, %2523, %2524 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2526 = torch.aten.item %2525 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_807 = torch.constant.int 2 | |
%2527 = torch.aten.select.int %2506, %int0_802, %int2_807 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2528 = torch.aten.item %2527 : !torch.vtensor<[1],si64> -> !torch.int | |
%2529 = torch.aten.eq.int %2528, %int0_802 : !torch.int, !torch.int -> !torch.bool | |
%2530 = torch.aten.Int.bool %2529 : !torch.bool -> !torch.int | |
%int2_808 = torch.constant.int 2 | |
%2531 = torch.aten.size.int %2481, %int2_808 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.int -> !torch.int | |
%2532 = torch.prim.NumToTensor.Scalar %2530 : !torch.int -> !torch.vtensor<[],i1> | |
%2533 = torch.prim.NumToTensor.Scalar %2531 : !torch.int -> !torch.vtensor<[],si64> | |
%2534 = torch.prim.NumToTensor.Scalar %2528 : !torch.int -> !torch.vtensor<[],si64> | |
%2535 = torch.aten.where.self %2532, %2533, %2534 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2536 = torch.aten.item %2535 : !torch.vtensor<[],si64> -> !torch.int | |
%2537 = torch.prim.ListConstruct %2516, %2526, %2536 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2538 = torch.aten.reshape %2481, %2537 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%2539 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%2540 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_809 = torch.constant.none | |
%true_810 = torch.constant.bool true | |
%false_811 = torch.constant.bool false | |
%float9.999990e-06_812 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_813 = torch.constant.float 0.000000e+00 | |
%2541 = torch.aten.instance_norm %2538, %2539, %2540, %none_809, %none_809, %true_810, %float0.000000e00_813, %float9.999990e-06_812, %false_811 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%2542 = torch.aten._shape_as_tensor %2481 : !torch.vtensor<[?,64,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_814 = torch.constant.int 0 | |
%int0_815 = torch.constant.int 0 | |
%2543 = torch.aten.select.int %2542, %int0_814, %int0_815 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2544 = torch.aten.item %2543 : !torch.vtensor<[1],si64> -> !torch.int | |
%2545 = torch.aten.eq.int %2544, %int0_814 : !torch.int, !torch.int -> !torch.bool | |
%2546 = torch.aten.Int.bool %2545 : !torch.bool -> !torch.int | |
%int0_816 = torch.constant.int 0 | |
%2547 = torch.aten.size.int %2541, %int0_816 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2548 = torch.prim.NumToTensor.Scalar %2546 : !torch.int -> !torch.vtensor<[],i1> | |
%2549 = torch.prim.NumToTensor.Scalar %2547 : !torch.int -> !torch.vtensor<[],si64> | |
%2550 = torch.prim.NumToTensor.Scalar %2544 : !torch.int -> !torch.vtensor<[],si64> | |
%2551 = torch.aten.where.self %2548, %2549, %2550 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2552 = torch.aten.item %2551 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_817 = torch.constant.int 1 | |
%2553 = torch.aten.select.int %2542, %int0_814, %int1_817 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2554 = torch.aten.item %2553 : !torch.vtensor<[1],si64> -> !torch.int | |
%2555 = torch.aten.eq.int %2554, %int0_814 : !torch.int, !torch.int -> !torch.bool | |
%2556 = torch.aten.Int.bool %2555 : !torch.bool -> !torch.int | |
%int1_818 = torch.constant.int 1 | |
%2557 = torch.aten.size.int %2541, %int1_818 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2558 = torch.prim.NumToTensor.Scalar %2556 : !torch.int -> !torch.vtensor<[],i1> | |
%2559 = torch.prim.NumToTensor.Scalar %2557 : !torch.int -> !torch.vtensor<[],si64> | |
%2560 = torch.prim.NumToTensor.Scalar %2554 : !torch.int -> !torch.vtensor<[],si64> | |
%2561 = torch.aten.where.self %2558, %2559, %2560 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2562 = torch.aten.item %2561 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_819 = torch.constant.int 2 | |
%2563 = torch.aten.select.int %2542, %int0_814, %int2_819 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2564 = torch.aten.item %2563 : !torch.vtensor<[1],si64> -> !torch.int | |
%2565 = torch.aten.eq.int %2564, %int0_814 : !torch.int, !torch.int -> !torch.bool | |
%2566 = torch.aten.Int.bool %2565 : !torch.bool -> !torch.int | |
%int2_820 = torch.constant.int 2 | |
%2567 = torch.aten.size.int %2541, %int2_820 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2568 = torch.prim.NumToTensor.Scalar %2566 : !torch.int -> !torch.vtensor<[],i1> | |
%2569 = torch.prim.NumToTensor.Scalar %2567 : !torch.int -> !torch.vtensor<[],si64> | |
%2570 = torch.prim.NumToTensor.Scalar %2564 : !torch.int -> !torch.vtensor<[],si64> | |
%2571 = torch.aten.where.self %2568, %2569, %2570 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2572 = torch.aten.item %2571 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_821 = torch.constant.int 3 | |
%2573 = torch.aten.select.int %2542, %int0_814, %int3_821 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2574 = torch.aten.item %2573 : !torch.vtensor<[1],si64> -> !torch.int | |
%2575 = torch.aten.eq.int %2574, %int0_814 : !torch.int, !torch.int -> !torch.bool | |
%2576 = torch.aten.Int.bool %2575 : !torch.bool -> !torch.int | |
%int4_822 = torch.constant.int 4 | |
%2577 = torch.aten.select.int %2542, %int0_814, %int4_822 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2578 = torch.aten.item %2577 : !torch.vtensor<[1],si64> -> !torch.int | |
%2579 = torch.aten.eq.int %2578, %int0_814 : !torch.int, !torch.int -> !torch.bool | |
%2580 = torch.aten.Int.bool %2579 : !torch.bool -> !torch.int | |
%2581 = torch.prim.ListConstruct %2552, %2562, %2572, %2574, %2578 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2582 = torch.aten.reshape %2541, %2581 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2583 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_823 = torch.constant.int 0 | |
%int1_824 = torch.constant.int 1 | |
%int0_825 = torch.constant.int 0 | |
%2584 = torch.aten.select.int %2583, %int0_823, %int0_825 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2585 = torch.aten.item %2584 : !torch.vtensor<[1],si64> -> !torch.int | |
%2586 = torch.aten.lt.int %2585, %int0_823 : !torch.int, !torch.int -> !torch.bool | |
%2587 = torch.aten.Int.bool %2586 : !torch.bool -> !torch.int | |
%2588 = torch.aten.mul.int %2587, %int1_824 : !torch.int, !torch.int -> !torch.int | |
%2589 = torch.aten.add.int %2585, %2588 : !torch.int, !torch.int -> !torch.int | |
%int1_826 = torch.constant.int 1 | |
%2590 = torch.aten.select.int %2583, %int0_823, %int1_826 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2591 = torch.aten.item %2590 : !torch.vtensor<[1],si64> -> !torch.int | |
%2592 = torch.aten.lt.int %2591, %int0_823 : !torch.int, !torch.int -> !torch.bool | |
%2593 = torch.aten.Int.bool %2592 : !torch.bool -> !torch.int | |
%2594 = torch.aten.mul.int %2593, %int1_824 : !torch.int, !torch.int -> !torch.int | |
%2595 = torch.aten.add.int %2591, %2594 : !torch.int, !torch.int -> !torch.int | |
%int2_827 = torch.constant.int 2 | |
%2596 = torch.aten.select.int %2583, %int0_823, %int2_827 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2597 = torch.aten.item %2596 : !torch.vtensor<[1],si64> -> !torch.int | |
%2598 = torch.aten.lt.int %2597, %int0_823 : !torch.int, !torch.int -> !torch.bool | |
%2599 = torch.aten.Int.bool %2598 : !torch.bool -> !torch.int | |
%2600 = torch.aten.mul.int %2599, %int1_824 : !torch.int, !torch.int -> !torch.int | |
%2601 = torch.aten.add.int %2597, %2600 : !torch.int, !torch.int -> !torch.int | |
%2602 = torch.prim.ListConstruct %2589, %2595, %2601 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_828 = torch.constant.bool false | |
%none_829 = torch.constant.none | |
%2603 = torch.aten.tensor %2602, %none_829, %none_829, %false_828 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_830, %indices_831 = torch.aten.sort %2603, %int0_823, %false_828 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_832 = torch.constant.int 0 | |
%2604 = torch.aten.select.int %values_830, %int0_823, %int0_832 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2605 = torch.aten.item %2604 : !torch.vtensor<[1],si64> -> !torch.int | |
%2606 = torch.aten.unsqueeze %2493, %2605 : !torch.vtensor<[64],f32>, !torch.int -> !torch.vtensor | |
%int1_833 = torch.constant.int 1 | |
%2607 = torch.aten.select.int %values_830, %int0_823, %int1_833 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2608 = torch.aten.item %2607 : !torch.vtensor<[1],si64> -> !torch.int | |
%2609 = torch.aten.unsqueeze %2606, %2608 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_834 = torch.constant.int 2 | |
%2610 = torch.aten.select.int %values_830, %int0_823, %int2_834 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2611 = torch.aten.item %2610 : !torch.vtensor<[1],si64> -> !torch.int | |
%2612 = torch.aten.unsqueeze %2609, %2611 : !torch.vtensor, !torch.int -> !torch.vtensor<[64,1,1,1],f32> | |
%2613 = torch.aten.mul.Tensor %2582, %2612 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[64,1,1,1],f32> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2614 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_835 = torch.constant.int 0 | |
%int1_836 = torch.constant.int 1 | |
%int0_837 = torch.constant.int 0 | |
%2615 = torch.aten.select.int %2614, %int0_835, %int0_837 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2616 = torch.aten.item %2615 : !torch.vtensor<[1],si64> -> !torch.int | |
%2617 = torch.aten.lt.int %2616, %int0_835 : !torch.int, !torch.int -> !torch.bool | |
%2618 = torch.aten.Int.bool %2617 : !torch.bool -> !torch.int | |
%2619 = torch.aten.mul.int %2618, %int1_836 : !torch.int, !torch.int -> !torch.int | |
%2620 = torch.aten.add.int %2616, %2619 : !torch.int, !torch.int -> !torch.int | |
%int1_838 = torch.constant.int 1 | |
%2621 = torch.aten.select.int %2614, %int0_835, %int1_838 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2622 = torch.aten.item %2621 : !torch.vtensor<[1],si64> -> !torch.int | |
%2623 = torch.aten.lt.int %2622, %int0_835 : !torch.int, !torch.int -> !torch.bool | |
%2624 = torch.aten.Int.bool %2623 : !torch.bool -> !torch.int | |
%2625 = torch.aten.mul.int %2624, %int1_836 : !torch.int, !torch.int -> !torch.int | |
%2626 = torch.aten.add.int %2622, %2625 : !torch.int, !torch.int -> !torch.int | |
%int2_839 = torch.constant.int 2 | |
%2627 = torch.aten.select.int %2614, %int0_835, %int2_839 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2628 = torch.aten.item %2627 : !torch.vtensor<[1],si64> -> !torch.int | |
%2629 = torch.aten.lt.int %2628, %int0_835 : !torch.int, !torch.int -> !torch.bool | |
%2630 = torch.aten.Int.bool %2629 : !torch.bool -> !torch.int | |
%2631 = torch.aten.mul.int %2630, %int1_836 : !torch.int, !torch.int -> !torch.int | |
%2632 = torch.aten.add.int %2628, %2631 : !torch.int, !torch.int -> !torch.int | |
%2633 = torch.prim.ListConstruct %2620, %2626, %2632 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_840 = torch.constant.bool false | |
%none_841 = torch.constant.none | |
%2634 = torch.aten.tensor %2633, %none_841, %none_841, %false_840 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_842, %indices_843 = torch.aten.sort %2634, %int0_835, %false_840 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_844 = torch.constant.int 0 | |
%2635 = torch.aten.select.int %values_842, %int0_835, %int0_844 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2636 = torch.aten.item %2635 : !torch.vtensor<[1],si64> -> !torch.int | |
%2637 = torch.aten.unsqueeze %2505, %2636 : !torch.vtensor<[64],f32>, !torch.int -> !torch.vtensor | |
%int1_845 = torch.constant.int 1 | |
%2638 = torch.aten.select.int %values_842, %int0_835, %int1_845 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2639 = torch.aten.item %2638 : !torch.vtensor<[1],si64> -> !torch.int | |
%2640 = torch.aten.unsqueeze %2637, %2639 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_846 = torch.constant.int 2 | |
%2641 = torch.aten.select.int %values_842, %int0_835, %int2_846 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2642 = torch.aten.item %2641 : !torch.vtensor<[1],si64> -> !torch.int | |
%2643 = torch.aten.unsqueeze %2640, %2642 : !torch.vtensor, !torch.int -> !torch.vtensor<[64,1,1,1],f32> | |
%int1_847 = torch.constant.int 1 | |
%2644 = torch.aten.add.Tensor %2613, %2643, %int1_847 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.vtensor<[64,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2645 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2646 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_848 = torch.constant.int 12 | |
%2647 = torch.aten.item %2645 : !torch.vtensor<[],f32> -> !torch.float | |
%2648 = torch.aten.item %2646 : !torch.vtensor<[],si8> -> !torch.int | |
%2649 = torch.aten.quantize_per_tensor %2644, %2647, %2648, %int12_848 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%2650 = torch.aten.int_repr %2649 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],si8> | |
%2651 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2652 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2653 = torch.aten.item %2651 : !torch.vtensor<[],f32> -> !torch.float | |
%2654 = torch.aten.item %2652 : !torch.vtensor<[],si8> -> !torch.int | |
%2655 = torch.aten._make_per_tensor_quantized_tensor %2650, %2653, %2654 : !torch.vtensor<[?,64,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%2656 = torch.aten.dequantize.self %2655 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2657 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2658 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_849 = torch.constant.int 12 | |
%2659 = torch.aten.item %2657 : !torch.vtensor<[],f32> -> !torch.float | |
%2660 = torch.aten.item %2658 : !torch.vtensor<[],si8> -> !torch.int | |
%2661 = torch.aten.quantize_per_tensor %35, %2659, %2660, %int12_849 : !torch.vtensor<[64,64,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[64,64,3,3,3],!torch.qint8> | |
%2662 = torch.aten.int_repr %2661 : !torch.vtensor<[64,64,3,3,3],!torch.qint8> -> !torch.vtensor<[64,64,3,3,3],si8> | |
%2663 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2664 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2665 = torch.aten.item %2663 : !torch.vtensor<[],f32> -> !torch.float | |
%2666 = torch.aten.item %2664 : !torch.vtensor<[],si8> -> !torch.int | |
%2667 = torch.aten._make_per_tensor_quantized_tensor %2662, %2665, %2666 : !torch.vtensor<[64,64,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[64,64,3,3,3],!torch.qint8> | |
%2668 = torch.aten.dequantize.self %2667 : !torch.vtensor<[64,64,3,3,3],!torch.qint8> -> !torch.vtensor<[64,64,3,3,3],f32> | |
%int1_850 = torch.constant.int 1 | |
%int1_851 = torch.constant.int 1 | |
%int1_852 = torch.constant.int 1 | |
%int1_853 = torch.constant.int 1 | |
%int1_854 = torch.constant.int 1 | |
%int1_855 = torch.constant.int 1 | |
%int1_856 = torch.constant.int 1 | |
%int1_857 = torch.constant.int 1 | |
%int1_858 = torch.constant.int 1 | |
%int0_859 = torch.constant.int 0 | |
%2669 = torch.prim.ListConstruct %int1_850, %int1_851, %int1_852 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2670 = torch.prim.ListConstruct %int1_853, %int1_854, %int1_855 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2671 = torch.prim.ListConstruct %int1_856, %int1_857, %int1_858 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2672 = torch.prim.ListConstruct %int0_859, %int0_859 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_860 = torch.constant.bool false | |
%none_861 = torch.constant.none | |
%int1_862 = torch.constant.int 1 | |
%2673 = torch.aten.convolution %2656, %2668, %none_861, %2671, %2669, %2670, %false_860, %2672, %int1_862 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.vtensor<[64,64,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2674 = torch.aten.relu %2673 : !torch.vtensor<[?,64,?,?,?],f32> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2675 = torch.vtensor.literal(dense<1.000000e+00> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2676 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_863 = torch.constant.int 12 | |
%2677 = torch.aten.item %2675 : !torch.vtensor<[],f32> -> !torch.float | |
%2678 = torch.aten.item %2676 : !torch.vtensor<[],si8> -> !torch.int | |
%2679 = torch.aten.quantize_per_tensor %2674, %2677, %2678, %int12_863 : !torch.vtensor<[?,64,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%2680 = torch.aten.int_repr %2679 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],si8> | |
%2681 = torch.vtensor.literal(dense<1.000000e+00> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2682 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2683 = torch.aten.item %2681 : !torch.vtensor<[],f32> -> !torch.float | |
%2684 = torch.aten.item %2682 : !torch.vtensor<[],si8> -> !torch.int | |
%2685 = torch.aten._make_per_tensor_quantized_tensor %2680, %2683, %2684 : !torch.vtensor<[?,64,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,64,?,?,?],!torch.qint8> | |
%2686 = torch.aten.dequantize.self %2685 : !torch.vtensor<[?,64,?,?,?],!torch.qint8> -> !torch.vtensor<[?,64,?,?,?],f32> | |
%2687 = torch.vtensor.literal(dense<64> : tensor<si64>) : !torch.vtensor<[],si64> | |
%2688 = torch.vtensor.literal(dense<128> : tensor<si64>) : !torch.vtensor<[],si64> | |
%2689 = torch.vtensor.literal(dense<128> : tensor<si64>) : !torch.vtensor<[],si64> | |
%2690 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_864 = torch.constant.int 0 | |
%int0_865 = torch.constant.int 0 | |
%int0_866 = torch.constant.int 0 | |
%2691 = torch.aten.select.int %2690, %int0_864, %int0_866 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2692 = torch.aten.item %2691 : !torch.vtensor<[1],si64> -> !torch.int | |
%2693 = torch.aten.lt.int %2692, %int0_864 : !torch.int, !torch.int -> !torch.bool | |
%2694 = torch.aten.Int.bool %2693 : !torch.bool -> !torch.int | |
%2695 = torch.aten.mul.int %2694, %int0_865 : !torch.int, !torch.int -> !torch.int | |
%2696 = torch.aten.add.int %2692, %2695 : !torch.int, !torch.int -> !torch.int | |
%2697 = torch.prim.ListConstruct %2696 : (!torch.int) -> !torch.list<int> | |
%false_867 = torch.constant.bool false | |
%none_868 = torch.constant.none | |
%2698 = torch.aten.tensor %2697, %none_868, %none_868, %false_867 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_869, %indices_870 = torch.aten.sort %2698, %int0_864, %false_867 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_871 = torch.constant.int 0 | |
%2699 = torch.aten.select.int %values_869, %int0_864, %int0_871 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2700 = torch.aten.item %2699 : !torch.vtensor<[1],si64> -> !torch.int | |
%2701 = torch.aten.unsqueeze %2687, %2700 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%2702 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_872 = torch.constant.int 0 | |
%int0_873 = torch.constant.int 0 | |
%int0_874 = torch.constant.int 0 | |
%2703 = torch.aten.select.int %2702, %int0_872, %int0_874 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2704 = torch.aten.item %2703 : !torch.vtensor<[1],si64> -> !torch.int | |
%2705 = torch.aten.lt.int %2704, %int0_872 : !torch.int, !torch.int -> !torch.bool | |
%2706 = torch.aten.Int.bool %2705 : !torch.bool -> !torch.int | |
%2707 = torch.aten.mul.int %2706, %int0_873 : !torch.int, !torch.int -> !torch.int | |
%2708 = torch.aten.add.int %2704, %2707 : !torch.int, !torch.int -> !torch.int | |
%2709 = torch.prim.ListConstruct %2708 : (!torch.int) -> !torch.list<int> | |
%false_875 = torch.constant.bool false | |
%none_876 = torch.constant.none | |
%2710 = torch.aten.tensor %2709, %none_876, %none_876, %false_875 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_877, %indices_878 = torch.aten.sort %2710, %int0_872, %false_875 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_879 = torch.constant.int 0 | |
%2711 = torch.aten.select.int %values_877, %int0_872, %int0_879 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2712 = torch.aten.item %2711 : !torch.vtensor<[1],si64> -> !torch.int | |
%2713 = torch.aten.unsqueeze %2688, %2712 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%2714 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%int0_880 = torch.constant.int 0 | |
%int0_881 = torch.constant.int 0 | |
%int0_882 = torch.constant.int 0 | |
%2715 = torch.aten.select.int %2714, %int0_880, %int0_882 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2716 = torch.aten.item %2715 : !torch.vtensor<[1],si64> -> !torch.int | |
%2717 = torch.aten.lt.int %2716, %int0_880 : !torch.int, !torch.int -> !torch.bool | |
%2718 = torch.aten.Int.bool %2717 : !torch.bool -> !torch.int | |
%2719 = torch.aten.mul.int %2718, %int0_881 : !torch.int, !torch.int -> !torch.int | |
%2720 = torch.aten.add.int %2716, %2719 : !torch.int, !torch.int -> !torch.int | |
%2721 = torch.prim.ListConstruct %2720 : (!torch.int) -> !torch.list<int> | |
%false_883 = torch.constant.bool false | |
%none_884 = torch.constant.none | |
%2722 = torch.aten.tensor %2721, %none_884, %none_884, %false_883 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[1],si64> | |
%values_885, %indices_886 = torch.aten.sort %2722, %int0_880, %false_883 : !torch.vtensor<[1],si64>, !torch.int, !torch.bool -> !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64> | |
%int0_887 = torch.constant.int 0 | |
%2723 = torch.aten.select.int %values_885, %int0_880, %int0_887 : !torch.vtensor<[1],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2724 = torch.aten.item %2723 : !torch.vtensor<[1],si64> -> !torch.int | |
%2725 = torch.aten.unsqueeze %2689, %2724 : !torch.vtensor<[],si64>, !torch.int -> !torch.vtensor<[1],si64> | |
%2726 = torch.prim.ListConstruct %2701, %2713, %2725 : (!torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>) -> !torch.list<vtensor> | |
%int0_888 = torch.constant.int 0 | |
%2727 = torch.aten.cat %2726, %int0_888 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[3],si64> | |
%2728 = torch.aten._shape_as_tensor %2686 : !torch.vtensor<[?,64,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%2729 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%2730 = torch.vtensor.literal(dense<0> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%2731 = torch.vtensor.literal(dense<2> : tensor<1xsi64>) : !torch.vtensor<[1],si64> | |
%none_889 = torch.constant.none | |
%int1_890 = torch.constant.int 1 | |
%2732 = torch.prim.ListConstruct %int1_890 : (!torch.int) -> !torch.list<int> | |
%2733 = torch.aten.ones %2732, %none_889, %none_889, %none_889, %none_889 : !torch.list<int>, !torch.none, !torch.none, !torch.none, !torch.none -> !torch.vtensor<[1],si64> | |
%int0_891 = torch.constant.int 0 | |
%int0_892 = torch.constant.int 0 | |
%2734 = torch.prim.NumToTensor.Scalar %int0_892 : !torch.int -> !torch.vtensor<[1],si64> | |
%2735 = torch.aten.index_select %2730, %int0_891, %2734 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%2736 = torch.aten.item %2735 : !torch.vtensor<[1],si64> -> !torch.int | |
%2737 = torch.aten.index_select %2731, %int0_891, %2734 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%2738 = torch.aten.item %2737 : !torch.vtensor<[1],si64> -> !torch.int | |
%2739 = torch.aten.index_select %2729, %int0_891, %2734 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%2740 = torch.aten.item %2739 : !torch.vtensor<[1],si64> -> !torch.int | |
%2741 = torch.aten.index_select %2733, %int0_891, %2734 : !torch.vtensor<[1],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1],si64> | |
%2742 = torch.aten.item %2741 : !torch.vtensor<[1],si64> -> !torch.int | |
%2743 = torch.aten.slice.Tensor %2728, %2740, %2736, %2738, %2742 : !torch.vtensor<[5],si64>, !torch.int, !torch.int, !torch.int, !torch.int -> !torch.vtensor<[2],si64> | |
%int4_893 = torch.constant.int 4 | |
%none_894 = torch.constant.none | |
%false_895 = torch.constant.bool false | |
%2744 = torch.aten.to.dtype %2727, %int4_893, %false_895, %false_895, %none_894 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool, !torch.bool, !torch.none -> !torch.vtensor<[3],si64> | |
%2745 = torch.prim.ListConstruct %2743, %2744 : (!torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>) -> !torch.list<vtensor> | |
%int0_896 = torch.constant.int 0 | |
%2746 = torch.aten.cat %2745, %int0_896 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[5],si64> | |
%2747 = torch.operator "onnx.Resize"(%2686, %none, %none, %2746) {torch.onnx.coordinate_transformation_mode = "asymmetric", torch.onnx.cubic_coeff_a = -7.500000e-01 : f32, torch.onnx.mode = "nearest", torch.onnx.nearest_mode = "floor"} : (!torch.vtensor<[?,64,?,?,?],f32>, !torch.none, !torch.none, !torch.vtensor<[5],si64>) -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2748 = torch.prim.ListConstruct %435, %2747 : (!torch.vtensor<[?,32,?,?,?],f32>, !torch.vtensor<[?,?,?,?,?],f32>) -> !torch.list<vtensor> | |
%int1_897 = torch.constant.int 1 | |
%2749 = torch.aten.cat %2748, %int1_897 : !torch.list<vtensor>, !torch.int -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2750 = torch.vtensor.literal(dense<1.000000e+00> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2751 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_898 = torch.constant.int 12 | |
%2752 = torch.aten.item %2750 : !torch.vtensor<[],f32> -> !torch.float | |
%2753 = torch.aten.item %2751 : !torch.vtensor<[],si8> -> !torch.int | |
%2754 = torch.aten.quantize_per_tensor %2749, %2752, %2753, %int12_898 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,?,?,?,?],!torch.qint8> | |
%2755 = torch.aten.int_repr %2754 : !torch.vtensor<[?,?,?,?,?],!torch.qint8> -> !torch.vtensor<[?,?,?,?,?],si8> | |
%2756 = torch.vtensor.literal(dense<1.000000e+00> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2757 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2758 = torch.aten.item %2756 : !torch.vtensor<[],f32> -> !torch.float | |
%2759 = torch.aten.item %2757 : !torch.vtensor<[],si8> -> !torch.int | |
%2760 = torch.aten._make_per_tensor_quantized_tensor %2755, %2758, %2759 : !torch.vtensor<[?,?,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,?,?,?,?],!torch.qint8> | |
%2761 = torch.aten.dequantize.self %2760 : !torch.vtensor<[?,?,?,?,?],!torch.qint8> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2762 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2763 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_899 = torch.constant.int 12 | |
%2764 = torch.aten.item %2762 : !torch.vtensor<[],f32> -> !torch.float | |
%2765 = torch.aten.item %2763 : !torch.vtensor<[],si8> -> !torch.int | |
%2766 = torch.aten.quantize_per_tensor %36, %2764, %2765, %int12_899 : !torch.vtensor<[96],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[96],!torch.qint8> | |
%2767 = torch.aten.int_repr %2766 : !torch.vtensor<[96],!torch.qint8> -> !torch.vtensor<[96],si8> | |
%2768 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2769 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2770 = torch.aten.item %2768 : !torch.vtensor<[],f32> -> !torch.float | |
%2771 = torch.aten.item %2769 : !torch.vtensor<[],si8> -> !torch.int | |
%2772 = torch.aten._make_per_tensor_quantized_tensor %2767, %2770, %2771 : !torch.vtensor<[96],si8>, !torch.float, !torch.int -> !torch.vtensor<[96],!torch.qint8> | |
%2773 = torch.aten.dequantize.self %2772 : !torch.vtensor<[96],!torch.qint8> -> !torch.vtensor<[96],f32> | |
%2774 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2775 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_900 = torch.constant.int 12 | |
%2776 = torch.aten.item %2774 : !torch.vtensor<[],f32> -> !torch.float | |
%2777 = torch.aten.item %2775 : !torch.vtensor<[],si8> -> !torch.int | |
%2778 = torch.aten.quantize_per_tensor %37, %2776, %2777, %int12_900 : !torch.vtensor<[96],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[96],!torch.qint8> | |
%2779 = torch.aten.int_repr %2778 : !torch.vtensor<[96],!torch.qint8> -> !torch.vtensor<[96],si8> | |
%2780 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2781 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2782 = torch.aten.item %2780 : !torch.vtensor<[],f32> -> !torch.float | |
%2783 = torch.aten.item %2781 : !torch.vtensor<[],si8> -> !torch.int | |
%2784 = torch.aten._make_per_tensor_quantized_tensor %2779, %2782, %2783 : !torch.vtensor<[96],si8>, !torch.float, !torch.int -> !torch.vtensor<[96],!torch.qint8> | |
%2785 = torch.aten.dequantize.self %2784 : !torch.vtensor<[96],!torch.qint8> -> !torch.vtensor<[96],f32> | |
%2786 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_901 = torch.constant.int 0 | |
%int0_902 = torch.constant.int 0 | |
%2787 = torch.aten.select.int %2786, %int0_901, %int0_902 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2788 = torch.aten.item %2787 : !torch.vtensor<[1],si64> -> !torch.int | |
%2789 = torch.aten.eq.int %2788, %int0_901 : !torch.int, !torch.int -> !torch.bool | |
%2790 = torch.aten.Int.bool %2789 : !torch.bool -> !torch.int | |
%int0_903 = torch.constant.int 0 | |
%2791 = torch.aten.size.int %2761, %int0_903 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%2792 = torch.prim.NumToTensor.Scalar %2790 : !torch.int -> !torch.vtensor<[],i1> | |
%2793 = torch.prim.NumToTensor.Scalar %2791 : !torch.int -> !torch.vtensor<[],si64> | |
%2794 = torch.prim.NumToTensor.Scalar %2788 : !torch.int -> !torch.vtensor<[],si64> | |
%2795 = torch.aten.where.self %2792, %2793, %2794 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2796 = torch.aten.item %2795 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_904 = torch.constant.int 1 | |
%2797 = torch.aten.select.int %2786, %int0_901, %int1_904 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2798 = torch.aten.item %2797 : !torch.vtensor<[1],si64> -> !torch.int | |
%2799 = torch.aten.eq.int %2798, %int0_901 : !torch.int, !torch.int -> !torch.bool | |
%2800 = torch.aten.Int.bool %2799 : !torch.bool -> !torch.int | |
%int1_905 = torch.constant.int 1 | |
%2801 = torch.aten.size.int %2761, %int1_905 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%2802 = torch.prim.NumToTensor.Scalar %2800 : !torch.int -> !torch.vtensor<[],i1> | |
%2803 = torch.prim.NumToTensor.Scalar %2801 : !torch.int -> !torch.vtensor<[],si64> | |
%2804 = torch.prim.NumToTensor.Scalar %2798 : !torch.int -> !torch.vtensor<[],si64> | |
%2805 = torch.aten.where.self %2802, %2803, %2804 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2806 = torch.aten.item %2805 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_906 = torch.constant.int 2 | |
%2807 = torch.aten.select.int %2786, %int0_901, %int2_906 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2808 = torch.aten.item %2807 : !torch.vtensor<[1],si64> -> !torch.int | |
%2809 = torch.aten.eq.int %2808, %int0_901 : !torch.int, !torch.int -> !torch.bool | |
%2810 = torch.aten.Int.bool %2809 : !torch.bool -> !torch.int | |
%int2_907 = torch.constant.int 2 | |
%2811 = torch.aten.size.int %2761, %int2_907 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.int -> !torch.int | |
%2812 = torch.prim.NumToTensor.Scalar %2810 : !torch.int -> !torch.vtensor<[],i1> | |
%2813 = torch.prim.NumToTensor.Scalar %2811 : !torch.int -> !torch.vtensor<[],si64> | |
%2814 = torch.prim.NumToTensor.Scalar %2808 : !torch.int -> !torch.vtensor<[],si64> | |
%2815 = torch.aten.where.self %2812, %2813, %2814 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2816 = torch.aten.item %2815 : !torch.vtensor<[],si64> -> !torch.int | |
%2817 = torch.prim.ListConstruct %2796, %2806, %2816 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2818 = torch.aten.reshape %2761, %2817 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%2819 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%2820 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_908 = torch.constant.none | |
%true_909 = torch.constant.bool true | |
%false_910 = torch.constant.bool false | |
%float9.999990e-06_911 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_912 = torch.constant.float 0.000000e+00 | |
%2821 = torch.aten.instance_norm %2818, %2819, %2820, %none_908, %none_908, %true_909, %float0.000000e00_912, %float9.999990e-06_911, %false_910 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%2822 = torch.aten._shape_as_tensor %2761 : !torch.vtensor<[?,?,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_913 = torch.constant.int 0 | |
%int0_914 = torch.constant.int 0 | |
%2823 = torch.aten.select.int %2822, %int0_913, %int0_914 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2824 = torch.aten.item %2823 : !torch.vtensor<[1],si64> -> !torch.int | |
%2825 = torch.aten.eq.int %2824, %int0_913 : !torch.int, !torch.int -> !torch.bool | |
%2826 = torch.aten.Int.bool %2825 : !torch.bool -> !torch.int | |
%int0_915 = torch.constant.int 0 | |
%2827 = torch.aten.size.int %2821, %int0_915 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2828 = torch.prim.NumToTensor.Scalar %2826 : !torch.int -> !torch.vtensor<[],i1> | |
%2829 = torch.prim.NumToTensor.Scalar %2827 : !torch.int -> !torch.vtensor<[],si64> | |
%2830 = torch.prim.NumToTensor.Scalar %2824 : !torch.int -> !torch.vtensor<[],si64> | |
%2831 = torch.aten.where.self %2828, %2829, %2830 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2832 = torch.aten.item %2831 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_916 = torch.constant.int 1 | |
%2833 = torch.aten.select.int %2822, %int0_913, %int1_916 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2834 = torch.aten.item %2833 : !torch.vtensor<[1],si64> -> !torch.int | |
%2835 = torch.aten.eq.int %2834, %int0_913 : !torch.int, !torch.int -> !torch.bool | |
%2836 = torch.aten.Int.bool %2835 : !torch.bool -> !torch.int | |
%int1_917 = torch.constant.int 1 | |
%2837 = torch.aten.size.int %2821, %int1_917 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2838 = torch.prim.NumToTensor.Scalar %2836 : !torch.int -> !torch.vtensor<[],i1> | |
%2839 = torch.prim.NumToTensor.Scalar %2837 : !torch.int -> !torch.vtensor<[],si64> | |
%2840 = torch.prim.NumToTensor.Scalar %2834 : !torch.int -> !torch.vtensor<[],si64> | |
%2841 = torch.aten.where.self %2838, %2839, %2840 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2842 = torch.aten.item %2841 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_918 = torch.constant.int 2 | |
%2843 = torch.aten.select.int %2822, %int0_913, %int2_918 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2844 = torch.aten.item %2843 : !torch.vtensor<[1],si64> -> !torch.int | |
%2845 = torch.aten.eq.int %2844, %int0_913 : !torch.int, !torch.int -> !torch.bool | |
%2846 = torch.aten.Int.bool %2845 : !torch.bool -> !torch.int | |
%int2_919 = torch.constant.int 2 | |
%2847 = torch.aten.size.int %2821, %int2_919 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%2848 = torch.prim.NumToTensor.Scalar %2846 : !torch.int -> !torch.vtensor<[],i1> | |
%2849 = torch.prim.NumToTensor.Scalar %2847 : !torch.int -> !torch.vtensor<[],si64> | |
%2850 = torch.prim.NumToTensor.Scalar %2844 : !torch.int -> !torch.vtensor<[],si64> | |
%2851 = torch.aten.where.self %2848, %2849, %2850 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%2852 = torch.aten.item %2851 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_920 = torch.constant.int 3 | |
%2853 = torch.aten.select.int %2822, %int0_913, %int3_920 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2854 = torch.aten.item %2853 : !torch.vtensor<[1],si64> -> !torch.int | |
%2855 = torch.aten.eq.int %2854, %int0_913 : !torch.int, !torch.int -> !torch.bool | |
%2856 = torch.aten.Int.bool %2855 : !torch.bool -> !torch.int | |
%int4_921 = torch.constant.int 4 | |
%2857 = torch.aten.select.int %2822, %int0_913, %int4_921 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2858 = torch.aten.item %2857 : !torch.vtensor<[1],si64> -> !torch.int | |
%2859 = torch.aten.eq.int %2858, %int0_913 : !torch.int, !torch.int -> !torch.bool | |
%2860 = torch.aten.Int.bool %2859 : !torch.bool -> !torch.int | |
%2861 = torch.prim.ListConstruct %2832, %2842, %2852, %2854, %2858 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2862 = torch.aten.reshape %2821, %2861 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%2863 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_922 = torch.constant.int 0 | |
%int1_923 = torch.constant.int 1 | |
%int0_924 = torch.constant.int 0 | |
%2864 = torch.aten.select.int %2863, %int0_922, %int0_924 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2865 = torch.aten.item %2864 : !torch.vtensor<[1],si64> -> !torch.int | |
%2866 = torch.aten.lt.int %2865, %int0_922 : !torch.int, !torch.int -> !torch.bool | |
%2867 = torch.aten.Int.bool %2866 : !torch.bool -> !torch.int | |
%2868 = torch.aten.mul.int %2867, %int1_923 : !torch.int, !torch.int -> !torch.int | |
%2869 = torch.aten.add.int %2865, %2868 : !torch.int, !torch.int -> !torch.int | |
%int1_925 = torch.constant.int 1 | |
%2870 = torch.aten.select.int %2863, %int0_922, %int1_925 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2871 = torch.aten.item %2870 : !torch.vtensor<[1],si64> -> !torch.int | |
%2872 = torch.aten.lt.int %2871, %int0_922 : !torch.int, !torch.int -> !torch.bool | |
%2873 = torch.aten.Int.bool %2872 : !torch.bool -> !torch.int | |
%2874 = torch.aten.mul.int %2873, %int1_923 : !torch.int, !torch.int -> !torch.int | |
%2875 = torch.aten.add.int %2871, %2874 : !torch.int, !torch.int -> !torch.int | |
%int2_926 = torch.constant.int 2 | |
%2876 = torch.aten.select.int %2863, %int0_922, %int2_926 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2877 = torch.aten.item %2876 : !torch.vtensor<[1],si64> -> !torch.int | |
%2878 = torch.aten.lt.int %2877, %int0_922 : !torch.int, !torch.int -> !torch.bool | |
%2879 = torch.aten.Int.bool %2878 : !torch.bool -> !torch.int | |
%2880 = torch.aten.mul.int %2879, %int1_923 : !torch.int, !torch.int -> !torch.int | |
%2881 = torch.aten.add.int %2877, %2880 : !torch.int, !torch.int -> !torch.int | |
%2882 = torch.prim.ListConstruct %2869, %2875, %2881 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_927 = torch.constant.bool false | |
%none_928 = torch.constant.none | |
%2883 = torch.aten.tensor %2882, %none_928, %none_928, %false_927 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_929, %indices_930 = torch.aten.sort %2883, %int0_922, %false_927 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_931 = torch.constant.int 0 | |
%2884 = torch.aten.select.int %values_929, %int0_922, %int0_931 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2885 = torch.aten.item %2884 : !torch.vtensor<[1],si64> -> !torch.int | |
%2886 = torch.aten.unsqueeze %2773, %2885 : !torch.vtensor<[96],f32>, !torch.int -> !torch.vtensor | |
%int1_932 = torch.constant.int 1 | |
%2887 = torch.aten.select.int %values_929, %int0_922, %int1_932 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2888 = torch.aten.item %2887 : !torch.vtensor<[1],si64> -> !torch.int | |
%2889 = torch.aten.unsqueeze %2886, %2888 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_933 = torch.constant.int 2 | |
%2890 = torch.aten.select.int %values_929, %int0_922, %int2_933 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2891 = torch.aten.item %2890 : !torch.vtensor<[1],si64> -> !torch.int | |
%2892 = torch.aten.unsqueeze %2889, %2891 : !torch.vtensor, !torch.int -> !torch.vtensor<[96,1,1,1],f32> | |
%2893 = torch.aten.mul.Tensor %2862, %2892 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[96,1,1,1],f32> -> !torch.vtensor<[?,96,?,?,?],f32> | |
%2894 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_934 = torch.constant.int 0 | |
%int1_935 = torch.constant.int 1 | |
%int0_936 = torch.constant.int 0 | |
%2895 = torch.aten.select.int %2894, %int0_934, %int0_936 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2896 = torch.aten.item %2895 : !torch.vtensor<[1],si64> -> !torch.int | |
%2897 = torch.aten.lt.int %2896, %int0_934 : !torch.int, !torch.int -> !torch.bool | |
%2898 = torch.aten.Int.bool %2897 : !torch.bool -> !torch.int | |
%2899 = torch.aten.mul.int %2898, %int1_935 : !torch.int, !torch.int -> !torch.int | |
%2900 = torch.aten.add.int %2896, %2899 : !torch.int, !torch.int -> !torch.int | |
%int1_937 = torch.constant.int 1 | |
%2901 = torch.aten.select.int %2894, %int0_934, %int1_937 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2902 = torch.aten.item %2901 : !torch.vtensor<[1],si64> -> !torch.int | |
%2903 = torch.aten.lt.int %2902, %int0_934 : !torch.int, !torch.int -> !torch.bool | |
%2904 = torch.aten.Int.bool %2903 : !torch.bool -> !torch.int | |
%2905 = torch.aten.mul.int %2904, %int1_935 : !torch.int, !torch.int -> !torch.int | |
%2906 = torch.aten.add.int %2902, %2905 : !torch.int, !torch.int -> !torch.int | |
%int2_938 = torch.constant.int 2 | |
%2907 = torch.aten.select.int %2894, %int0_934, %int2_938 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2908 = torch.aten.item %2907 : !torch.vtensor<[1],si64> -> !torch.int | |
%2909 = torch.aten.lt.int %2908, %int0_934 : !torch.int, !torch.int -> !torch.bool | |
%2910 = torch.aten.Int.bool %2909 : !torch.bool -> !torch.int | |
%2911 = torch.aten.mul.int %2910, %int1_935 : !torch.int, !torch.int -> !torch.int | |
%2912 = torch.aten.add.int %2908, %2911 : !torch.int, !torch.int -> !torch.int | |
%2913 = torch.prim.ListConstruct %2900, %2906, %2912 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_939 = torch.constant.bool false | |
%none_940 = torch.constant.none | |
%2914 = torch.aten.tensor %2913, %none_940, %none_940, %false_939 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_941, %indices_942 = torch.aten.sort %2914, %int0_934, %false_939 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_943 = torch.constant.int 0 | |
%2915 = torch.aten.select.int %values_941, %int0_934, %int0_943 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2916 = torch.aten.item %2915 : !torch.vtensor<[1],si64> -> !torch.int | |
%2917 = torch.aten.unsqueeze %2785, %2916 : !torch.vtensor<[96],f32>, !torch.int -> !torch.vtensor | |
%int1_944 = torch.constant.int 1 | |
%2918 = torch.aten.select.int %values_941, %int0_934, %int1_944 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2919 = torch.aten.item %2918 : !torch.vtensor<[1],si64> -> !torch.int | |
%2920 = torch.aten.unsqueeze %2917, %2919 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_945 = torch.constant.int 2 | |
%2921 = torch.aten.select.int %values_941, %int0_934, %int2_945 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2922 = torch.aten.item %2921 : !torch.vtensor<[1],si64> -> !torch.int | |
%2923 = torch.aten.unsqueeze %2920, %2922 : !torch.vtensor, !torch.int -> !torch.vtensor<[96,1,1,1],f32> | |
%int1_946 = torch.constant.int 1 | |
%2924 = torch.aten.add.Tensor %2893, %2923, %int1_946 : !torch.vtensor<[?,96,?,?,?],f32>, !torch.vtensor<[96,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,96,?,?,?],f32> | |
%2925 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2926 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_947 = torch.constant.int 12 | |
%2927 = torch.aten.item %2925 : !torch.vtensor<[],f32> -> !torch.float | |
%2928 = torch.aten.item %2926 : !torch.vtensor<[],si8> -> !torch.int | |
%2929 = torch.aten.quantize_per_tensor %2924, %2927, %2928, %int12_947 : !torch.vtensor<[?,96,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,96,?,?,?],!torch.qint8> | |
%2930 = torch.aten.int_repr %2929 : !torch.vtensor<[?,96,?,?,?],!torch.qint8> -> !torch.vtensor<[?,96,?,?,?],si8> | |
%2931 = torch.vtensor.literal(dense<2.500000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2932 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2933 = torch.aten.item %2931 : !torch.vtensor<[],f32> -> !torch.float | |
%2934 = torch.aten.item %2932 : !torch.vtensor<[],si8> -> !torch.int | |
%2935 = torch.aten._make_per_tensor_quantized_tensor %2930, %2933, %2934 : !torch.vtensor<[?,96,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,96,?,?,?],!torch.qint8> | |
%2936 = torch.aten.dequantize.self %2935 : !torch.vtensor<[?,96,?,?,?],!torch.qint8> -> !torch.vtensor<[?,96,?,?,?],f32> | |
%2937 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2938 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_948 = torch.constant.int 12 | |
%2939 = torch.aten.item %2937 : !torch.vtensor<[],f32> -> !torch.float | |
%2940 = torch.aten.item %2938 : !torch.vtensor<[],si8> -> !torch.int | |
%2941 = torch.aten.quantize_per_tensor %38, %2939, %2940, %int12_948 : !torch.vtensor<[32,96,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32,96,3,3,3],!torch.qint8> | |
%2942 = torch.aten.int_repr %2941 : !torch.vtensor<[32,96,3,3,3],!torch.qint8> -> !torch.vtensor<[32,96,3,3,3],si8> | |
%2943 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2944 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2945 = torch.aten.item %2943 : !torch.vtensor<[],f32> -> !torch.float | |
%2946 = torch.aten.item %2944 : !torch.vtensor<[],si8> -> !torch.int | |
%2947 = torch.aten._make_per_tensor_quantized_tensor %2942, %2945, %2946 : !torch.vtensor<[32,96,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[32,96,3,3,3],!torch.qint8> | |
%2948 = torch.aten.dequantize.self %2947 : !torch.vtensor<[32,96,3,3,3],!torch.qint8> -> !torch.vtensor<[32,96,3,3,3],f32> | |
%int1_949 = torch.constant.int 1 | |
%int1_950 = torch.constant.int 1 | |
%int1_951 = torch.constant.int 1 | |
%int1_952 = torch.constant.int 1 | |
%int1_953 = torch.constant.int 1 | |
%int1_954 = torch.constant.int 1 | |
%int1_955 = torch.constant.int 1 | |
%int1_956 = torch.constant.int 1 | |
%int1_957 = torch.constant.int 1 | |
%int0_958 = torch.constant.int 0 | |
%2949 = torch.prim.ListConstruct %int1_949, %int1_950, %int1_951 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2950 = torch.prim.ListConstruct %int1_952, %int1_953, %int1_954 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2951 = torch.prim.ListConstruct %int1_955, %int1_956, %int1_957 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%2952 = torch.prim.ListConstruct %int0_958, %int0_958 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_959 = torch.constant.bool false | |
%none_960 = torch.constant.none | |
%int1_961 = torch.constant.int 1 | |
%2953 = torch.aten.convolution %2936, %2948, %none_960, %2951, %2949, %2950, %false_959, %2952, %int1_961 : !torch.vtensor<[?,96,?,?,?],f32>, !torch.vtensor<[32,96,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,32,?,?,?],f32> | |
%2954 = torch.aten.relu %2953 : !torch.vtensor<[?,32,?,?,?],f32> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%2955 = torch.vtensor.literal(dense<1.000000e+00> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2956 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_962 = torch.constant.int 12 | |
%2957 = torch.aten.item %2955 : !torch.vtensor<[],f32> -> !torch.float | |
%2958 = torch.aten.item %2956 : !torch.vtensor<[],si8> -> !torch.int | |
%2959 = torch.aten.quantize_per_tensor %2954, %2957, %2958, %int12_962 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%2960 = torch.aten.int_repr %2959 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],si8> | |
%2961 = torch.vtensor.literal(dense<1.000000e+00> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2962 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2963 = torch.aten.item %2961 : !torch.vtensor<[],f32> -> !torch.float | |
%2964 = torch.aten.item %2962 : !torch.vtensor<[],si8> -> !torch.int | |
%2965 = torch.aten._make_per_tensor_quantized_tensor %2960, %2963, %2964 : !torch.vtensor<[?,32,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%2966 = torch.aten.dequantize.self %2965 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%2967 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2968 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_963 = torch.constant.int 12 | |
%2969 = torch.aten.item %2967 : !torch.vtensor<[],f32> -> !torch.float | |
%2970 = torch.aten.item %2968 : !torch.vtensor<[],si8> -> !torch.int | |
%2971 = torch.aten.quantize_per_tensor %39, %2969, %2970, %int12_963 : !torch.vtensor<[32],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%2972 = torch.aten.int_repr %2971 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],si8> | |
%2973 = torch.vtensor.literal(dense<1.562500e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2974 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2975 = torch.aten.item %2973 : !torch.vtensor<[],f32> -> !torch.float | |
%2976 = torch.aten.item %2974 : !torch.vtensor<[],si8> -> !torch.int | |
%2977 = torch.aten._make_per_tensor_quantized_tensor %2972, %2975, %2976 : !torch.vtensor<[32],si8>, !torch.float, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%2978 = torch.aten.dequantize.self %2977 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],f32> | |
%2979 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2980 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_964 = torch.constant.int 12 | |
%2981 = torch.aten.item %2979 : !torch.vtensor<[],f32> -> !torch.float | |
%2982 = torch.aten.item %2980 : !torch.vtensor<[],si8> -> !torch.int | |
%2983 = torch.aten.quantize_per_tensor %40, %2981, %2982, %int12_964 : !torch.vtensor<[32],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%2984 = torch.aten.int_repr %2983 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],si8> | |
%2985 = torch.vtensor.literal(dense<9.765625E-4> : tensor<f32>) : !torch.vtensor<[],f32> | |
%2986 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%2987 = torch.aten.item %2985 : !torch.vtensor<[],f32> -> !torch.float | |
%2988 = torch.aten.item %2986 : !torch.vtensor<[],si8> -> !torch.int | |
%2989 = torch.aten._make_per_tensor_quantized_tensor %2984, %2987, %2988 : !torch.vtensor<[32],si8>, !torch.float, !torch.int -> !torch.vtensor<[32],!torch.qint8> | |
%2990 = torch.aten.dequantize.self %2989 : !torch.vtensor<[32],!torch.qint8> -> !torch.vtensor<[32],f32> | |
%2991 = torch.vtensor.literal(dense<[0, 8, -1]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_965 = torch.constant.int 0 | |
%int0_966 = torch.constant.int 0 | |
%2992 = torch.aten.select.int %2991, %int0_965, %int0_966 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%2993 = torch.aten.item %2992 : !torch.vtensor<[1],si64> -> !torch.int | |
%2994 = torch.aten.eq.int %2993, %int0_965 : !torch.int, !torch.int -> !torch.bool | |
%2995 = torch.aten.Int.bool %2994 : !torch.bool -> !torch.int | |
%int0_967 = torch.constant.int 0 | |
%2996 = torch.aten.size.int %2966, %int0_967 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%2997 = torch.prim.NumToTensor.Scalar %2995 : !torch.int -> !torch.vtensor<[],i1> | |
%2998 = torch.prim.NumToTensor.Scalar %2996 : !torch.int -> !torch.vtensor<[],si64> | |
%2999 = torch.prim.NumToTensor.Scalar %2993 : !torch.int -> !torch.vtensor<[],si64> | |
%3000 = torch.aten.where.self %2997, %2998, %2999 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%3001 = torch.aten.item %3000 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_968 = torch.constant.int 1 | |
%3002 = torch.aten.select.int %2991, %int0_965, %int1_968 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3003 = torch.aten.item %3002 : !torch.vtensor<[1],si64> -> !torch.int | |
%3004 = torch.aten.eq.int %3003, %int0_965 : !torch.int, !torch.int -> !torch.bool | |
%3005 = torch.aten.Int.bool %3004 : !torch.bool -> !torch.int | |
%int1_969 = torch.constant.int 1 | |
%3006 = torch.aten.size.int %2966, %int1_969 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%3007 = torch.prim.NumToTensor.Scalar %3005 : !torch.int -> !torch.vtensor<[],i1> | |
%3008 = torch.prim.NumToTensor.Scalar %3006 : !torch.int -> !torch.vtensor<[],si64> | |
%3009 = torch.prim.NumToTensor.Scalar %3003 : !torch.int -> !torch.vtensor<[],si64> | |
%3010 = torch.aten.where.self %3007, %3008, %3009 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%3011 = torch.aten.item %3010 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_970 = torch.constant.int 2 | |
%3012 = torch.aten.select.int %2991, %int0_965, %int2_970 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3013 = torch.aten.item %3012 : !torch.vtensor<[1],si64> -> !torch.int | |
%3014 = torch.aten.eq.int %3013, %int0_965 : !torch.int, !torch.int -> !torch.bool | |
%3015 = torch.aten.Int.bool %3014 : !torch.bool -> !torch.int | |
%int2_971 = torch.constant.int 2 | |
%3016 = torch.aten.size.int %2966, %int2_971 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.int -> !torch.int | |
%3017 = torch.prim.NumToTensor.Scalar %3015 : !torch.int -> !torch.vtensor<[],i1> | |
%3018 = torch.prim.NumToTensor.Scalar %3016 : !torch.int -> !torch.vtensor<[],si64> | |
%3019 = torch.prim.NumToTensor.Scalar %3013 : !torch.int -> !torch.vtensor<[],si64> | |
%3020 = torch.aten.where.self %3017, %3018, %3019 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%3021 = torch.aten.item %3020 : !torch.vtensor<[],si64> -> !torch.int | |
%3022 = torch.prim.ListConstruct %3001, %3011, %3021 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%3023 = torch.aten.reshape %2966, %3022 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.list<int> -> !torch.vtensor<[?,8,?],f32> | |
%3024 = torch.vtensor.literal(dense<1.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%3025 = torch.vtensor.literal(dense<0.000000e+00> : tensor<8xf32>) : !torch.vtensor<[8],f32> | |
%none_972 = torch.constant.none | |
%true_973 = torch.constant.bool true | |
%false_974 = torch.constant.bool false | |
%float9.999990e-06_975 = torch.constant.float 9.9999997473787516E-6 | |
%float0.000000e00_976 = torch.constant.float 0.000000e+00 | |
%3026 = torch.aten.instance_norm %3023, %3024, %3025, %none_972, %none_972, %true_973, %float0.000000e00_976, %float9.999990e-06_975, %false_974 : !torch.vtensor<[?,8,?],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[8],f32>, !torch.none, !torch.none, !torch.bool, !torch.float, !torch.float, !torch.bool -> !torch.vtensor<[?,8,?],f32> | |
%3027 = torch.aten._shape_as_tensor %2966 : !torch.vtensor<[?,32,?,?,?],f32> -> !torch.vtensor<[5],si64> | |
%int0_977 = torch.constant.int 0 | |
%int0_978 = torch.constant.int 0 | |
%3028 = torch.aten.select.int %3027, %int0_977, %int0_978 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3029 = torch.aten.item %3028 : !torch.vtensor<[1],si64> -> !torch.int | |
%3030 = torch.aten.eq.int %3029, %int0_977 : !torch.int, !torch.int -> !torch.bool | |
%3031 = torch.aten.Int.bool %3030 : !torch.bool -> !torch.int | |
%int0_979 = torch.constant.int 0 | |
%3032 = torch.aten.size.int %3026, %int0_979 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%3033 = torch.prim.NumToTensor.Scalar %3031 : !torch.int -> !torch.vtensor<[],i1> | |
%3034 = torch.prim.NumToTensor.Scalar %3032 : !torch.int -> !torch.vtensor<[],si64> | |
%3035 = torch.prim.NumToTensor.Scalar %3029 : !torch.int -> !torch.vtensor<[],si64> | |
%3036 = torch.aten.where.self %3033, %3034, %3035 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%3037 = torch.aten.item %3036 : !torch.vtensor<[],si64> -> !torch.int | |
%int1_980 = torch.constant.int 1 | |
%3038 = torch.aten.select.int %3027, %int0_977, %int1_980 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3039 = torch.aten.item %3038 : !torch.vtensor<[1],si64> -> !torch.int | |
%3040 = torch.aten.eq.int %3039, %int0_977 : !torch.int, !torch.int -> !torch.bool | |
%3041 = torch.aten.Int.bool %3040 : !torch.bool -> !torch.int | |
%int1_981 = torch.constant.int 1 | |
%3042 = torch.aten.size.int %3026, %int1_981 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%3043 = torch.prim.NumToTensor.Scalar %3041 : !torch.int -> !torch.vtensor<[],i1> | |
%3044 = torch.prim.NumToTensor.Scalar %3042 : !torch.int -> !torch.vtensor<[],si64> | |
%3045 = torch.prim.NumToTensor.Scalar %3039 : !torch.int -> !torch.vtensor<[],si64> | |
%3046 = torch.aten.where.self %3043, %3044, %3045 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%3047 = torch.aten.item %3046 : !torch.vtensor<[],si64> -> !torch.int | |
%int2_982 = torch.constant.int 2 | |
%3048 = torch.aten.select.int %3027, %int0_977, %int2_982 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3049 = torch.aten.item %3048 : !torch.vtensor<[1],si64> -> !torch.int | |
%3050 = torch.aten.eq.int %3049, %int0_977 : !torch.int, !torch.int -> !torch.bool | |
%3051 = torch.aten.Int.bool %3050 : !torch.bool -> !torch.int | |
%int2_983 = torch.constant.int 2 | |
%3052 = torch.aten.size.int %3026, %int2_983 : !torch.vtensor<[?,8,?],f32>, !torch.int -> !torch.int | |
%3053 = torch.prim.NumToTensor.Scalar %3051 : !torch.int -> !torch.vtensor<[],i1> | |
%3054 = torch.prim.NumToTensor.Scalar %3052 : !torch.int -> !torch.vtensor<[],si64> | |
%3055 = torch.prim.NumToTensor.Scalar %3049 : !torch.int -> !torch.vtensor<[],si64> | |
%3056 = torch.aten.where.self %3053, %3054, %3055 : !torch.vtensor<[],i1>, !torch.vtensor<[],si64>, !torch.vtensor<[],si64> -> !torch.vtensor<[],si64> | |
%3057 = torch.aten.item %3056 : !torch.vtensor<[],si64> -> !torch.int | |
%int3_984 = torch.constant.int 3 | |
%3058 = torch.aten.select.int %3027, %int0_977, %int3_984 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3059 = torch.aten.item %3058 : !torch.vtensor<[1],si64> -> !torch.int | |
%3060 = torch.aten.eq.int %3059, %int0_977 : !torch.int, !torch.int -> !torch.bool | |
%3061 = torch.aten.Int.bool %3060 : !torch.bool -> !torch.int | |
%int4_985 = torch.constant.int 4 | |
%3062 = torch.aten.select.int %3027, %int0_977, %int4_985 : !torch.vtensor<[5],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3063 = torch.aten.item %3062 : !torch.vtensor<[1],si64> -> !torch.int | |
%3064 = torch.aten.eq.int %3063, %int0_977 : !torch.int, !torch.int -> !torch.bool | |
%3065 = torch.aten.Int.bool %3064 : !torch.bool -> !torch.int | |
%3066 = torch.prim.ListConstruct %3037, %3047, %3057, %3059, %3063 : (!torch.int, !torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%3067 = torch.aten.reshape %3026, %3066 : !torch.vtensor<[?,8,?],f32>, !torch.list<int> -> !torch.vtensor<[?,?,?,?,?],f32> | |
%3068 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_986 = torch.constant.int 0 | |
%int1_987 = torch.constant.int 1 | |
%int0_988 = torch.constant.int 0 | |
%3069 = torch.aten.select.int %3068, %int0_986, %int0_988 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3070 = torch.aten.item %3069 : !torch.vtensor<[1],si64> -> !torch.int | |
%3071 = torch.aten.lt.int %3070, %int0_986 : !torch.int, !torch.int -> !torch.bool | |
%3072 = torch.aten.Int.bool %3071 : !torch.bool -> !torch.int | |
%3073 = torch.aten.mul.int %3072, %int1_987 : !torch.int, !torch.int -> !torch.int | |
%3074 = torch.aten.add.int %3070, %3073 : !torch.int, !torch.int -> !torch.int | |
%int1_989 = torch.constant.int 1 | |
%3075 = torch.aten.select.int %3068, %int0_986, %int1_989 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3076 = torch.aten.item %3075 : !torch.vtensor<[1],si64> -> !torch.int | |
%3077 = torch.aten.lt.int %3076, %int0_986 : !torch.int, !torch.int -> !torch.bool | |
%3078 = torch.aten.Int.bool %3077 : !torch.bool -> !torch.int | |
%3079 = torch.aten.mul.int %3078, %int1_987 : !torch.int, !torch.int -> !torch.int | |
%3080 = torch.aten.add.int %3076, %3079 : !torch.int, !torch.int -> !torch.int | |
%int2_990 = torch.constant.int 2 | |
%3081 = torch.aten.select.int %3068, %int0_986, %int2_990 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3082 = torch.aten.item %3081 : !torch.vtensor<[1],si64> -> !torch.int | |
%3083 = torch.aten.lt.int %3082, %int0_986 : !torch.int, !torch.int -> !torch.bool | |
%3084 = torch.aten.Int.bool %3083 : !torch.bool -> !torch.int | |
%3085 = torch.aten.mul.int %3084, %int1_987 : !torch.int, !torch.int -> !torch.int | |
%3086 = torch.aten.add.int %3082, %3085 : !torch.int, !torch.int -> !torch.int | |
%3087 = torch.prim.ListConstruct %3074, %3080, %3086 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_991 = torch.constant.bool false | |
%none_992 = torch.constant.none | |
%3088 = torch.aten.tensor %3087, %none_992, %none_992, %false_991 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_993, %indices_994 = torch.aten.sort %3088, %int0_986, %false_991 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_995 = torch.constant.int 0 | |
%3089 = torch.aten.select.int %values_993, %int0_986, %int0_995 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3090 = torch.aten.item %3089 : !torch.vtensor<[1],si64> -> !torch.int | |
%3091 = torch.aten.unsqueeze %2978, %3090 : !torch.vtensor<[32],f32>, !torch.int -> !torch.vtensor | |
%int1_996 = torch.constant.int 1 | |
%3092 = torch.aten.select.int %values_993, %int0_986, %int1_996 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3093 = torch.aten.item %3092 : !torch.vtensor<[1],si64> -> !torch.int | |
%3094 = torch.aten.unsqueeze %3091, %3093 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_997 = torch.constant.int 2 | |
%3095 = torch.aten.select.int %values_993, %int0_986, %int2_997 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3096 = torch.aten.item %3095 : !torch.vtensor<[1],si64> -> !torch.int | |
%3097 = torch.aten.unsqueeze %3094, %3096 : !torch.vtensor, !torch.int -> !torch.vtensor<[32,1,1,1],f32> | |
%3098 = torch.aten.mul.Tensor %3067, %3097 : !torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[32,1,1,1],f32> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%3099 = torch.vtensor.literal(dense<[1, 2, 3]> : tensor<3xsi64>) : !torch.vtensor<[3],si64> | |
%int0_998 = torch.constant.int 0 | |
%int1_999 = torch.constant.int 1 | |
%int0_1000 = torch.constant.int 0 | |
%3100 = torch.aten.select.int %3099, %int0_998, %int0_1000 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3101 = torch.aten.item %3100 : !torch.vtensor<[1],si64> -> !torch.int | |
%3102 = torch.aten.lt.int %3101, %int0_998 : !torch.int, !torch.int -> !torch.bool | |
%3103 = torch.aten.Int.bool %3102 : !torch.bool -> !torch.int | |
%3104 = torch.aten.mul.int %3103, %int1_999 : !torch.int, !torch.int -> !torch.int | |
%3105 = torch.aten.add.int %3101, %3104 : !torch.int, !torch.int -> !torch.int | |
%int1_1001 = torch.constant.int 1 | |
%3106 = torch.aten.select.int %3099, %int0_998, %int1_1001 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3107 = torch.aten.item %3106 : !torch.vtensor<[1],si64> -> !torch.int | |
%3108 = torch.aten.lt.int %3107, %int0_998 : !torch.int, !torch.int -> !torch.bool | |
%3109 = torch.aten.Int.bool %3108 : !torch.bool -> !torch.int | |
%3110 = torch.aten.mul.int %3109, %int1_999 : !torch.int, !torch.int -> !torch.int | |
%3111 = torch.aten.add.int %3107, %3110 : !torch.int, !torch.int -> !torch.int | |
%int2_1002 = torch.constant.int 2 | |
%3112 = torch.aten.select.int %3099, %int0_998, %int2_1002 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3113 = torch.aten.item %3112 : !torch.vtensor<[1],si64> -> !torch.int | |
%3114 = torch.aten.lt.int %3113, %int0_998 : !torch.int, !torch.int -> !torch.bool | |
%3115 = torch.aten.Int.bool %3114 : !torch.bool -> !torch.int | |
%3116 = torch.aten.mul.int %3115, %int1_999 : !torch.int, !torch.int -> !torch.int | |
%3117 = torch.aten.add.int %3113, %3116 : !torch.int, !torch.int -> !torch.int | |
%3118 = torch.prim.ListConstruct %3105, %3111, %3117 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%false_1003 = torch.constant.bool false | |
%none_1004 = torch.constant.none | |
%3119 = torch.aten.tensor %3118, %none_1004, %none_1004, %false_1003 : !torch.list<int>, !torch.none, !torch.none, !torch.bool -> !torch.vtensor<[3],si64> | |
%values_1005, %indices_1006 = torch.aten.sort %3119, %int0_998, %false_1003 : !torch.vtensor<[3],si64>, !torch.int, !torch.bool -> !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64> | |
%int0_1007 = torch.constant.int 0 | |
%3120 = torch.aten.select.int %values_1005, %int0_998, %int0_1007 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3121 = torch.aten.item %3120 : !torch.vtensor<[1],si64> -> !torch.int | |
%3122 = torch.aten.unsqueeze %2990, %3121 : !torch.vtensor<[32],f32>, !torch.int -> !torch.vtensor | |
%int1_1008 = torch.constant.int 1 | |
%3123 = torch.aten.select.int %values_1005, %int0_998, %int1_1008 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3124 = torch.aten.item %3123 : !torch.vtensor<[1],si64> -> !torch.int | |
%3125 = torch.aten.unsqueeze %3122, %3124 : !torch.vtensor, !torch.int -> !torch.vtensor | |
%int2_1009 = torch.constant.int 2 | |
%3126 = torch.aten.select.int %values_1005, %int0_998, %int2_1009 : !torch.vtensor<[3],si64>, !torch.int, !torch.int -> !torch.vtensor<[1],si64> | |
%3127 = torch.aten.item %3126 : !torch.vtensor<[1],si64> -> !torch.int | |
%3128 = torch.aten.unsqueeze %3125, %3127 : !torch.vtensor, !torch.int -> !torch.vtensor<[32,1,1,1],f32> | |
%int1_1010 = torch.constant.int 1 | |
%3129 = torch.aten.add.Tensor %3098, %3128, %int1_1010 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.vtensor<[32,1,1,1],f32>, !torch.int -> !torch.vtensor<[?,32,?,?,?],f32> | |
%3130 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3131 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_1011 = torch.constant.int 12 | |
%3132 = torch.aten.item %3130 : !torch.vtensor<[],f32> -> !torch.float | |
%3133 = torch.aten.item %3131 : !torch.vtensor<[],si8> -> !torch.int | |
%3134 = torch.aten.quantize_per_tensor %3129, %3132, %3133, %int12_1011 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%3135 = torch.aten.int_repr %3134 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],si8> | |
%3136 = torch.vtensor.literal(dense<6.250000e-02> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3137 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%3138 = torch.aten.item %3136 : !torch.vtensor<[],f32> -> !torch.float | |
%3139 = torch.aten.item %3137 : !torch.vtensor<[],si8> -> !torch.int | |
%3140 = torch.aten._make_per_tensor_quantized_tensor %3135, %3138, %3139 : !torch.vtensor<[?,32,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%3141 = torch.aten.dequantize.self %3140 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%3142 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3143 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_1012 = torch.constant.int 12 | |
%3144 = torch.aten.item %3142 : !torch.vtensor<[],f32> -> !torch.float | |
%3145 = torch.aten.item %3143 : !torch.vtensor<[],si8> -> !torch.int | |
%3146 = torch.aten.quantize_per_tensor %41, %3144, %3145, %int12_1012 : !torch.vtensor<[32,32,3,3,3],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[32,32,3,3,3],!torch.qint8> | |
%3147 = torch.aten.int_repr %3146 : !torch.vtensor<[32,32,3,3,3],!torch.qint8> -> !torch.vtensor<[32,32,3,3,3],si8> | |
%3148 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3149 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%3150 = torch.aten.item %3148 : !torch.vtensor<[],f32> -> !torch.float | |
%3151 = torch.aten.item %3149 : !torch.vtensor<[],si8> -> !torch.int | |
%3152 = torch.aten._make_per_tensor_quantized_tensor %3147, %3150, %3151 : !torch.vtensor<[32,32,3,3,3],si8>, !torch.float, !torch.int -> !torch.vtensor<[32,32,3,3,3],!torch.qint8> | |
%3153 = torch.aten.dequantize.self %3152 : !torch.vtensor<[32,32,3,3,3],!torch.qint8> -> !torch.vtensor<[32,32,3,3,3],f32> | |
%int1_1013 = torch.constant.int 1 | |
%int1_1014 = torch.constant.int 1 | |
%int1_1015 = torch.constant.int 1 | |
%int1_1016 = torch.constant.int 1 | |
%int1_1017 = torch.constant.int 1 | |
%int1_1018 = torch.constant.int 1 | |
%int1_1019 = torch.constant.int 1 | |
%int1_1020 = torch.constant.int 1 | |
%int1_1021 = torch.constant.int 1 | |
%int0_1022 = torch.constant.int 0 | |
%3154 = torch.prim.ListConstruct %int1_1013, %int1_1014, %int1_1015 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%3155 = torch.prim.ListConstruct %int1_1016, %int1_1017, %int1_1018 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%3156 = torch.prim.ListConstruct %int1_1019, %int1_1020, %int1_1021 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%3157 = torch.prim.ListConstruct %int0_1022, %int0_1022 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_1023 = torch.constant.bool false | |
%none_1024 = torch.constant.none | |
%int1_1025 = torch.constant.int 1 | |
%3158 = torch.aten.convolution %3141, %3153, %none_1024, %3156, %3154, %3155, %false_1023, %3157, %int1_1025 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.vtensor<[32,32,3,3,3],f32>, !torch.none, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,32,?,?,?],f32> | |
%3159 = torch.aten.relu %3158 : !torch.vtensor<[?,32,?,?,?],f32> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%3160 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3161 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_1026 = torch.constant.int 12 | |
%3162 = torch.aten.item %3160 : !torch.vtensor<[],f32> -> !torch.float | |
%3163 = torch.aten.item %3161 : !torch.vtensor<[],si8> -> !torch.int | |
%3164 = torch.aten.quantize_per_tensor %3159, %3162, %3163, %int12_1026 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%3165 = torch.aten.int_repr %3164 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],si8> | |
%3166 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3167 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%3168 = torch.aten.item %3166 : !torch.vtensor<[],f32> -> !torch.float | |
%3169 = torch.aten.item %3167 : !torch.vtensor<[],si8> -> !torch.int | |
%3170 = torch.aten._make_per_tensor_quantized_tensor %3165, %3168, %3169 : !torch.vtensor<[?,32,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,32,?,?,?],!torch.qint8> | |
%3171 = torch.aten.dequantize.self %3170 : !torch.vtensor<[?,32,?,?,?],!torch.qint8> -> !torch.vtensor<[?,32,?,?,?],f32> | |
%3172 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3173 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_1027 = torch.constant.int 12 | |
%3174 = torch.aten.item %3172 : !torch.vtensor<[],f32> -> !torch.float | |
%3175 = torch.aten.item %3173 : !torch.vtensor<[],si8> -> !torch.int | |
%3176 = torch.aten.quantize_per_tensor %42, %3174, %3175, %int12_1027 : !torch.vtensor<[1,32,1,1,1],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[1,32,1,1,1],!torch.qint8> | |
%3177 = torch.aten.int_repr %3176 : !torch.vtensor<[1,32,1,1,1],!torch.qint8> -> !torch.vtensor<[1,32,1,1,1],si8> | |
%3178 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3179 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%3180 = torch.aten.item %3178 : !torch.vtensor<[],f32> -> !torch.float | |
%3181 = torch.aten.item %3179 : !torch.vtensor<[],si8> -> !torch.int | |
%3182 = torch.aten._make_per_tensor_quantized_tensor %3177, %3180, %3181 : !torch.vtensor<[1,32,1,1,1],si8>, !torch.float, !torch.int -> !torch.vtensor<[1,32,1,1,1],!torch.qint8> | |
%3183 = torch.aten.dequantize.self %3182 : !torch.vtensor<[1,32,1,1,1],!torch.qint8> -> !torch.vtensor<[1,32,1,1,1],f32> | |
%3184 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3185 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_1028 = torch.constant.int 12 | |
%3186 = torch.aten.item %3184 : !torch.vtensor<[],f32> -> !torch.float | |
%3187 = torch.aten.item %3185 : !torch.vtensor<[],si8> -> !torch.int | |
%3188 = torch.aten.quantize_per_tensor %43, %3186, %3187, %int12_1028 : !torch.vtensor<[1],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[1],!torch.qint8> | |
%3189 = torch.aten.int_repr %3188 : !torch.vtensor<[1],!torch.qint8> -> !torch.vtensor<[1],si8> | |
%3190 = torch.vtensor.literal(dense<0.001953125> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3191 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%3192 = torch.aten.item %3190 : !torch.vtensor<[],f32> -> !torch.float | |
%3193 = torch.aten.item %3191 : !torch.vtensor<[],si8> -> !torch.int | |
%3194 = torch.aten._make_per_tensor_quantized_tensor %3189, %3192, %3193 : !torch.vtensor<[1],si8>, !torch.float, !torch.int -> !torch.vtensor<[1],!torch.qint8> | |
%3195 = torch.aten.dequantize.self %3194 : !torch.vtensor<[1],!torch.qint8> -> !torch.vtensor<[1],f32> | |
%int0_1029 = torch.constant.int 0 | |
%int0_1030 = torch.constant.int 0 | |
%int0_1031 = torch.constant.int 0 | |
%int1_1032 = torch.constant.int 1 | |
%int1_1033 = torch.constant.int 1 | |
%int1_1034 = torch.constant.int 1 | |
%int1_1035 = torch.constant.int 1 | |
%int1_1036 = torch.constant.int 1 | |
%int1_1037 = torch.constant.int 1 | |
%int0_1038 = torch.constant.int 0 | |
%3196 = torch.prim.ListConstruct %int0_1029, %int0_1030, %int0_1031 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%3197 = torch.prim.ListConstruct %int1_1032, %int1_1033, %int1_1034 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%3198 = torch.prim.ListConstruct %int1_1035, %int1_1036, %int1_1037 : (!torch.int, !torch.int, !torch.int) -> !torch.list<int> | |
%3199 = torch.prim.ListConstruct %int0_1038, %int0_1038 : (!torch.int, !torch.int) -> !torch.list<int> | |
%false_1039 = torch.constant.bool false | |
%int1_1040 = torch.constant.int 1 | |
%3200 = torch.aten.convolution %3171, %3183, %3195, %3198, %3196, %3197, %false_1039, %3199, %int1_1040 : !torch.vtensor<[?,32,?,?,?],f32>, !torch.vtensor<[1,32,1,1,1],f32>, !torch.vtensor<[1],f32>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.list<int>, !torch.int -> !torch.vtensor<[?,1,?,?,?],f32> | |
%3201 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3202 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%int12_1041 = torch.constant.int 12 | |
%3203 = torch.aten.item %3201 : !torch.vtensor<[],f32> -> !torch.float | |
%3204 = torch.aten.item %3202 : !torch.vtensor<[],si8> -> !torch.int | |
%3205 = torch.aten.quantize_per_tensor %3200, %3203, %3204, %int12_1041 : !torch.vtensor<[?,1,?,?,?],f32>, !torch.float, !torch.int, !torch.int -> !torch.vtensor<[?,1,?,?,?],!torch.qint8> | |
%3206 = torch.aten.int_repr %3205 : !torch.vtensor<[?,1,?,?,?],!torch.qint8> -> !torch.vtensor<[?,1,?,?,?],si8> | |
%3207 = torch.vtensor.literal(dense<1.250000e-01> : tensor<f32>) : !torch.vtensor<[],f32> | |
%3208 = torch.vtensor.literal(dense<0> : tensor<si8>) : !torch.vtensor<[],si8> | |
%3209 = torch.aten.item %3207 : !torch.vtensor<[],f32> -> !torch.float | |
%3210 = torch.aten.item %3208 : !torch.vtensor<[],si8> -> !torch.int | |
%3211 = torch.aten._make_per_tensor_quantized_tensor %3206, %3209, %3210 : !torch.vtensor<[?,1,?,?,?],si8>, !torch.float, !torch.int -> !torch.vtensor<[?,1,?,?,?],!torch.qint8> | |
%3212 = torch.aten.dequantize.self %3211 : !torch.vtensor<[?,1,?,?,?],!torch.qint8> -> !torch.vtensor<[?,1,?,?,?],f32> | |
%3213 = torch.aten.sigmoid %3212 : !torch.vtensor<[?,1,?,?,?],f32> -> !torch.vtensor<[1,1,64,128,128],f32> | |
return %3213 : !torch.vtensor<[1,1,64,128,128],f32> | |
} | |
} | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment