(self)
| 358 | @register_class |
| 359 | class Clip(CaffeOpExporter): |
| 360 | def parse(self): |
| 361 | self.layer.bottom[:] = [var.name for var in self.op.inputs if not var.is_parameter] |
| 362 | self.layer.top[:] = [var.name for var in self.op.outputs] |
| 363 | |
| 364 | min_val = refine_value(self.op.attributes.get('min')) |
| 365 | max_val = refine_value(self.op.attributes.get('max')) |
| 366 | if len(self.op.parameters) == 2: |
| 367 | min_val = convert_any_to_numpy(self.op.parameters[0].value).item() |
| 368 | max_val = convert_any_to_numpy(self.op.parameters[1].value).item() |
| 369 | |
| 370 | if min_val == 0.0 and max_val == 6.0: |
| 371 | self.layer.type = 'ReLU6' |
| 372 | else: |
| 373 | self.layer.clip_param.min = min_val |
| 374 | self.layer.clip_param.max = max_val |
| 375 | return self.layer |
| 376 | |
| 377 | |
| 378 | @register_class |
nothing calls this directly
no test coverage detected