Method parse

ppq/parser/caffe/caffe_export_utils.py:270–282

(self)

        self.layer.eltwise_param.operation = ppl_caffe_pb2.EltwiseParameter.PROD

    def parse(self):
        self.set_attr()
        self.layer.bottom[:] = [var.name for var in self.op.inputs if not var.is_parameter]
        self.layer.top[:] = [var.name for var in self.op.outputs]

        for var in self.op.parameters:
            value = convert_any_to_numpy(var.value)
            if value.size != 1:
                raise AttributeError(f'Now don\'t support Mul op with initializer in shape {value.shape} convert to caffe')
            # Mul only has two inputs, thus in this loop means the bottom has only one item
            self.layer.eltwise_param.coeff.append(value.item())

        return self.layer


@register_class