§ „Wêh ãó–—ddlmZmZmZmZmZmZmZmZm Z m Z m Z m Z ddl mZddlmZmZmZddlmZmZmZddlmZmZmZmZmZmZmZdS)é) ÚIndexÚ MapTracerÚ MeshAxisNameÚMeshComputationÚMeshExecutableÚPmapExecutableÚglobal_aval_to_result_handlerÚglobal_avals_to_results_handlerÚglobal_result_handlersÚparallel_callableÚ shard_argsÚ xla_pmap_p)Úthread_resources)Úare_op_shardings_equalÚis_op_sharding_replicatedÚop_sharding_to_indices)Ú ArrayMappingÚ UNSPECIFIEDÚarray_mapping_to_axis_resources)ÚChunkedÚ NoShardingÚ ReplicatedÚ ShardedAxisÚ ShardingSpecÚ UnstackedÚspec_to_indicesN) Újax._src.interpreters.pxlarrrrrrr r r r r rÚ jax._src.meshrÚjax._src.op_shardingsrrrÚjax._src.sharding_implsrrÚ _UNSPECIFIEDrÚjax._src.sharding_specsrrrrrrr©óú[/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/jax/interpreters/pxla.pyúr&swðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ððððððððððððððððð ðððððððððð ðððððððððððððððððððr$