U yh]@sUddlZddlZddlmZddlZeedddZeedddZGdd d Z Gd d d Z Gd d d ej Z e ej eeej e<eed<eeed<dS)N)List)qenginereturncCsl|dks|dks|dkrd}nJ|dkr,d}n<|dkr:d}n.|dkrHd }n |d krVd }nd }t|d |S)Nnonerfbgemmqnnpackonednnx86z* is not a valid value for quantized engine RuntimeError)rretrS/var/www/html/venv/lib/python3.8/site-packages/torch/backends/quantized/__init__.py_get_qengine_id srcCsdddddd}||dS)Nrrr r r )rrr r rz *undefined)get)rZ all_enginesrrr_get_qengine_strsrc@s*eZdZedddZeddddZdS) _QEngineProprcCsttjSN)rtorch_CZ _get_qengine)selfobjobjtyperrr__get__"sz_QEngineProp.__get__N)valrcCstjt|dSr)rrZ _set_qenginerrrr!rrr__set__%sz_QEngineProp.__set__)__name__ __module__ __qualname__strr r#rrrrr!src@s,eZdZeedddZddddZdS)_SupportedQEnginesProprcCstj}dd|DS)NcSsg|] }t|qSr)r).0Zqerrr ,sz2_SupportedQEnginesProp.__get__..)rrZ_supported_qengines)rrrZqenginesrrrr *s z_SupportedQEnginesProp.__get__NcCs tddS)NzAssignment not supportedrr"rrrr#.sz_SupportedQEnginesProp.__set__)r$r%r&rr'r r#rrrrr()sr(cs0eZdZfddZddZeZeZZ S)QuantizedEnginecst|||_dSr)super__init__m)rr.name __class__rrr-3s zQuantizedEngine.__init__cCs |j|Sr)r.__getattribute__)rattrrrr __getattr__7szQuantizedEngine.__getattr__) r$r%r&r-r4renginer(supported_engines __classcell__rrr0rr+2s r+r5r6)systypestypingrrr'intrrrr( ModuleTyper+modulesr$__annotations__rrrrs