with `distributed_save`, it reads the snapshot while it is being written.
      For snapshots written with regular `save`, it waits for the snapshot until
      it's finished. The default is `False` for backward compatibility. Users of
      `distributed_save` are recommended to set it to `True`.

* `tf.tpu.experimental.embedding.TPUEmbeddingV2`