(self)
| 349 | global_pooling_arg.i = 1 |
| 350 | |
| 351 | def convert_ops(self): |
| 352 | layer_names = set() |
| 353 | for layer in self._caffe_layers.layer: |
| 354 | caffe_op = self._caffe_net.get_op(layer.name) |
| 355 | if caffe_op not in self._skip_ops: |
| 356 | mace_check(layer.name not in layer_names, |
| 357 | "There is duplicate layer name '%s' in your model" |
| 358 | % layer.name) |
| 359 | mace_check(layer.type in self._op_converters, |
| 360 | "Mace does not support caffe op type %s yet" |
| 361 | % layer.type) |
| 362 | layer_names.add(layer.name) |
| 363 | self._op_converters[layer.type](caffe_op) |
| 364 | |
| 365 | def add_tensor(self, name, shape, data_type, value): |
| 366 | tensor = self._mace_net_def.tensors.add() |
no test coverage detected