tfm.optimization.ema_optimizer.maybe_merge_call

Maybe invoke fn via merge_call which may or may not be fulfilled.

tfm.optimization.ema_optimizer.maybe_merge_call(
    fn, strategy, *args, **kwargs
)

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 adapted from tensorflow/python/distribute/merge_call_interim.py.

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-02-02 UTC.

Args