* Add `wait` to `tf.data.Dataset.load`. If `True`, for snapshots written
      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`

# START CROSS-COMPILE CONFIGS
# Set execution platform to Linux x86
# Note: Lot of the "host_" flags such as "host_cpu" and "host_crosstool_top"
# flags seem to be actually used to specify the execution platform details. It
# seems it is this way because these flags are old and predate the distinction
# between host and execution platform.
build:cross_compile_base --host_cpu=k8