tf.tpu.experimental.embedding.serving_embedding_lookup