xref: /aosp_15_r20/external/pytorch/test/onnx/expect/TestOperators.test_argmax.expect (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1ir_version: 8
2producer_name: "pytorch"
3producer_version: "CURRENT_VERSION"
4graph {
5  node {
6    input: "onnx::ArgMax_0"
7    output: "1"
8    name: "ArgMax_0"
9    op_type: "ArgMax"
10    attribute {
11      name: "axis"
12      i: 1
13      type: INT
14    }
15    attribute {
16      name: "keepdims"
17      i: 0
18      type: INT
19    }
20    attribute {
21      name: "select_last_index"
22      i: 0
23      type: INT
24    }
25  }
26  name: "main_graph"
27  input {
28    name: "onnx::ArgMax_0"
29    type {
30      tensor_type {
31        elem_type: 1
32        shape {
33          dim {
34            dim_value: 4
35          }
36          dim {
37            dim_value: 4
38          }
39        }
40      }
41    }
42  }
43  output {
44    name: "1"
45    type {
46      tensor_type {
47        elem_type: 7
48        shape {
49          dim {
50            dim_value: 4
51          }
52        }
53      }
54    }
55  }
56}
57opset_import {
58  version: 17
59}
60