tf.compat.v1.train.replica_device_setter