1ir_version: 8 2producer_name: "pytorch" 3producer_version: "CURRENT_VERSION" 4graph { 5 node { 6 output: "onnx::Loop_33" 7 name: "Constant_9" 8 op_type: "Constant" 9 attribute { 10 name: "value" 11 t { 12 data_type: 9 13 raw_data: "\001" 14 } 15 type: TENSOR 16 } 17 } 18 node { 19 output: "5" 20 name: "Constant_10" 21 op_type: "Constant" 22 attribute { 23 name: "value" 24 t { 25 dims: 1 26 data_type: 7 27 raw_data: "\000\000\000\000\000\000\000\000" 28 } 29 type: TENSOR 30 } 31 } 32 node { 33 input: "input" 34 output: "onnx::Gather_6" 35 name: "Shape_11" 36 op_type: "Shape" 37 } 38 node { 39 output: "onnx::Gather_7" 40 name: "Constant_12" 41 op_type: "Constant" 42 attribute { 43 name: "value" 44 t { 45 data_type: 7 46 raw_data: "\000\000\000\000\000\000\000\000" 47 } 48 type: TENSOR 49 } 50 } 51 node { 52 input: "onnx::Gather_6" 53 input: "onnx::Gather_7" 54 output: "onnx::Unsqueeze_8" 55 name: "Gather_13" 56 op_type: "Gather" 57 attribute { 58 name: "axis" 59 i: 0 60 type: INT 61 } 62 } 63 node { 64 output: "onnx::Unsqueeze_9" 65 name: "Constant_14" 66 op_type: "Constant" 67 attribute { 68 name: "value" 69 t { 70 dims: 1 71 data_type: 7 72 raw_data: "\000\000\000\000\000\000\000\000" 73 } 74 type: TENSOR 75 } 76 } 77 node { 78 input: "onnx::Unsqueeze_8" 79 input: "onnx::Unsqueeze_9" 80 output: "onnx::Concat_10" 81 name: "Unsqueeze_15" 82 op_type: "Unsqueeze" 83 } 84 node { 85 input: "offsets" 86 input: "onnx::Concat_10" 87 output: "onnx::Slice_11" 88 name: "Concat_16" 89 op_type: "Concat" 90 attribute { 91 name: "axis" 92 i: 0 93 type: INT 94 } 95 } 96 node { 97 output: "onnx::Slice_12" 98 name: "Constant_17" 99 op_type: "Constant" 100 attribute { 101 name: "value" 102 t { 103 dims: 1 104 data_type: 7 105 raw_data: "\000\000\000\000\000\000\000\000" 106 } 107 type: TENSOR 108 } 109 } 110 node { 111 output: "onnx::Slice_13" 112 name: "Constant_18" 113 op_type: "Constant" 114 attribute { 115 name: "value" 116 t { 117 dims: 1 118 data_type: 7 119 raw_data: "\001\000\000\000\000\000\000\000" 120 } 121 type: TENSOR 122 } 123 } 124 node { 125 output: "onnx::Slice_14" 126 name: "Constant_19" 127 op_type: "Constant" 128 attribute { 129 name: "value" 130 t { 131 dims: 1 132 data_type: 7 133 raw_data: "\377\377\377\377\377\377\377\177" 134 } 135 type: TENSOR 136 } 137 } 138 node { 139 output: "onnx::Slice_15" 140 name: "Constant_20" 141 op_type: "Constant" 142 attribute { 143 name: "value" 144 t { 145 dims: 1 146 data_type: 7 147 raw_data: "\001\000\000\000\000\000\000\000" 148 } 149 type: TENSOR 150 } 151 } 152 node { 153 input: "onnx::Slice_11" 154 input: "onnx::Slice_13" 155 input: "onnx::Slice_14" 156 input: "onnx::Slice_12" 157 input: "onnx::Slice_15" 158 output: "onnx::Shape_16" 159 name: "Slice_21" 160 op_type: "Slice" 161 } 162 node { 163 input: "onnx::Shape_16" 164 output: "onnx::Gather_17" 165 name: "Shape_22" 166 op_type: "Shape" 167 } 168 node { 169 output: "onnx::Gather_18" 170 name: "Constant_23" 171 op_type: "Constant" 172 attribute { 173 name: "value" 174 t { 175 data_type: 7 176 raw_data: "\000\000\000\000\000\000\000\000" 177 } 178 type: TENSOR 179 } 180 } 181 node { 182 input: "onnx::Gather_17" 183 input: "onnx::Gather_18" 184 output: "onnx::Loop_19" 185 name: "Gather_24" 186 op_type: "Gather" 187 attribute { 188 name: "axis" 189 i: 0 190 type: INT 191 } 192 } 193 node { 194 input: "onnx::Loop_19" 195 input: "onnx::Loop_33" 196 output: "20" 197 name: "Loop_25" 198 op_type: "Loop" 199 attribute { 200 name: "body" 201 g { 202 node { 203 input: "onnx::Slice_11" 204 input: "21" 205 output: "23" 206 name: "Gather_26" 207 op_type: "Gather" 208 attribute { 209 name: "axis" 210 i: 0 211 type: INT 212 } 213 } 214 node { 215 input: "onnx::Shape_16" 216 input: "21" 217 output: "24" 218 name: "Gather_27" 219 op_type: "Gather" 220 attribute { 221 name: "axis" 222 i: 0 223 type: INT 224 } 225 } 226 node { 227 output: "25" 228 name: "Constant_28" 229 op_type: "Constant" 230 attribute { 231 name: "value" 232 t { 233 dims: 1 234 data_type: 7 235 raw_data: "\000\000\000\000\000\000\000\000" 236 } 237 type: TENSOR 238 } 239 } 240 node { 241 input: "23" 242 input: "25" 243 output: "26" 244 name: "Unsqueeze_29" 245 op_type: "Unsqueeze" 246 } 247 node { 248 output: "27" 249 name: "Constant_30" 250 op_type: "Constant" 251 attribute { 252 name: "value" 253 t { 254 dims: 1 255 data_type: 7 256 raw_data: "\000\000\000\000\000\000\000\000" 257 } 258 type: TENSOR 259 } 260 } 261 node { 262 input: "24" 263 input: "27" 264 output: "28" 265 name: "Unsqueeze_31" 266 op_type: "Unsqueeze" 267 } 268 node { 269 input: "input" 270 input: "26" 271 input: "28" 272 input: "5" 273 output: "29" 274 name: "Slice_32" 275 op_type: "Slice" 276 } 277 node { 278 input: "weight" 279 input: "29" 280 output: "30" 281 name: "Gather_33" 282 op_type: "Gather" 283 attribute { 284 name: "axis" 285 i: 0 286 type: INT 287 } 288 } 289 node { 290 input: "30" 291 output: "31" 292 name: "ReduceMean_34" 293 op_type: "ReduceMean" 294 attribute { 295 name: "axes" 296 ints: 0 297 type: INTS 298 } 299 attribute { 300 name: "keepdims" 301 i: 0 302 type: INT 303 } 304 } 305 node { 306 input: "onnx::Loop_33" 307 output: "32" 308 name: "Cast_35" 309 op_type: "Cast" 310 attribute { 311 name: "to" 312 i: 9 313 type: INT 314 } 315 } 316 name: "sub_graph" 317 input { 318 name: "21" 319 type { 320 tensor_type { 321 elem_type: 7 322 shape { 323 } 324 } 325 } 326 } 327 input { 328 name: "22" 329 type { 330 tensor_type { 331 elem_type: 9 332 shape { 333 } 334 } 335 } 336 } 337 output { 338 name: "32" 339 type { 340 tensor_type { 341 elem_type: 9 342 shape { 343 } 344 } 345 } 346 } 347 output { 348 name: "31" 349 type { 350 tensor_type { 351 elem_type: 1 352 shape { 353 dim { 354 dim_param: "Loop20_dim_1" 355 } 356 } 357 } 358 } 359 } 360 } 361 type: GRAPH 362 } 363 } 364 name: "main_graph" 365 initializer { 366 dims: 10 367 dims: 8 368 data_type: 1 369 name: "weight" 370 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?\027X\277\277\253\206a?\354\335\226\277L\032o\277\251J\021\277\311\360\215\276\312\274\013\300\252\320\273>\220\"p?\267\020\000<R\262\240\276\343\016\224\2779\241\353?8;\202\277\023\020\234?E\370#>\222\233\314?\334\360?\275|t\303\277\214\351\000\300\3065\302\2775\206\306>X\251\227\277x\2160?U^\251?d\221\350?\237F.?\rp9?9X\004=/c\324\277SL\360\277\'\274<?t\375l?\342\270l?\240\352:>\332\356\226\275\211\035\241>*\271\204\277>\025W>\036K\035?\036\233\200=\035\313\250\276\017\003\346\277\374p_?\313WD?!\006\351\275\232\\q\277\230\007A?" 371 } 372 input { 373 name: "input" 374 type { 375 tensor_type { 376 elem_type: 7 377 shape { 378 dim { 379 dim_value: 4 380 } 381 } 382 } 383 } 384 } 385 input { 386 name: "offsets" 387 type { 388 tensor_type { 389 elem_type: 7 390 shape { 391 dim { 392 dim_value: 1 393 } 394 } 395 } 396 } 397 } 398 input { 399 name: "weight" 400 type { 401 tensor_type { 402 elem_type: 1 403 shape { 404 dim { 405 dim_value: 10 406 } 407 dim { 408 dim_value: 8 409 } 410 } 411 } 412 } 413 } 414 output { 415 name: "20" 416 type { 417 tensor_type { 418 elem_type: 1 419 shape { 420 dim { 421 dim_param: "Loop20_dim_0" 422 } 423 dim { 424 dim_param: "Loop20_dim_1" 425 } 426 } 427 } 428 } 429 } 430} 431opset_import { 432 version: 17 433} 434