BVh=dZddlmZddlmZehdZiZiZdZ dZ dZ edg Gd d e Z edg Gd d e Zy)zClass to represent a device.) tf_export) pywrap_tfe>CPUEPUGPUTPUCUSTOMc |dSt|SN)strinps W/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/device_spec.py_as_str_or_noner*#c(*c |dSt|Sr )intr s r_as_int_or_noner"rrc@|dvr|jSt|S)N)cpugpu)upperr) device_types r_as_device_str_or_noner&s&N"      %%r DeviceSpec)v1ceZdZdZdZ ddZdZedZdZ dZ d Z e d Z e d Ze d Ze d Ze dZdZedZeddZedZdZdZdZy) DeviceSpecV2aRepresents a (possibly partial) specification for a TensorFlow device. `DeviceSpec`s are used throughout TensorFlow to describe where state is stored and computations occur. Using `DeviceSpec` allows you to parse device spec strings to verify their validity, merge them or compose them programmatically. Example: ```python # Place the operations on device "GPU:0" in the "ps" job. device_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0) with tf.device(device_spec.to_string()): # Both my_var and squared_var will be placed on /job:ps/device:GPU:0. my_var = tf.Variable(..., name="my_variable") squared_var = tf.square(my_var) ``` With eager execution disabled (by default in TensorFlow 1.x and by calling disable_eager_execution() in TensorFlow 2.x), the following syntax can be used: ```python tf.compat.v1.disable_eager_execution() # Same as previous device_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0) # No need of .to_string() method. with tf.device(device_spec): my_var = tf.Variable(..., name="my_variable") squared_var = tf.square(my_var) ``` If a `DeviceSpec` is partially specified, it will be merged with other `DeviceSpec`s according to the scope in which it is defined. `DeviceSpec` components defined in inner scopes take precedence over those defined in outer scopes. ```python gpu0_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0) with tf.device(DeviceSpec(job="train").to_string()): with tf.device(gpu0_spec.to_string()): # Nodes created here will be assigned to /job:ps/device:GPU:0. with tf.device(DeviceSpec(device_type="GPU", device_index=1).to_string()): # Nodes created here will be assigned to /job:train/device:GPU:1. ``` A `DeviceSpec` consists of 5 components -- each of which is optionally specified: * Job: The job name. * Replica: The replica index. * Task: The task index. * Device type: The device type string (e.g. "CPU" or "GPU"). * Device index: The device index. )_job_replica_task _device_type _device_index _as_string_hashNczt||_t||_t||_t ||_t||_|j|j|j|j|j |j|_ t|j|_ y)afCreate a new `DeviceSpec` object. Args: job: string. Optional job name. replica: int. Optional replica index. task: int. Optional task index. device_type: Optional device type string (e.g. "CPU" or "GPU") device_index: int. Optional device index. If left unspecified, device represents 'any' device_index. jobreplicataskr device_indexN) rr rr!r"rr#r$_components_to_stringr%hash to_stringr&)selfr)r*r+rr,s r__init__zDeviceSpecV2.__init__ks  $DI#G,DM &DJ.{;D(6D00 II ZZ%%'' 1)DO dnn&'DJrc|jS)zReturn a string representation of this `DeviceSpec`. Returns: a string of the form /job:/replica:/task:/device::. )r%r0s rr/zDeviceSpecV2.to_strings ??rc*||j|S)a;Construct a `DeviceSpec` from a string. Args: spec: a string of the form /job:/replica:/task:/device:CPU: or /job:/replica:/task:/device:GPU: as cpu and gpu are mutually exclusive. All entries are optional. Returns: A DeviceSpec. )_string_to_components)clsspecs r from_stringzDeviceSpecV2.from_strings ))$/ 00rc$|j|S)aParse a `DeviceSpec` name into its components. **2.x behavior change**: In TensorFlow 1.x, this function mutates its own state and returns itself. In 2.x, DeviceSpecs are immutable, and this function will return a DeviceSpec which contains the spec. * Recommended: ``` # my_spec and my_updated_spec are unrelated. my_spec = tf.DeviceSpec.from_string("/CPU:0") my_updated_spec = tf.DeviceSpec.from_string("/GPU:0") with tf.device(my_updated_spec): ... ``` * Will work in 1.x and 2.x (though deprecated in 2.x): ``` my_spec = tf.DeviceSpec.from_string("/CPU:0") my_updated_spec = my_spec.parse_from_string("/GPU:0") with tf.device(my_updated_spec): ... ``` * Will NOT work in 2.x: ``` my_spec = tf.DeviceSpec.from_string("/CPU:0") my_spec.parse_from_string("/GPU:0") # <== Will not update my_spec with tf.device(my_spec): ... ``` In general, `DeviceSpec.from_string` should completely replace `DeviceSpec.parse_from_string`, and `DeviceSpec.replace` should completely replace setting attributes directly. Args: spec: an optional string of the form /job:/replica:/task:/device:CPU: or /job:/replica:/task:/device:GPU: as cpu and gpu are mutually exclusive. All entries are optional. Returns: The `DeviceSpec`. Raises: ValueError: if the spec was not valid. )r8r0r7s rparse_from_stringzDeviceSpecV2.parse_from_stringsj   D !!rc>|j|j|S)a!Returns a new DeviceSpec which incorporates `dev`. When combining specs, `dev` will take precedence over the current spec. So for instance: ``` first_spec = tf.DeviceSpec(job=0, device_type="CPU") second_spec = tf.DeviceSpec(device_type="GPU") combined_spec = first_spec.make_merged_spec(second_spec) ``` is equivalent to: ``` combined_spec = tf.DeviceSpec(job=0, device_type="GPU") ``` Args: dev: a `DeviceSpec` Returns: A new `DeviceSpec` which combines `self` and `dev` ) __class___get_combined_propertiesr0devs rmake_merged_speczDeviceSpecV2.make_merged_specs , 4>>488= >>rc t|j|j|j|j|j }|j ||jdi|S)aConvenience method for making a new DeviceSpec by overriding fields. For instance: ``` my_spec = DeviceSpec=(job="my_job", device="CPU") my_updated_spec = my_spec.replace(device="GPU") my_other_spec = my_spec.replace(device=None) ``` Args: **kwargs: This method takes the same args as the DeviceSpec constructor Returns: A DeviceSpec with the fields specified in kwargs overridden. r()dictr)r*r+rr,updater=)r0kwargs init_kwargss rreplacezDeviceSpecV2.replacesY  HH YY$$&& (Kv 4>> (K ((rc|jSr )r r3s rr)zDeviceSpecV2.job s 99rc|jSr )r!r3s rr*zDeviceSpecV2.replicas ==rc|jSr )r"r3s rr+zDeviceSpecV2.tasks ::rc|jSr )r#r3s rrzDeviceSpecV2.device_types   rc|jSr )r$r3s rr,zDeviceSpecV2.device_indexs   rcf|j |jn |j|j |jn |j|j |jn |j|j |jn |j|j |jfS|jfS)a*Combine the current DeviceSpec with another DeviceSpec. The combination of DeviceSpecs is will give priority to dev. Args: dev: a `DeviceSpec` Returns: A tuple of (job, replica, task, device_type, device_index) which represents the combination of self and dev. r(r?s rr>z%DeviceSpecV2._get_combined_propertiess77&DHH{{. DLLHH(dii??6DO>O  rcti}tj}|D]3}|j|j j dd5|t z}|S)N:)setrTF_ListPluggablePhysicalDevicesadddecodesplit_VALID_DEVICE_TYPES)valid_device_typesphysical_devicesdevices r_get_valid_device_typesz$DeviceSpecV2._get_valid_device_types2s`R!AAC"<V]]_2237:;<+.AA rctj|}||S|}d\}}}}}|xsd}|jdDcgc]}|jd} }tj } | D]} t | } | s| dk(r| ddk(r| d}%| dk(r| dd k(r| d}8| dk(r| dd k(r| d}K| dk(s| dk(rW| dj | vrB|td |d | dj }| dk(s| dd k7st| d}| dk(r6| ddk(r.|td |d | d}| dd k7st| d}| s| ddk7std| dd|d|||||f} | t|<| Scc}w)aEStateless portion of device spec string parsing. Args: spec: An optional string specifying a device specification. Returns: The parsed components of `spec`. Note that the result of this function must go through attribute setters of DeviceSpec, and should therefore NOT be used directly. NNNNN/rPrr)rQr*r+zEMultiple device types are not allowed while parsing the device spec: .*rZzUnknown attribute 'z1' is encountered while parsing the device spec: 'z'.) _STRING_TO_COMPONENTS_CACHEgetrVrr[lenr ValueErrorr)r7 cached_resultraw_specr)r*r+rr,xsplitsrXylyoutputs rr5z"DeviceSpecV2._string_to_components;s 033D9M H4P1C$ \ :2D$(JJsO 4qaggcl 4F 4%==? H q6b 7qtu}!# 1W1*aD' 1W11$Qw"'! 8J(J  $ ??CfAGH H! + 1W1qt9L 1W1)  $ ??CfAGH H!+ qTS[qt9L AaDBJ017>>BV2GH H3H87D+| Stateless portion of `to_string` (separated to allow caching).z/job:z /replica:z/task:rbz/device:rPr^)_COMPONENTS_TO_STRING_CACHEreappendr join) r)r*r+rr,keyrhrndevice_index_strings rr-z"DeviceSpecV2._components_to_stringqs {L 9C/33C8M  F  mmGcM" mmK#g,./  mmHs4y()  !!,/ mm{4GHI WWV_F'-$ Mrctt||jxr!|j|jk(S)aFChecks if the `other` DeviceSpec is same as the current instance, eg have same value for all the internal fields. Args: other: Another DeviceSpec Returns: Return `True` if `other` is also a DeviceSpec instance and has same value as the current instance. Return `False` otherwise. ) isinstancer=r/)r0others r__eq__zDeviceSpecV2.__eq__s2 udnn - 2 NN  1 13rc|jSr )r&r3s r__hash__zDeviceSpecV2.__hash__s ::rc d|jd|jd|jd|jd|jd S)Nzr(r3s r__repr__zDeviceSpecV2.__repr__sN 488*Jt||nGDII;O''(8I8I7J" NOrr]r )__name__ __module__ __qualname____doc__ __slots__r1r/ classmethodr8r;rArHpropertyr)r*r+rr,r> staticmethodr[r5r-rxrzr|rCrrrr.s6p&) (: 1 15"n?0)6          (33j23 OrrceZdZejZej Zej jdZejjdZejjdZ ejjdZ ejjdZ dZ dZ dZd Zejje _ejje_y ) DeviceSpecV1cBt||_d\|_|_yN)NN)rr r%r&)r0r)s rr)zDeviceSpecV1.jobs$DI",DOTZrcBt||_d\|_|_yr)rr!r%r&)r0r*s rr*zDeviceSpecV1.replicas#G,DM",DOTZrcBt||_d\|_|_yr)rr"r%r&)r0r+s rr+zDeviceSpecV1.tasks &DJ",DOTZrcBt||_d\|_|_yr)rr#r%r&)r0rs rrzDeviceSpecV1.device_types.{;D",DOTZrcBt||_d\|_|_yr)rr$r%r&)r0r,s rr,zDeviceSpecV1.device_indexs(6D",DOTZrcn|jt|j|_|jSr )r&r.r/r3s rrzzDeviceSpecV1.__hash__s) zz()dj ::rc|jM|j|j|j|j|j |j |_|jS)Nr()r%r-r)r*r+rr,r3s rr/zDeviceSpecV1.to_stringsV 22hh,,yy&&(( 3*do ??rcf|j|\|_|_|_|_|_|Sr )r5r)r*r+rr,r:s rr;zDeviceSpecV1.parse_from_strings144T:TXt|TY(8  Krcd|j|\|_|_|_|_|_y)zMerge the properties of "dev" into this `DeviceSpec`. Note: Will be removed in TensorFlow 2.x since DeviceSpecs will become immutable. Args: dev: a `DeviceSpec`. N)r>r)r*r+rr,r?s r merge_fromzDeviceSpecV1.merge_froms.77<TXt|TY(8 rN)r}r~rrrrr)setterr*r+rr,rzr/r;rrCrrrrs  '$$)------""-#-##-$-  =#,,44)*<<DDrrN)r tensorflow.python.util.tf_exportrtensorflow.pythonr frozensetrWrdrprrrobjectrrrCrrrs#6( FG ! ++&