Searched refs:embedding_dimension (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column_test.py | 49 embedding_dimension = 2
51 categorical_column, dimension=embedding_dimension)
53 self.assertEqual(embedding_dimension, embedding_column.dimension)
57 self.assertEqual((embedding_dimension,), embedding_column._variable_shape)
65 embedding_dimension = 2
68 dimension=embedding_dimension,
72 self.assertEqual(embedding_dimension, embedding_column.dimension)
76 self.assertEqual((embedding_dimension,), embedding_column._variable_shape)
95 embedding_dimension = 2
103 self.assertAllEqual((vocabulary_size, embedding_dimension), shape)
[all …]
|
| D | tpu_embedding.py | 527 embedding_dimension=self._table_to_config_dict[table].dimension,
822 embedding_dimension=table_config.dimension,
902 embedding_dimension=table_config.dimension,
910 embedding_dimension=table_config.dimension,
1088 embedding_dimension, argument
1100 shape=(vocabulary_size, embedding_dimension),
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_test.py | 2653 embedding_dimension = 2
2666 dimension=embedding_dimension,
2697 embedding_dimension = 2
2711 dimension=embedding_dimension,
3056 embedding_dimension = 2
3059 dimension=embedding_dimension)
3090 embedding_dimension = 2
3093 dimension=embedding_dimension)
4787 embedding_dimension = 2
4789 categorical_column, dimension=embedding_dimension)
[all …]
|
| D | sequence_feature_column_test.py | 106 def _get_initializer(embedding_dimension, embedding_values): argument
108 self.assertAllEqual((vocabulary_size, embedding_dimension), shape)
185 embedding_dimension = 2
192 def _get_initializer(embedding_dimension, embedding_values): argument
195 self.assertAllEqual((vocabulary_size, embedding_dimension), shape)
217 dimension=embedding_dimension,
218 initializer=_get_initializer(embedding_dimension, embedding_values))
960 embedding_dimension = 2
967 self.assertAllEqual((vocabulary_size, embedding_dimension), shape)
975 categorical_column, dimension=embedding_dimension,
[all …]
|
| D | feature_column_v2_test.py | 3194 embedding_dimension = 2
3208 dimension=embedding_dimension,
3239 embedding_dimension = 2
3253 dimension=embedding_dimension,
3534 embedding_dimension = 2
3537 dimension=embedding_dimension)
3568 embedding_dimension = 2
3571 dimension=embedding_dimension)
3818 embedding_dimension = 2
3832 dimension=embedding_dimension,
[all …]
|
| /external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/ |
| D | sequence_feature_column_test.py | 103 def _get_initializer(embedding_dimension, embedding_values): argument
105 self.assertAllEqual((vocabulary_size, embedding_dimension), shape)
181 embedding_dimension = 2
188 def _get_initializer(embedding_dimension, embedding_values): argument
191 self.assertAllEqual((vocabulary_size, embedding_dimension), shape)
213 dimension=embedding_dimension,
214 initializer=_get_initializer(embedding_dimension, embedding_values))
908 embedding_dimension = 2
915 self.assertAllEqual((vocabulary_size, embedding_dimension), shape)
924 dimension=embedding_dimension,
[all …]
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | state_saving_rnn_estimator_test.py | 153 embedding_dimension = 8
181 dimension=embedding_dimension,
194 embedding_dimension = 8
226 dimension=embedding_dimension,
249 embedding_dimension = 8
276 dimension=embedding_dimension,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | feature_column_ops_test.py | 1442 embedding_dimension = 5
1452 expected_input_shape = np.array([4, 3, embedding_dimension])
1457 embedding_dimension)
1471 embedding_dimension = 5
1481 expected_input_shape = np.array([4, 3, embedding_dimension])
1486 embedding_dimension)
1500 embedding_dimension = 3
1513 hashed_ids_column, embedding_dimension, combiner="sum")
1526 expected_input_shape = [4, 3, embedding_dimension]
1531 expected_gradient_values = sorted([0., 3., 2., 1.] * embedding_dimension)
[all …]
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals.py | 462 embedding_dimension, argument
536 embedding_dimension,
|