Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py130 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
134 embedding_shape=embedding_shape,
Dfeature_column_v2.py1009 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
1015 shape=embedding_shape,
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py923 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
927 embedding_shape=embedding_shape,
1703 embedding_shape, argument
1725 self._embedding_shape = embedding_shape
2659 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
2671 if embedding_weights.get_shape() != embedding_shape:
2680 embedding_weights.get_shape(), embedding_shape))
2684 shape=embedding_shape,
Dfeature_column_v2.py2899 embedding_shape = (num_buckets, self.dimension)
2903 shape=embedding_shape,
2947 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
2953 shape=embedding_shape,
3125 embedding_shape = (self._num_buckets, self._dimension)
3128 shape=embedding_shape,
Dfeature_column_test.py5245 embedding_shape = (vocabulary_size, embedding_dimension)
5246 zeros_embedding_values = np.zeros(embedding_shape)
5248 self.assertAllEqual(embedding_shape, shape)
5324 embedding_shape = (vocabulary_size, embedding_dimension)
5325 zeros_embedding_values = np.zeros(embedding_shape)
5328 self.assertAllEqual(embedding_shape, shape)
6012 embedding_shape = (vocabulary_size, embedding_dimension)
6013 zeros_embedding_values = np.zeros(embedding_shape)
6016 self.assertAllEqual(embedding_shape, shape)
6105 embedding_shape = (vocabulary_size, embedding_dimension)
[all …]
Dfeature_column_v2_test.py5203 embedding_shape = (vocabulary_size, embedding_dimension)
5204 zeros_embedding_values = np.zeros(embedding_shape)
5207 self.assertAllEqual(embedding_shape, shape)
5285 embedding_shape = (vocabulary_size, embedding_dimension)
5286 zeros_embedding_values = np.zeros(embedding_shape)
5289 self.assertAllEqual(embedding_shape, shape)