xref: /aosp_15_r20/external/pytorch/test/onnx/expect/TestOperators.test_chunk.expect (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1ir_version: 8
2producer_name: "pytorch"
3producer_version: "CURRENT_VERSION"
4graph {
5  node {
6    input: "onnx::Shape_0"
7    output: "onnx::Gather_1"
8    name: "Shape_6"
9    op_type: "Shape"
10  }
11  node {
12    output: "onnx::Gather_2"
13    name: "Constant_7"
14    op_type: "Constant"
15    attribute {
16      name: "value"
17      t {
18        dims: 1
19        data_type: 7
20        raw_data: "\000\000\000\000\000\000\000\000"
21      }
22      type: TENSOR
23    }
24  }
25  node {
26    input: "onnx::Gather_1"
27    input: "onnx::Gather_2"
28    output: "onnx::Add_3"
29    name: "Gather_8"
30    op_type: "Gather"
31    attribute {
32      name: "axis"
33      i: 0
34      type: INT
35    }
36  }
37  node {
38    output: "onnx::Slice_4"
39    name: "Constant_9"
40    op_type: "Constant"
41    attribute {
42      name: "value"
43      t {
44        dims: 1
45        data_type: 7
46        raw_data: "\000\000\000\000\000\000\000\000"
47      }
48      type: TENSOR
49    }
50  }
51  node {
52    output: "onnx::Add_5"
53    name: "Constant_10"
54    op_type: "Constant"
55    attribute {
56      name: "value"
57      t {
58        dims: 1
59        data_type: 7
60        raw_data: "\001\000\000\000\000\000\000\000"
61      }
62      type: TENSOR
63    }
64  }
65  node {
66    input: "onnx::Add_3"
67    input: "onnx::Add_5"
68    output: "onnx::Div_6"
69    name: "Add_11"
70    op_type: "Add"
71  }
72  node {
73    output: "onnx::Div_7"
74    name: "Constant_12"
75    op_type: "Constant"
76    attribute {
77      name: "value"
78      t {
79        dims: 1
80        data_type: 7
81        raw_data: "\002\000\000\000\000\000\000\000"
82      }
83      type: TENSOR
84    }
85  }
86  node {
87    input: "onnx::Div_6"
88    input: "onnx::Div_7"
89    output: "onnx::Mul_8"
90    name: "Div_13"
91    op_type: "Div"
92  }
93  node {
94    output: "onnx::Mul_9"
95    name: "Constant_14"
96    op_type: "Constant"
97    attribute {
98      name: "value"
99      t {
100        dims: 1
101        data_type: 7
102        raw_data: "\001\000\000\000\000\000\000\000"
103      }
104      type: TENSOR
105    }
106  }
107  node {
108    input: "onnx::Mul_8"
109    input: "onnx::Mul_9"
110    output: "onnx::Slice_10"
111    name: "Mul_15"
112    op_type: "Mul"
113  }
114  node {
115    input: "onnx::Shape_0"
116    input: "onnx::Slice_4"
117    input: "onnx::Slice_10"
118    input: "onnx::Gather_2"
119    output: "11"
120    name: "Slice_16"
121    op_type: "Slice"
122  }
123  node {
124    output: "onnx::Mul_12"
125    name: "Constant_17"
126    op_type: "Constant"
127    attribute {
128      name: "value"
129      t {
130        dims: 1
131        data_type: 7
132        raw_data: "\002\000\000\000\000\000\000\000"
133      }
134      type: TENSOR
135    }
136  }
137  node {
138    input: "onnx::Mul_8"
139    input: "onnx::Mul_12"
140    output: "onnx::Slice_13"
141    name: "Mul_18"
142    op_type: "Mul"
143  }
144  node {
145    input: "onnx::Shape_0"
146    input: "onnx::Slice_10"
147    input: "onnx::Slice_13"
148    input: "onnx::Gather_2"
149    output: "14"
150    name: "Slice_19"
151    op_type: "Slice"
152  }
153  name: "main_graph"
154  input {
155    name: "onnx::Shape_0"
156    type {
157      tensor_type {
158        elem_type: 1
159        shape {
160          dim {
161            dim_value: 3
162          }
163        }
164      }
165    }
166  }
167  output {
168    name: "11"
169    type {
170      tensor_type {
171        elem_type: 1
172        shape {
173          dim {
174            dim_value: 2
175          }
176        }
177      }
178    }
179  }
180  output {
181    name: "14"
182    type {
183      tensor_type {
184        elem_type: 1
185        shape {
186          dim {
187            dim_value: 1
188          }
189        }
190      }
191    }
192  }
193}
194opset_import {
195  version: 17
196}
197