o #Õjoã@s¨ddlZGdd„dƒZeejjdgdgdgƒeejjdgdgd gƒeejjdggdgƒeejjdggd gƒeejjdggd gƒeejjdggd gƒeejj dggd gƒeejj dggd gƒeejj dggd gƒeejj dggd gƒeejj dggd gƒeejjdggd gƒeejjdggd gƒeejjdggd gƒeejjjddggd gƒgZeejjjjdgdgdgƒeejjjjdgdgd gƒgZejjejjgZGd d „d ƒZdS) éNcs eZdZdZ‡fdd„Z‡ZS)Ú LayerInfoz7 Store the argnames of the inputs and outputs. cs&tƒ ¡||_||_||_||_dS©N)ÚsuperÚ__init__ÚlayerÚ input_namesÚ weight_namesÚ output_names)Úselfrrrr ©Ú __class__©úl/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/quantization/imperative/ptq_registry.pyrs  zLayerInfo.__init__)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚ __classcell__r r r rrsrZInputÚFilterÚOutputÚXÚYZOutcsheZdZdZiZiZdZ‡fdd„Zedd„ƒZ edd„ƒZ ed d „ƒZ ed d „ƒZ ed d„ƒZ ‡ZS)Ú PTQRegistryzH Register the supported layers for PTQ and provide layers info. Fcstƒ ¡dSr)rr)r r r rrIszPTQRegistry.__init__cCsD|jstD]}||j|j<qtt}|D]}||j|j<qd|_dS)NT)Ú is_initedÚPTQ_LAYERS_INFOÚsupported_layers_maprÚQUANT_LAYERS_INFOÚregistered_layers_map)ÚclsÚ layer_infoZall_layers_infor r rÚ_initLs zPTQRegistry._initcCó&| ¡||jvpt|t|j ¡ƒƒS)zè Analyze whether the layer supports quantization. Args: layer(Layer): The input layer can be a python class or an instance. Returns: flag(bool): Whther the layer is supported. )r rÚ isinstanceÚtupleÚkeys©rrr r rÚis_supported_layerWó ÿzPTQRegistry.is_supported_layercCr!)zó Analyze whether the layer is register layer_info. Args: layer(Layer): The input layer can be a python class or an instance. Returns: flag(bool): Wether the layer is register layer_info. )r rr"r#r$r%r r rÚis_registered_layerer'zPTQRegistry.is_registered_layercCs|tvp t|ttƒƒS)zë Analyze whether the layer is simulated quant layer. Args: layer(Layer): The input layer can be a python class or an instance. Returns: flag(bool): Whther the layer is supported. )ÚSIMULATED_LAYERSr"r#r%r r rÚis_simulated_quant_layerss ÿz$PTQRegistry.is_simulated_quant_layercCsD| |¡s Jdƒ‚|j ¡D]\}}||kst||ƒr|SqdS)zè Get the infomation for the layer. Args: layer(Layer): The input layer can be a python class or an instance. Returns: layer_info(LayerInfo): The layer info of the input layer. z The input layer is not register.N)r(rÚitemsr")rrZ layer_keyrr r rr€s ÿþÿÿzPTQRegistry.layer_info)rrrrrrrrÚ classmethodr r&r(r*rrr r r rr@s       r)ZpaddlerÚnnZConv2DZLinearZ BatchNorm2DZAdaptiveMaxPool2DZAdaptiveAvgPool2DZ AvgPool2DZ MaxPool2DZReLUZReLU6Z HardswishZSwishZSigmoidZSoftmaxZTanhZquantÚaddrZ quant_layersZQuantizedConv2DZQuantizedLinearrr)rr r r rÚs<ñ üÿù