Home
last modified time | relevance | path

Searched refs:_num_buckets (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py127 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
251 num_buckets = sorted_columns[0]._num_buckets # pylint: disable=protected-access
254 if num_buckets != c._num_buckets: # pylint: disable=protected-access
259 … sorted_columns[0], num_buckets, c, c._num_buckets)) # pylint: disable=protected-access
409 return (self.categorical_column._num_buckets, self.dimension)
544 return (self.categorical_column._num_buckets, self.dimension)
Dfeature_column_v2.py336 num_buckets = sorted_columns[0]._num_buckets # pylint: disable=protected-access
339 if num_buckets != c._num_buckets: # pylint: disable=protected-access
344 … sorted_columns[0], num_buckets, c, c._num_buckets)) # pylint: disable=protected-access
455 return (self.categorical_column._num_buckets, self.dimension)
642 return (self.categorical_column._num_buckets,
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py554 … shape=(self._feature_column._num_buckets, self._units), # pylint: disable=protected-access
918 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
1979 def _num_buckets(self): member in _CategoricalColumn
2063 shape=(column._num_buckets, units), # pylint: disable=protected-access
2416 def _num_buckets(self): member in _BucketizedColumn
2634 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
2784 def _num_buckets(self): member in _HashedCategoricalColumn
2837 def _num_buckets(self): member in _VocabularyFileCategoricalColumn
2888 def _num_buckets(self): member in _VocabularyListCategoricalColumn
2941 def _num_buckets(self): member in _IdentityCategoricalColumn
[all …]
Dfeature_column_v2.py1094 num_buckets = c0._num_buckets # pylint: disable=protected-access
1114 if num_buckets != c._num_buckets: # pylint: disable=protected-access
1119 c0, num_buckets, c, c._num_buckets)) # pylint: disable=protected-access
3036 def _num_buckets(self): member in BucketizedColumn
3183 … else self.categorical_column._num_buckets) # pylint: disable=protected-access
3234 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
3393 self._num_buckets = num_buckets
3408 embedding_shape = (self._num_buckets, self._dimension)
3655 def _num_buckets(self): member in HashedCategoricalColumn
3769 def _num_buckets(self): member in VocabularyFileCategoricalColumn
[all …]
Dfeature_column_test.py443 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 …]
Dfeature_column_v2_test.py1527 def _num_buckets(self): member in CrossedColumnTest.test_old_linear_model_with_weights._TestColumnWithWeights
2465 def _num_buckets(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn