o ?e @sHdZddlmZddlmZedgdddZedgdd d Zd S) z5A module for interm merge-call related internal APIs.)distribute_lib) tf_exportz7__internal__.distribute.strategy_supports_no_merge_call)v1cCs tsdSt}|j S)zFReturns if the current `Strategy` can operate in pure replica context.T)rZ has_strategyZ get_strategyextendedZ_use_merge_call)strategyrp/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/distribute/merge_call_interim.pystrategy_supports_no_merge_calls r z0__internal__.distribute.interim.maybe_merge_callcOs0tr||g|Ri|Stj|||dS)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 rZget_replica_contextZ merge_call)fnrr r rrrmaybe_merge_calls r N)__doc__Ztensorflow.python.distributerZ tensorflow.python.util.tf_exportrr r rrrrs