experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

API changes to allow TPU embedding table sizes that do not fit in 32-bit

Browse Source 
integers. Larger tables do not work correctly yet; this is just changing some
of the external APIs.

PiperOrigin-RevId: 351470612
Change-Id: I6bae16c0a8573e507fb065b9334410063759e6f9
This commit is contained in:
A. Unique TensorFlower 2021-01-12 16:14:57 -08:00 committed by TensorFlower Gardener
parent c46ef8d3e4
commit 3fc7a34c77
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/protobuf/tpu/tpu_embedding_configuration.proto
View File

@ -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.