• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1ir_version: 8
2producer_name: "pytorch"
3producer_version: "CURRENT_VERSION"
4graph {
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}
249opset_import {
250  version: 17
251}
252