pytorch

1
ir_version: 5
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
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
}
90
opset_import {
91
  version: 10
92
}
93