pytorch

1
ir_version: 7
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
5
  node {
6
    input: "input"
7
    output: "1"
8
    name: "ReduceMean_0"
9
    op_type: "ReduceMean"
10
    attribute {
11
      name: "axes"
12
      ints: 1
13
      type: INTS
14
    }
15
    attribute {
16
      name: "keepdims"
17
      i: 0
18
      type: INT
19
    }
20
  }
21
  name: "main_graph"
22
  input {
23
    name: "input"
24
    type {
25
      tensor_type {
26
        elem_type: 1
27
        shape {
28
          dim {
29
            dim_value: 2
30
          }
31
          dim {
32
            dim_param: "input_dim_1"
33
          }
34
          dim {
35
            dim_param: "input_dim_2"
36
          }
37
        }
38
      }
39
    }
40
  }
41
  output {
42
    name: "1"
43
    type {
44
      tensor_type {
45
        elem_type: 1
46
        shape {
47
          dim {
48
            dim_value: 2
49
          }
50
          dim {
51
            dim_param: "input_dim_2"
52
          }
53
        }
54
      }
55
    }
56
  }
57
}
58
opset_import {
59
  version: 12
60
}
61