
    vh                     p   d dl Zd dlmZ d Zi dddddd	d
dddddddddddddddddej                  fddefddd d!ej                  fd"d#ej                  fd$d%ej                  fd&ej                  fd'ej                  fd(ej                  fd)ej                  fd*Zd dlZej"                  rhej                  ZeZd+Zej                  Zej                  Zej                  Z	ej                  Z
ej                  Zej                  Zej                  Z[[[yd d,lmZ  eee      Z[[[[y)-    N)
xla_clientc                 H    t        j                  | j                               S )N)_gzipcompressheap_profile)clients    M/opt/face_recognition/venv/lib/python3.12/site-packages/jax/lib/xla_client.py_heap_profiler
      s    	++-	..    FftType)zJjax.lib.xla_client.FftType was removed in JAX v0.6.0; use jax.lax.FftType.NPaddingType)zpjax.lib.xla_client.PaddingType was removed in JAX v0.6.0; this type is unused by JAX so there is no replacement.Ndtype_to_etype)z@dtype_to_etype was removed in JAX v0.6.0; use StableHLO instead.Nshape_from_pyval)zBshape_from_pyval was removed in JAX v0.6.0; use StableHLO instead.Nops)z:ops has been removed in JAX v0.6.0; use StableHLO instead.Nregister_custom_call_target)z}register_custom_call_target has been removed in JAX v0.6.0; use the JAX FFI instead (https://docs.jax.dev/en/latest/ffi.html)NPrimitiveType)zDPrimitiveType has been removed in JAX v0.6.0; use StableHLO instead.NShape)z<Shape has been removed in JAX v0.6.0; use StableHLO instead.N
XlaBuilder)zAXlaBuilder has been removed in JAX v0.6.0; use StableHLO instead.NXlaComputation)zEXlaComputation has been removed in JAX v0.6.0; use StableHLO instead.N	ArrayImpl)zSjax.lib.xla_client.ArrayImpl has been removed in JAX v0.6.0; use jax.Array instead.Nget_topology_for_deviceszjjax.lib.xla_client.get_topology_for_devices was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0r   z^jax.lib.xla_client.heap_profile was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0mlir_api_version)zbjax.lib.xla_client.mlir_api_version was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0:   ClientzXjax.lib.xla_client.Client was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0CompileOptionsz`jax.lib.xla_client.CompileOptions was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0DeviceAssignmentzbjax.lib.xla_client.DeviceAssignment was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0zWjax.lib.xla_client.Frame was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0z]jax.lib.xla_client.HloSharding was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0z\jax.lib.xla_client.OpSharding was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0z[jax.lib.xla_client.Traceback was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0)FrameHloSharding
OpSharding	Tracebackr   )deprecation_getattr)gzipr   jax._src.libr   _xcr
   r   r   r   r   r   r   r   r    _deprecationstyping_typingTYPE_CHECKINGr   r   jax._src.deprecationsr!   _deprecation_getattr__name____getattr__ r   r	   <module>r.      sm    */C C  C   !C(  )C2 
 3C: " $;CH  ICP  QCX  YC`  aCj  kCz 8 	$$!{CH - 	ICV  WCd % 	

eCr - 	sC@ 1 	ACR% 			- 	- 	( 	yCJ 
 99,::&%%.))
))%+~~*mm)  P$X}=+r   