AVh `dZddlmZddlmZedgdZedgdZy ) z5A module for interm merge-call related internal APIs.)distribute_lib) tf_exportz7__internal__.distribute.strategy_supports_no_merge_call)v1ctjsytj}|jj S)zFReturns if the current `Strategy` can operate in pure replica context.T)r has_strategy get_strategyextended_use_merge_call)strategys _/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/distribute/merge_call_interim.pystrategy_supports_no_merge_callr s9  $ $ &   ( ( *(    . . 0 00z0__internal__.distribute.interim.maybe_merge_callcztr ||g|i|Stjj|||S)aMaybe invoke `fn` via `merge_call` which may or may not be fulfilled. The caller of this utility function requests to invoke `fn` via `merge_call` at `tf.distribute.Strategy`'s best efforts. It is `tf.distribute`'s internal whether the request is honored, depending on the `Strategy`. See `tf.distribute.ReplicaContext.merge_call()` for more information. This is an interim API which is subject to removal and does not guarantee backward-compatibility. Args: fn: the function to be invoked. strategy: the `tf.distribute.Strategy` to call `fn` with. *args: the positional arguments to be passed in to `fn`. **kwargs: the keyword arguments to be passed in to `fn`. Returns: The return value of the `fn` call. )argskwargs)r rget_replica_context merge_call)fnr rrs r maybe_merge_callrsG*%& h ( ( ((  - - / : : f ; &&rN)__doc__tensorflow.python.distributer tensorflow.python.util.tf_exportrr rrr rsJ<76 DL1M1 ="E&F&r