Searched refs:get_weight_key_name (Results 1 – 1 of 1) sorted by relevance
184 def get_weight_key_name(self): member in _TPUBaseEmbeddingColumn256 def get_weight_key_name(self): member in _TPUEmbeddingColumn354 def get_weight_key_name(self): member in _TPUSharedEmbeddingColumn