1ir_version: 7 2producer_name: "pytorch" 3producer_version: "CURRENT_VERSION" 4graph { 5 node { 6 input: "emb.weight" 7 input: "input_1" 8 output: "onnx::Add_3" 9 name: "ATen_1" 10 op_type: "ATen" 11 attribute { 12 name: "custom_attributes_json" 13 s: "{\"padding_idx\":-1,\"scale_grad_by_freq\":false,\"sparse\":false}" 14 type: STRING 15 } 16 attribute { 17 name: "operator" 18 s: "embedding" 19 type: STRING 20 } 21 attribute { 22 name: "overload_name" 23 s: "" 24 type: STRING 25 } 26 domain: "org.pytorch.aten" 27 } 28 node { 29 input: "onnx::Add_3" 30 input: "input_2" 31 output: "onnx::Shape_4" 32 name: "Add_2" 33 op_type: "Add" 34 } 35 node { 36 input: "onnx::Shape_4" 37 output: "onnx::Gather_5" 38 name: "Shape_3" 39 op_type: "Shape" 40 } 41 node { 42 output: "onnx::Gather_6" 43 name: "Constant_4" 44 op_type: "Constant" 45 attribute { 46 name: "value" 47 t { 48 data_type: 7 49 raw_data: "\000\000\000\000\000\000\000\000" 50 } 51 type: TENSOR 52 } 53 } 54 node { 55 input: "onnx::Gather_5" 56 input: "onnx::Gather_6" 57 output: "onnx::Unsqueeze_7" 58 name: "Gather_5" 59 op_type: "Gather" 60 attribute { 61 name: "axis" 62 i: 0 63 type: INT 64 } 65 } 66 node { 67 input: "onnx::Unsqueeze_7" 68 output: "onnx::Concat_8" 69 name: "Unsqueeze_6" 70 op_type: "Unsqueeze" 71 attribute { 72 name: "axes" 73 ints: 0 74 type: INTS 75 } 76 } 77 node { 78 input: "onnx::Concat_8" 79 output: "onnx::ConstantOfShape_9" 80 name: "Concat_7" 81 op_type: "Concat" 82 attribute { 83 name: "axis" 84 i: 0 85 type: INT 86 } 87 } 88 node { 89 input: "onnx::ConstantOfShape_9" 90 output: "10" 91 name: "ConstantOfShape_8" 92 op_type: "ConstantOfShape" 93 attribute { 94 name: "value" 95 t { 96 dims: 1 97 data_type: 1 98 raw_data: "\000\000\200?" 99 } 100 type: TENSOR 101 } 102 } 103 name: "main_graph" 104 initializer { 105 dims: 4 106 dims: 8 107 data_type: 1 108 name: "emb.weight" 109 raw_data: "\264\314\344\275\017A\376\276\313\374&>J\266a\277s\306\\=\212\032+?\211[t\275\344[\357\276Dk\\\276OKb?\234\'B\277A\334\274\2767N\257\276\320s\263\277\371+\244>:\314\202\277K\200L??\001\275\275\236u4\2774\032\315\277\214\004\224>Z\320\372>\267B\305\276\346G6\277N\265.\276\343\316\272\277t\364a>\201)|>p\223\251\277Qm2?\346\275)\277\354\235\233?" 110 } 111 input { 112 name: "input_1" 113 type { 114 tensor_type { 115 elem_type: 7 116 shape { 117 dim { 118 dim_param: "input_1_dim_0" 119 } 120 } 121 } 122 } 123 } 124 input { 125 name: "input_2" 126 type { 127 tensor_type { 128 elem_type: 1 129 shape { 130 dim { 131 dim_param: "input_2_dim_0" 132 } 133 dim { 134 dim_param: "input_2_dim_1" 135 } 136 } 137 } 138 } 139 } 140 output { 141 name: "10" 142 type { 143 tensor_type { 144 elem_type: 1 145 shape { 146 dim { 147 dim_param: "ConstantOfShape10_dim_0" 148 } 149 } 150 } 151 } 152 } 153} 154opset_import { 155 version: 12 156} 157opset_import { 158 domain: "org.pytorch.aten" 159 version: 1 160} 161