Method set_attr

ppq/parser/caffe/caffe_export_utils.py:267–268

(self)

    # TODO: Can optimize some case to Scale + Reshape
    #  (lhs_shape[1] == rhs_shape[1] and all(i == 1 for i in rhs_shape[2:]))
    def set_attr(self):
        self.layer.eltwise_param.operation = ppl_caffe_pb2.EltwiseParameter.PROD

    def parse(self):
        self.set_attr()