tf.distribute.experimental_set_strategy

Set a tf.distribute.Strategy as current without with strategy.scope().

tf.distribute.experimental_set_strategy(strategy1) f() tf.distribute.experimental_set_strategy(strategy2) g() tf.distribute.experimental_set_strategy(None) h() 

is equivalent to:

with strategy1.scope(): f() with strategy2.scope(): g() h() 

In general, you should use the with strategy.scope(): API, but this alternative may be convenient in notebooks where you would have to put each cell in a with strategy.scope(): block.

strategy A tf.distribute.Strategy object or None.

RuntimeError If called inside a with strategy.scope():.