https://www.tensorflow.org/api_docs/python/tf/tpu/experimental/embedding/QuantizationConfig?authuser=1
tf.tpu.experimental.embedding.QuantizationConfig | TensorFlow v2.16.1
Settings for simulated quantization of the tpu embedding table.
tftpuexperimentalembeddingtensorflow