VpfPGddZeZGddeZdS)ceZdZdZdS) _UnconstrainedPartitionSingletoncdS)N UNCONSTRAINEDselfs W/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/partition_spec.py__repr__z)_UnconstrainedPartitionSingleton.__repr__s ?N)__name__ __module__ __qualname__r rr r rrs#r rc.eZdZdZeZdZdZdZdZ dS) PartitionSpecahTuple describing how to partition an array across a mesh of devices. Each element is either ``None``, a string, or a tuple of strings. See the documentation of :class:`jax.sharding.NamedSharding` for more details. This class exists so JAX's pytree utilities can distinguish a partition specifications from tuples that should be treated as pytrees. cdSNr)r partitionss r __init__zPartitionSpec.__init__(sDr cBtt|Sr)tuple__new__r)clsrs r rzPartitionSpec.__new__+s == 3 33r c<dt|zS)NzPartitionSpec%s)rr rs r r zPartitionSpec.__repr__.s u~~d33 33r c.tt|fSr)rrrs r __reduce__zPartitionSpec.__reduce__1s 5;; ''r N) r r r__doc___UNCONSTRAINED_PARTITIONrrrr rrr r rrsa+-   444444(((((r rN)rrrrrr r rsr<;==(((((E(((((r