Searched defs:get_embedding_table_size (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ | ||
H A D | feature_column.py | 315 def get_embedding_table_size(self): member in _TPUBaseEmbeddingColumn |
H A D | feature_column_v2.py | 475 def get_embedding_table_size(self): member in _TPUEmbeddingColumnV2 |