AVhddZddlZejdZejdZdZdZy)zJUtility to re-use variables created on first device on subsequent devices.Nz_\d/z_\d$cd|ytjd|}tjd|}|S)NVariable/)_VARIABLE_UNIQUIFYING_REGEXsub"_VARIABLE_UNIQUIFYING_REGEX_AT_END)names d/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/distribute/shared_variable_creator.py_canonicalize_variable_namer s3 \  $ ( (d 3$ + / /D 9$ +cZittsJfd}fd}dk(r|S|S)a{Construct the variable creator function for device `device_id`. Constructs custom variable creator functions for the given device. On first device (device_id == 0), it creates the variable using the `next_creator`, and stores it in the provided `shared_variable_store`. On all other devices (device_id > 0), it tries to re-use the variable already created with the same name. If no such variable exists, it throws an error. Additionally, we de-uniquify variable names before checking for matches. This helps re-use variables which are intended to be the same but have different names due to variable uniquification happening upstream. Since this might mean we may have multiple variables with the same canonical name, we store them in a list per canonical name and return them in the same order as well. Args: shared_variable_store: A dictionary that we will use to store variables created on the first device, and re-used by creators for other devices. device_id: Integer index of the device whose creator should be constructed. Returns: An appropriate creator function based on device_id. ct|jd}|di|}|vrg|<|j||S)z6Create the variable using `next_creator` and store it.r )r getappend) next_creatorkwargscanonical_namevshared_variable_stores r create_new_variablez$make_fn..create_new_variable>sO0F1CDNvA22.0N+.)003 Hr c~|jd}t|} j|d}||}|dz|<|S#ttf$rt dj |wxYw)z>Re-use existing variable from store with same name (in order).r rz>Tried to create variable {} with mismatching name on device {})rr KeyError IndexError RuntimeErrorformat) rrr rvariable_indexr device_idrvariable_scope_access_indexs r reuse_variablezmake_fn..reuse_variableHs ::f D06N #266~qIn  / ?a5CQ4F!.1 h j !#  J &y ! ###s #A+A/r) isinstanceint)rr rr"r!s`` @r make_fnr%"s:2!# Is ## # #$!^  r )__doc__recompilerr r r%rr r r)s:Q (bjj1%/RZZ%8";r