pytorch

1
ir_version: 8
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
5
  node {
6
    input: "x"
7
    output: "onnx::Gather_1"
8
    name: "Shape_7"
9
    op_type: "Shape"
10
  }
11
  node {
12
    output: "onnx::Gather_2"
13
    name: "Constant_8"
14
    op_type: "Constant"
15
    attribute {
16
      name: "value"
17
      t {
18
        data_type: 7
19
        raw_data: "\000\000\000\000\000\000\000\000"
20
      }
21
      type: TENSOR
22
    }
23
  }
24
  node {
25
    input: "onnx::Gather_1"
26
    input: "onnx::Gather_2"
27
    output: "onnx::Unsqueeze_3"
28
    name: "Gather_9"
29
    op_type: "Gather"
30
    attribute {
31
      name: "axis"
32
      i: 0
33
      type: INT
34
    }
35
  }
36
  node {
37
    output: "onnx::Unsqueeze_7"
38
    name: "Constant_10"
39
    op_type: "Constant"
40
    attribute {
41
      name: "value"
42
      t {
43
        dims: 1
44
        data_type: 7
45
        raw_data: "\000\000\000\000\000\000\000\000"
46
      }
47
      type: TENSOR
48
    }
49
  }
50
  node {
51
    input: "onnx::Unsqueeze_3"
52
    input: "onnx::Unsqueeze_7"
53
    output: "onnx::Concat_8"
54
    name: "Unsqueeze_11"
55
    op_type: "Unsqueeze"
56
  }
57
  node {
58
    output: "onnx::Concat_25"
59
    name: "Constant_12"
60
    op_type: "Constant"
61
    attribute {
62
      name: "value"
63
      t {
64
        dims: 1
65
        data_type: 7
66
        raw_data: "\001\000\000\000\000\000\000\000"
67
      }
68
      type: TENSOR
69
    }
70
  }
71
  node {
72
    output: "onnx::Concat_26"
73
    name: "Constant_13"
74
    op_type: "Constant"
75
    attribute {
76
      name: "value"
77
      t {
78
        dims: 1
79
        data_type: 7
80
        raw_data: "\002\000\000\000\000\000\000\000"
81
      }
82
      type: TENSOR
83
    }
84
  }
85
  node {
86
    output: "onnx::Concat_27"
87
    name: "Constant_14"
88
    op_type: "Constant"
89
    attribute {
90
      name: "value"
91
      t {
92
        dims: 1
93
        data_type: 7
94
        raw_data: "\377\377\377\377\377\377\377\377"
95
      }
96
      type: TENSOR
97
    }
98
  }
99
  node {
100
    input: "onnx::Concat_8"
101
    input: "onnx::Concat_25"
102
    input: "onnx::Concat_26"
103
    input: "onnx::Concat_27"
104
    output: "onnx::Reshape_15"
105
    name: "Concat_15"
106
    op_type: "Concat"
107
    attribute {
108
      name: "axis"
109
      i: 0
110
      type: INT
111
    }
112
  }
113
  node {
114
    input: "x"
115
    input: "onnx::Reshape_15"
116
    output: "onnx::Transpose_16"
117
    name: "Reshape_16"
118
    op_type: "Reshape"
119
    attribute {
120
      name: "allowzero"
121
      i: 0
122
      type: INT
123
    }
124
  }
125
  node {
126
    input: "onnx::Transpose_16"
127
    output: "x.1"
128
    name: "Transpose_17"
129
    op_type: "Transpose"
130
    attribute {
131
      name: "perm"
132
      ints: 0
133
      ints: 2
134
      ints: 1
135
      ints: 3
136
      type: INTS
137
    }
138
  }
139
  node {
140
    input: "x.1"
141
    output: "onnx::Reshape_18"
142
    name: "Softmax_18"
143
    op_type: "Softmax"
144
    attribute {
145
      name: "axis"
146
      i: 1
147
      type: INT
148
    }
149
  }
150
  node {
151
    output: "onnx::Unsqueeze_19"
152
    name: "Constant_19"
153
    op_type: "Constant"
154
    attribute {
155
      name: "value"
156
      t {
157
        dims: 1
158
        data_type: 7
159
        raw_data: "\000\000\000\000\000\000\000\000"
160
      }
161
      type: TENSOR
162
    }
163
  }
164
  node {
165
    input: "onnx::Unsqueeze_3"
166
    input: "onnx::Unsqueeze_19"
167
    output: "onnx::Concat_20"
168
    name: "Unsqueeze_20"
169
    op_type: "Unsqueeze"
170
  }
171
  node {
172
    output: "onnx::Concat_28"
173
    name: "Constant_21"
174
    op_type: "Constant"
175
    attribute {
176
      name: "value"
177
      t {
178
        dims: 1
179
        data_type: 7
180
        raw_data: "\377\377\377\377\377\377\377\377"
181
      }
182
      type: TENSOR
183
    }
184
  }
185
  node {
186
    input: "onnx::Concat_20"
187
    input: "onnx::Concat_28"
188
    output: "onnx::Reshape_23"
189
    name: "Concat_22"
190
    op_type: "Concat"
191
    attribute {
192
      name: "axis"
193
      i: 0
194
      type: INT
195
    }
196
  }
197
  node {
198
    input: "onnx::Reshape_18"
199
    input: "onnx::Reshape_23"
200
    output: "24"
201
    name: "Reshape_23"
202
    op_type: "Reshape"
203
    attribute {
204
      name: "allowzero"
205
      i: 0
206
      type: INT
207
    }
208
  }
209
  name: "main_graph"
210
  input {
211
    name: "x"
212
    type {
213
      tensor_type {
214
        elem_type: 1
215
        shape {
216
          dim {
217
            dim_param: "x_dim_0"
218
          }
219
          dim {
220
            dim_value: 1
221
          }
222
          dim {
223
            dim_value: 128
224
          }
225
          dim {
226
            dim_value: 1
227
          }
228
        }
229
      }
230
    }
231
  }
232
  output {
233
    name: "24"
234
    type {
235
      tensor_type {
236
        elem_type: 1
237
        shape {
238
          dim {
239
            dim_param: "x_dim_0"
240
          }
241
          dim {
242
            dim_value: 128
243
          }
244
        }
245
      }
246
    }
247
  }
248
}
249
opset_import {
250
  version: 17
251
}
252