pytorch

1
ir_version: 7
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
5
  node {
6
    input: "input"
7
    output: "onnx::Softmax_1"
8
    name: "Transpose_0"
9
    op_type: "Transpose"
10
    attribute {
11
      name: "perm"
12
      ints: 1
13
      ints: 0
14
      type: INTS
15
    }
16
  }
17
  node {
18
    input: "onnx::Softmax_1"
19
    output: "onnx::Transpose_2"
20
    name: "Softmax_1"
21
    op_type: "Softmax"
22
    attribute {
23
      name: "axis"
24
      i: 1
25
      type: INT
26
    }
27
  }
28
  node {
29
    input: "onnx::Transpose_2"
30
    output: "3"
31
    name: "Transpose_2"
32
    op_type: "Transpose"
33
    attribute {
34
      name: "perm"
35
      ints: 1
36
      ints: 0
37
      type: INTS
38
    }
39
  }
40
  name: "main_graph"
41
  input {
42
    name: "input"
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_1"
52
          }
53
        }
54
      }
55
    }
56
  }
57
  output {
58
    name: "3"
59
    type {
60
      tensor_type {
61
        elem_type: 1
62
        shape {
63
          dim {
64
            dim_value: 2
65
          }
66
          dim {
67
            dim_param: "input_dim_1"
68
          }
69
        }
70
      }
71
    }
72
  }
73
}
74
opset_import {
75
  version: 12
76
}
77