Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py91 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
95 embedding_shape=embedding_shape,
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py903 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
907 embedding_shape=embedding_shape,
1682 embedding_shape, argument
1704 self._embedding_shape = embedding_shape
2546 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
2558 if embedding_weights.get_shape() != embedding_shape:
2567 embedding_weights.get_shape(), embedding_shape))
2571 shape=embedding_shape,
Dfeature_column_v2.py3038 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
3042 shape=embedding_shape,
3081 …embedding_shape = (self.categorical_column._num_buckets, self.dimension) # pylint: disable=protec…
3087 shape=embedding_shape,
3248 embedding_shape = (self._num_buckets, self._dimension)
3251 shape=embedding_shape,
Dfeature_column_test.py5216 embedding_shape = (vocabulary_size, embedding_dimension)
5217 zeros_embedding_values = np.zeros(embedding_shape)
5219 self.assertAllEqual(embedding_shape, shape)
5296 embedding_shape = (vocabulary_size, embedding_dimension)
5297 zeros_embedding_values = np.zeros(embedding_shape)
5300 self.assertAllEqual(embedding_shape, shape)
5927 embedding_shape = (vocabulary_size, embedding_dimension)
5928 zeros_embedding_values = np.zeros(embedding_shape)
5930 self.assertAllEqual(embedding_shape, shape)
6020 embedding_shape = (vocabulary_size, embedding_dimension)
[all …]
Dfeature_column_v2_test.py6737 embedding_shape = (vocabulary_size, embedding_dimension)
6738 zeros_embedding_values = np.zeros(embedding_shape)
6741 self.assertAllEqual(embedding_shape, shape)
7017 embedding_shape = (vocabulary_size, embedding_dimension)
7018 zeros_embedding_values = np.zeros(embedding_shape)
7021 self.assertAllEqual(embedding_shape, shape)
7099 embedding_shape = (vocabulary_size, embedding_dimension)
7100 zeros_embedding_values = np.zeros(embedding_shape)
7103 self.assertAllEqual(embedding_shape, shape)
7626 embedding_shape = (vocabulary_size, embedding_dimension)
[all …]