Method set_attr

ppq/parser/caffe/caffe_export_utils.py:221–226

(self)

@register_class
class Softmax(CaffeOpExporter):
    def set_attr(self):
        axis = refine_value(self.op.attributes.get('axis', -1))
        if not (axis == -1 or axis == len(self.op.inputs[0].shape) - 1):
            logger.warning(f'Converting to caffe Softmax, the axis={axis}, which is not the last axis. '
                           'This may result to incorrect caffe model')
        self.layer.softmax_param.axis = axis


@register_class