pytorch

1
ir_version: 8
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
5
  node {
6
    input: "onnx::ReduceMean_0"
7
    output: "onnx::Sub_1"
8
    name: "ReduceMean_2"
9
    op_type: "ReduceMean"
10
    attribute {
11
      name: "axes"
12
      ints: 0
13
      ints: 1
14
      type: INTS
15
    }
16
    attribute {
17
      name: "keepdims"
18
      i: 1
19
      type: INT
20
    }
21
  }
22
  node {
23
    input: "onnx::ReduceMean_0"
24
    output: "onnx::Gather_2"
25
    name: "Shape_3"
26
    op_type: "Shape"
27
  }
28
  node {
29
    output: "onnx::Gather_3"
30
    name: "Constant_4"
31
    op_type: "Constant"
32
    attribute {
33
      name: "value"
34
      t {
35
        dims: 2
36
        data_type: 7
37
        raw_data: "\000\000\000\000\000\000\000\000\001\000\000\000\000\000\000\000"
38
      }
39
      type: TENSOR
40
    }
41
  }
42
  node {
43
    input: "onnx::Gather_2"
44
    input: "onnx::Gather_3"
45
    output: "onnx::ReduceProd_4"
46
    name: "Gather_5"
47
    op_type: "Gather"
48
    attribute {
49
      name: "axis"
50
      i: 0
51
      type: INT
52
    }
53
  }
54
  node {
55
    input: "onnx::ReduceProd_4"
56
    output: "onnx::Cast_5"
57
    name: "ReduceProd_6"
58
    op_type: "ReduceProd"
59
    attribute {
60
      name: "keepdims"
61
      i: 0
62
      type: INT
63
    }
64
  }
65
  node {
66
    input: "onnx::ReduceMean_0"
67
    input: "onnx::Sub_1"
68
    output: "onnx::Mul_6"
69
    name: "Sub_7"
70
    op_type: "Sub"
71
  }
72
  node {
73
    input: "onnx::Mul_6"
74
    input: "onnx::Mul_6"
75
    output: "onnx::ReduceMean_7"
76
    name: "Mul_8"
77
    op_type: "Mul"
78
  }
79
  node {
80
    input: "onnx::ReduceMean_7"
81
    output: "onnx::Mul_8"
82
    name: "ReduceMean_9"
83
    op_type: "ReduceMean"
84
    attribute {
85
      name: "axes"
86
      ints: 0
87
      ints: 1
88
      type: INTS
89
    }
90
    attribute {
91
      name: "keepdims"
92
      i: 1
93
      type: INT
94
    }
95
  }
96
  node {
97
    input: "onnx::Cast_5"
98
    output: "onnx::Mul_9"
99
    name: "Cast_10"
100
    op_type: "Cast"
101
    attribute {
102
      name: "to"
103
      i: 1
104
      type: INT
105
    }
106
  }
107
  node {
108
    input: "onnx::Mul_8"
109
    input: "onnx::Mul_9"
110
    output: "onnx::Div_10"
111
    name: "Mul_11"
112
    op_type: "Mul"
113
  }
114
  node {
115
    output: "onnx::Sub_11"
116
    name: "Constant_12"
117
    op_type: "Constant"
118
    attribute {
119
      name: "value"
120
      t {
121
        data_type: 1
122
        raw_data: "\000\000\200?"
123
      }
124
      type: TENSOR
125
    }
126
  }
127
  node {
128
    input: "onnx::Mul_9"
129
    input: "onnx::Sub_11"
130
    output: "onnx::Div_12"
131
    name: "Sub_13"
132
    op_type: "Sub"
133
  }
134
  node {
135
    input: "onnx::Div_10"
136
    input: "onnx::Div_12"
137
    output: "onnx::Sqrt_13"
138
    name: "Div_14"
139
    op_type: "Div"
140
  }
141
  node {
142
    input: "onnx::Sqrt_13"
143
    output: "14"
144
    name: "Sqrt_15"
145
    op_type: "Sqrt"
146
  }
147
  name: "main_graph"
148
  input {
149
    name: "onnx::ReduceMean_0"
150
    type {
151
      tensor_type {
152
        elem_type: 1
153
        shape {
154
          dim {
155
            dim_value: 2
156
          }
157
          dim {
158
            dim_value: 3
159
          }
160
          dim {
161
            dim_value: 4
162
          }
163
        }
164
      }
165
    }
166
  }
167
  output {
168
    name: "14"
169
    type {
170
      tensor_type {
171
        elem_type: 1
172
        shape {
173
          dim {
174
            dim_value: 1
175
          }
176
          dim {
177
            dim_value: 1
178
          }
179
          dim {
180
            dim_value: 4
181
          }
182
        }
183
      }
184
    }
185
  }
186
}
187
opset_import {
188
  version: 17
189
}
190