pytorch

1
ir_version: 4
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
5
  node {
6
    input: "_caffe2::GenerateProposals_0"
7
    input: "_caffe2::GenerateProposals_1"
8
    input: "_caffe2::GenerateProposals_2"
9
    input: "_caffe2::GenerateProposals_3"
10
    output: "4"
11
    output: "5"
12
    name: "GenerateProposals_0"
13
    op_type: "GenerateProposals"
14
    attribute {
15
      name: "spatial_scale"
16
      f: 2
17
      type: FLOAT
18
    }
19
    attribute {
20
      name: "pre_nms_topN"
21
      i: 6000
22
      type: INT
23
    }
24
    attribute {
25
      name: "post_nms_topN"
26
      i: 300
27
      type: INT
28
    }
29
    attribute {
30
      name: "nms_thresh"
31
      f: 0.7
32
      type: FLOAT
33
    }
34
    attribute {
35
      name: "min_size"
36
      f: 16
37
      type: FLOAT
38
    }
39
    attribute {
40
      name: "angle_bound_on"
41
      i: 1
42
      type: INT
43
    }
44
    attribute {
45
      name: "angle_bound_lo"
46
      i: -90
47
      type: INT
48
    }
49
    attribute {
50
      name: "angle_bound_hi"
51
      i: 90
52
      type: INT
53
    }
54
    attribute {
55
      name: "clip_angle_thresh"
56
      f: 1
57
      type: FLOAT
58
    }
59
    attribute {
60
      name: "legacy_plus_one"
61
      i: 1
62
      type: INT
63
    }
64
    domain: "org.pytorch._caffe2"
65
  }
66
  name: "main_graph"
67
  input {
68
    name: "_caffe2::GenerateProposals_0"
69
    type {
70
      tensor_type {
71
        elem_type: 1
72
        shape {
73
          dim {
74
            dim_value: 3
75
          }
76
          dim {
77
            dim_value: 4
78
          }
79
          dim {
80
            dim_value: 10
81
          }
82
          dim {
83
            dim_value: 8
84
          }
85
        }
86
      }
87
    }
88
  }
89
  input {
90
    name: "_caffe2::GenerateProposals_1"
91
    type {
92
      tensor_type {
93
        elem_type: 1
94
        shape {
95
          dim {
96
            dim_value: 3
97
          }
98
          dim {
99
            dim_value: 16
100
          }
101
          dim {
102
            dim_value: 10
103
          }
104
          dim {
105
            dim_value: 8
106
          }
107
        }
108
      }
109
    }
110
  }
111
  input {
112
    name: "_caffe2::GenerateProposals_2"
113
    type {
114
      tensor_type {
115
        elem_type: 1
116
        shape {
117
          dim {
118
            dim_value: 3
119
          }
120
          dim {
121
            dim_value: 3
122
          }
123
        }
124
      }
125
    }
126
  }
127
  input {
128
    name: "_caffe2::GenerateProposals_3"
129
    type {
130
      tensor_type {
131
        elem_type: 1
132
        shape {
133
          dim {
134
            dim_value: 4
135
          }
136
          dim {
137
            dim_value: 4
138
          }
139
        }
140
      }
141
    }
142
  }
143
  output {
144
    name: "4"
145
    type {
146
      tensor_type {
147
        elem_type: 1
148
        shape {
149
          dim {
150
            dim_param: "GenerateProposals4_dim_0"
151
          }
152
          dim {
153
            dim_param: "GenerateProposals4_dim_1"
154
          }
155
        }
156
      }
157
    }
158
  }
159
  output {
160
    name: "5"
161
    type {
162
      tensor_type {
163
        elem_type: 1
164
        shape {
165
          dim {
166
            dim_param: "GenerateProposals5_dim_0"
167
          }
168
        }
169
      }
170
    }
171
  }
172
}
173
opset_import {
174
  version: 9
175
}
176
opset_import {
177
  domain: "org.pytorch._caffe2"
178
  version: 0
179
}
180