Ë
    Ž”vhS  ã                   óV  — d dl mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2 d dl3m4Z4 d dl5m6Z6m7Z7m8Z8m9Z9 d dl:m;Z; ddefiZ<d dl=Z>e>j~                  reZ[>[yd dl@mAZB  eBeCe<«      ZD[B[>[y)	é    )0ÚAxisContextÚConstantHandlerÚDEVICE_TO_DEVICE_TYPEÚLoweringParametersÚLoweringResultÚLoweringRuleÚLoweringRuleContextÚModuleContextÚRECV_FROM_HOST_TYPEÚSEND_TO_HOST_TYPEÚShapePolyLoweringStateÚTokenÚTokenSetÚValueÚcall_loweringÚ
_loweringsÚ_platform_specific_loweringsÚaval_to_ir_typeÚaval_to_ir_typesÚcore_call_loweringÚcustom_callÚdense_bool_elementsÚdense_bool_arrayÚdense_int_arrayÚdense_int_elementsÚdtype_to_ir_typeÚflatten_ir_typesÚflatten_ir_valuesr   Úunflatten_ir_values_like_typesÚi32_attrÚi64_attrÚirÚir_constantÚir_type_handlersÚjaxpr_subcompÚ	lower_funÚlower_jaxpr_to_funÚlower_jaxpr_to_moduleÚlowerable_effectsÚmake_ir_contextÚmerge_mlir_modulesÚmodule_to_bytecodeÚmodule_to_stringÚregister_constant_handlerÚregister_loweringÚshape_tensorÚ
token_type)ÚMesh)ÚMeshAxisNameÚReplicaAxisContextÚSPMDAxisContextÚShardingContext)Úemit_python_callbackr   zImlir.custom_call is deprecated; use the APIs provided by jax.ffi instead.N)Údeprecation_getattr)EÚjax._src.interpreters.mlirr   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   Ú_call_loweringr   r   r   r   r   r   Ú_custom_callr   r   r   r   r   r   r   Úflatten_lowering_ir_argsr   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   Újax._src.meshr2   Újax._src.sharding_implsr3   r4   r5   r6   Újax._src.callbackr7   Ú_deprecationsÚtypingÚ_typingÚTYPE_CHECKINGÚjax._src.deprecationsr8   Ú_deprecation_getattrÚ__name__Ú__getattr__© ó    úP/opt/face_recognition/venv/lib/python3.12/site-packages/jax/interpreters/mlir.pyú<module>rK      s¨   ð÷1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1ó 1õf '÷ó õð ØSØðð€ó à
×ÒØ€+ð Ùõ Pá$ X¨}Ó=€+ØØÙrI   