tf.distribute.StrategyExtended

Additional APIs for algorithms that need to be distribution-aware.