Searched refs:EmbeddingColumn (Results 1 – 4 of 4) sorted by relevance
420 class _TPUEmbeddingColumnV2(_TPUBaseEmbeddingColumn, fc_lib.EmbeddingColumn):433 return fc_lib.EmbeddingColumn.__new__(512 return fc_lib.EmbeddingColumn._get_dense_tensor(517 return fc_lib.EmbeddingColumn._get_dense_tensor(534 return fc_lib.EmbeddingColumn.create_state(548 return fc_lib.EmbeddingColumn.get_dense_tensor(553 return fc_lib.EmbeddingColumn.get_dense_tensor(567 return fc_lib.EmbeddingColumn._get_sequence_dense_tensor(572 return fc_lib.EmbeddingColumn._get_sequence_dense_tensor(594 return fc_lib.EmbeddingColumn.get_sequence_dense_tensor([all …]
31 fc_lib.BucketizedColumn, fc_lib.CrossedColumn, fc_lib.EmbeddingColumn,
592 return EmbeddingColumn(2817 class EmbeddingColumn( class2839 return super(EmbeddingColumn, cls).__new__(
5390 new_embedding_column = fc.EmbeddingColumn.from_config(5397 new_embedding_column = fc.EmbeddingColumn.from_config(5450 new_embedding_column = fc.EmbeddingColumn.from_config(5456 new_embedding_column = fc.EmbeddingColumn.from_config(