Searched refs:_EmbeddingColumn (Results 1 – 2 of 2) sorted by relevance
361 class _TPUEmbeddingColumn(_TPUBaseEmbeddingColumn, fc._EmbeddingColumn):381 return fc._EmbeddingColumn.__new__(456 return fc._EmbeddingColumn._get_dense_tensor(461 return fc._EmbeddingColumn._get_dense_tensor(480 return fc._EmbeddingColumn._get_sequence_dense_tensor(485 return fc._EmbeddingColumn._get_sequence_dense_tensor(
934 return _EmbeddingColumn(2476 class _EmbeddingColumn( class2495 return super(_EmbeddingColumn, cls).__new__(