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