o ?e=@sdZddlmZddlmZehdZiZiZddZ ddZ d d Z ed gd Gd dde Z ed gd Gddde ZdS)zClass to represent a device.) tf_export) pywrap_tfe>ZTPUZCUSTOMZEPUZGPUZCPUcC|durdSt|SN)strinpr h/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/framework/device_spec.py_as_str_or_noner cCrr)intrr r r _as_int_or_none"r rcCs|dvr|St|S)N)cpuZgpu)upperr ) device_typer r r _as_device_str_or_none&srZ DeviceSpec)v1c@seZdZdZdZ     d(ddZddZedd Zd d Z d d Z ddZ e ddZ e ddZe ddZe ddZe ddZddZeddZed)ddZed d!Zd"d#Zd$d%Zd&d'ZdS)* 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_hashNcCsdt||_t||_t||_t||_t||_|j|j|j|j|j|jd|_ t | |_ dS)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) r rrrrrrr_components_to_stringrhash to_stringr)selfrrrrr r r r __init__ks     zDeviceSpecV2.__init__cC|jS)zReturn a string representation of this `DeviceSpec`. Returns: a string of the form /job:/replica:/task:/device::. )rr$r r r r#szDeviceSpecV2.to_stringcCs|||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)clsspecr r r from_strings zDeviceSpecV2.from_stringcCs ||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. )r+r$r*r r r parse_from_strings 5zDeviceSpecV2.parse_from_stringcCs|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_propertiesr$devr r r make_merged_specszDeviceSpecV2.make_merged_speccKs6t|j|j|j|j|jd}|||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. rNr )dictrrrrr updater.)r$kwargsZ init_kwargsr r r replaces zDeviceSpecV2.replacecCr&r)rr'r r r r zDeviceSpecV2.jobcCr&r)rr'r r r rr7zDeviceSpecV2.replicacCr&r)rr'r r r rr7zDeviceSpecV2.taskcCr&r)rr'r r r rr7zDeviceSpecV2.device_typecCr&r)rr'r r r r r7zDeviceSpecV2.device_indexcCsj|jdur|jn|j|jdur|jn|j|jdur|jn|j|jdur&|jn|j|jdur1|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. Nrr0r r r r/s z%DeviceSpecV2._get_combined_propertiescCs>ti}t}|D]}||ddq |tB}|S)N:)setrZTF_ListPluggablePhysicalDevicesadddecodesplit_VALID_DEVICE_TYPES)valid_device_typesZphysical_devicesdevicer r r _get_valid_device_types2s z$DeviceSpecV2._get_valid_device_typesc Cst|}|dur |S|}d\}}}}}|pd}dd|dD}t} |D]} t| } | r| dkr?| dd kr?| d }q(| dkrN| dd krN| d }q(| dkr]| dd kr]| d }q(| d kse| dkr| d| vr|durytd |d| d}| dkr| d dkrt| d }q(| dkr| ddkr|durtd |d| d }| ddkrt| d}q(| r| ddkrtd| dd|dq(|||||f} | t|<| S)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. NNNNNNcSsg|]}|dqS)r8)r=).0xr r r Osz6DeviceSpecV2._string_to_components../rrr9rrzEMultiple device types are not allowed while parsing the device spec: .*r@zUnknown attribute 'z1' is encountered while parsing the device spec: 'z'.) _STRING_TO_COMPONENTS_CACHEgetr=rrAlenr ValueErrorr ) r* cached_resultZraw_specrrrrr Zsplitsr?yZlyoutputr r r r(;sV           z"DeviceSpecV2._string_to_componentsc Cs|||||f}t|}|dur|Sg}|dur|d||dur,|dt||dur9|dt||durPd}|durGt|}|d||fd|}|t|<|S)z>Stateless portion of `to_string` (separated to allow caching).Nz/job:z /replica:z/task:rJz /device:%s:%srC)_COMPONENTS_TO_STRING_CACHErMappendrjoin) rrrrr keyrPrRZdevice_index_stringr r r r!qs&  z"DeviceSpecV2._components_to_stringcCst||jo ||kS)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#)r$otherr r r __eq__s zDeviceSpecV2.__eq__cCr&r)rr'r r r __hash__szDeviceSpecV2.__hash__c Cs.d|jd|jd|jd|jd|jd S)Nzrr'r r r __repr__s zDeviceSpecV2.__repr__rBr)__name__ __module__ __qualname____doc__ __slots__r%r# classmethodr+r-r2r6propertyrrrrr r/ staticmethodrAr(r!rYrZr[r r r r r.sF8  7       5  rc@seZdZejZejZejjddZejjddZej jddZ ej jddZ ej jd d Z d d Z d dZ ddZddZej je _ejje_dS) DeviceSpecV1cCt||_d\|_|_dSN)NN)r rrr)r$rr r r r zDeviceSpecV1.jobcCrerf)rrrr)r$rr r r rrgzDeviceSpecV1.replicacCrerf)rrrr)r$rr r r rrgzDeviceSpecV1.taskcCrerf)rrrr)r$rr r r rrgzDeviceSpecV1.device_typecCrerf)rrrr)r$r r r r r rgzDeviceSpecV1.device_indexcCs|jdur t||_|jSr)rr"r#r'r r r rZs zDeviceSpecV1.__hash__cCs0|jdur|j|j|j|j|j|jd|_|jS)Nr)rr!rrrrr r'r r r r#s zDeviceSpecV1.to_stringcCs"||\|_|_|_|_|_|Sr)r(rrrrr r,r r r r-szDeviceSpecV1.parse_from_stringcCs"||\|_|_|_|_|_dS)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/rrrrr r0r r r merge_froms zDeviceSpecV1.merge_fromN)r\r]r^rr_r`rsetterrrrr rZr#r-rhr r r r rds&       rdN)r_Z tensorflow.python.util.tf_exportrZtensorflow.pythonr frozensetr>rLrSr rrobjectrrdr r r r s     w