tf.distribute.experimental_set_strategy

<