diff --git a/tensorflow/core/protobuf/tpu/tpu_embedding_configuration.proto b/tensorflow/core/protobuf/tpu/tpu_embedding_configuration.proto
index 7e321158091..a42dac3828a 100644
--- a/tensorflow/core/protobuf/tpu/tpu_embedding_configuration.proto
+++ b/tensorflow/core/protobuf/tpu/tpu_embedding_configuration.proto
@@ -11,7 +11,7 @@ message TPUEmbeddingConfiguration {
     // Name of the table.
     string name = 1;
     // Size of the vocabulary (i.e., number of rows) in the table.
-    int32 vocabulary_size = 2;
+    int64 vocabulary_size = 2;
     // The embedding dimension (i.e., the width of the embedding table).
     int32 dimension = 3;
     // Number of features mapped to this table.