xref: /aosp_15_r20/external/pytorch/test/onnx/expect/TestOperators.test_topk.expect (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1ir_version: 5
2producer_name: "pytorch"
3producer_version: "CURRENT_VERSION"
4graph {
5  node {
6    output: "onnx::Reshape_2"
7    name: "Constant_1"
8    op_type: "Constant"
9    attribute {
10      name: "value"
11      t {
12        dims: 1
13        data_type: 7
14        raw_data: "\001\000\000\000\000\000\000\000"
15      }
16      type: TENSOR
17    }
18  }
19  node {
20    input: "onnx::Reshape_1"
21    input: "onnx::Reshape_2"
22    output: "onnx::TopK_3"
23    name: "Reshape_2"
24    op_type: "Reshape"
25  }
26  node {
27    input: "onnx::TopK_0"
28    input: "onnx::TopK_3"
29    output: "4"
30    output: "5"
31    name: "TopK_3"
32    op_type: "TopK"
33    attribute {
34      name: "axis"
35      i: -1
36      type: INT
37    }
38  }
39  name: "main_graph"
40  input {
41    name: "onnx::TopK_0"
42    type {
43      tensor_type {
44        elem_type: 1
45        shape {
46          dim {
47            dim_value: 5
48          }
49        }
50      }
51    }
52  }
53  input {
54    name: "onnx::Reshape_1"
55    type {
56      tensor_type {
57        elem_type: 7
58        shape {
59        }
60      }
61    }
62  }
63  output {
64    name: "4"
65    type {
66      tensor_type {
67        elem_type: 1
68        shape {
69          dim {
70            dim_param: "TopK4_dim_0"
71          }
72        }
73      }
74    }
75  }
76  output {
77    name: "5"
78    type {
79      tensor_type {
80        elem_type: 7
81        shape {
82          dim {
83            dim_param: "TopK5_dim_0"
84          }
85        }
86      }
87    }
88  }
89}
90opset_import {
91  version: 10
92}
93