pytorch

TestOperators.test_embedding_bags.expect
433 строки · 8.7 Кб
Перенос по словам
1
ir_version: 8
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
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
}
431
opset_import {
432
  version: 17
433
}
434
pytorch

Использование cookies
