Searched refs:_wide_embedding_lookup_arguments (Results 1 – 3 of 3) sorted by relevance
266 def _wide_embedding_lookup_arguments(self, input_tensor): member in _FeatureColumn434 def _wide_embedding_lookup_arguments(self, input_tensor): member in _SparseColumn830 def _wide_embedding_lookup_arguments(self, input_tensor): member in _WeightedSparseColumn1165 def _wide_embedding_lookup_arguments(self, input_tensor): member in _EmbeddingColumn2148 def _wide_embedding_lookup_arguments(self, input_tensor): member in _BucketizedColumn2357 def _wide_embedding_lookup_arguments(self, input_tensor): member in _CrossedColumn
407 … column._wide_embedding_lookup_arguments(transformed_tensor)) # pylint: disable=protected-access497 embedding_lookup_arguments = column._wide_embedding_lookup_arguments(
787 self.assertEqual(1, sc._wide_embedding_lookup_arguments(None).vocab_size)