Method set_attr

ppq/parser/caffe/caffe_export_utils.py:238–240

(self)

@register_class
class ReduceL2(CaffeOpExporter):
    def set_attr(self):
        self.layer.reducel2_param.axes = refine_value(self.op.attributes.get('axes'))
        self.layer.reducel2_param.keepdims = refine_value(self.op.attributes.get('keepdims', 1))


@register_class