tf.distribute.experimental.ValueContext