pytorch
1ir_version: 8
2producer_name: "pytorch"
3producer_version: "CURRENT_VERSION"
4graph {
5node {
6input: "onnx::GatherElements_0"
7input: "onnx::GatherElements_1"
8output: "2"
9name: "GatherElements_0"
10op_type: "GatherElements"
11attribute {
12name: "axis"
13i: 1
14type: INT
15}
16}
17name: "main_graph"
18input {
19name: "onnx::GatherElements_0"
20type {
21tensor_type {
22elem_type: 1
23shape {
24dim {
25dim_value: 3
26}
27dim {
28dim_value: 4
29}
30dim {
31dim_value: 3
32}
33}
34}
35}
36}
37input {
38name: "onnx::GatherElements_1"
39type {
40tensor_type {
41elem_type: 7
42shape {
43dim {
44dim_value: 3
45}
46dim {
47dim_value: 2
48}
49dim {
50dim_value: 3
51}
52}
53}
54}
55}
56output {
57name: "2"
58type {
59tensor_type {
60elem_type: 1
61shape {
62dim {
63dim_value: 3
64}
65dim {
66dim_value: 2
67}
68dim {
69dim_value: 3
70}
71}
72}
73}
74}
75}
76opset_import {
77version: 17
78}
79