Searched refs:_num_buckets (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column.py | 548 … shape=(self._feature_column._num_buckets, self._units), # pylint: disable=protected-access
903 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
1921 def _num_buckets(self): member in _CategoricalColumn
2005 shape=(column._num_buckets, units), # pylint: disable=protected-access
2358 def _num_buckets(self): member in _BucketizedColumn
2546 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
2690 def _num_buckets(self): member in _HashedCategoricalColumn
2743 def _num_buckets(self): member in _VocabularyFileCategoricalColumn
2794 def _num_buckets(self): member in _VocabularyListCategoricalColumn
2857 def _num_buckets(self): member in _IdentityCategoricalColumn
[all …]
|
| D | feature_column_v2.py | 1049 num_buckets = c0._num_buckets # pylint: disable=protected-access
1067 if num_buckets != c._num_buckets: # pylint: disable=protected-access
1072 c0, num_buckets, c, c._num_buckets)) # pylint: disable=protected-access
2918 def _num_buckets(self): member in BucketizedColumn
3038 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
3081 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
3235 self._num_buckets = num_buckets
3248 embedding_shape = (self._num_buckets, self._dimension)
3483 def _num_buckets(self): member in HashedCategoricalColumn
3594 def _num_buckets(self): member in VocabularyFileCategoricalColumn
[all …]
|
| D | feature_column_test.py | 443 self.assertEqual(6, b._num_buckets)
707 self.assertEqual(10, column._num_buckets)
870 self.assertEqual(4, wire_column._num_buckets)
893 self.assertEqual(4, wire_column._num_buckets)
998 self.assertEqual(15, crossed._num_buckets)
1177 def _num_buckets(self): member in CrossedColumnTest.test_linear_model_with_weights._TestColumnWithWeights
1263 def _num_buckets(self): member in CrossedColumnTest.test_keras_linear_model_with_weights._TestColumnWithWeights
1455 def _num_buckets(self): member in LinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
2138 def _num_buckets(self): member in _LinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
3385 self.assertEqual(3, column._num_buckets)
[all …]
|
| D | feature_column_v2_test.py | 1475 def _num_buckets(self): member in CrossedColumnTest.test_old_linear_model_with_weights._TestColumnWithWeights
2452 def _num_buckets(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column.py | 91 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
248 return (self.categorical_column._num_buckets, self.dimension)
346 return (self.categorical_column._num_buckets, self.dimension)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | feature_column.py | 488 def _num_buckets(self): member in _SparseColumn
864 def _num_buckets(self): member in _WeightedSparseColumn
2174 def _num_buckets(self): member in _BucketizedColumn
2404 def _num_buckets(self): member in _CrossedColumn
|