Home
last modified time | relevance | path

Searched defs:get_embedding_table_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dfeature_column.py315 def get_embedding_table_size(self): member in _TPUBaseEmbeddingColumn
418 def get_embedding_table_size(self): member in _TPUEmbeddingColumn
557 def get_embedding_table_size(self): member in _TPUSharedEmbeddingColumn
H A Dfeature_column_v2.py475 def get_embedding_table_size(self): member in _TPUEmbeddingColumnV2
669 def get_embedding_table_size(self): member in _TPUSharedEmbeddingColumnV2