tfr.keras.strategy_utils.strategy_scope

Gets the strategy.scope() for training with strategy.

tfr.keras.strategy_utils.strategy_scope(
    strategy: Optional[tf.distribute.Strategy]
) -> Any

Example usage:

with strategy_scope(strategy):
  model = ...

strategy Distributed training strategy is used.

ContextManager for the distributed training strategy.