Searched defs:get_embedding_table_size (Results 1 – 1 of 1) sorted by relevance
176 def get_embedding_table_size(self): member in _TPUBaseEmbeddingColumn246 def get_embedding_table_size(self): member in _TPUEmbeddingColumn344 def get_embedding_table_size(self): member in _TPUSharedEmbeddingColumn