Searched refs:embedding_column (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column_test.py | 50 embedding_column = tpu_fc.embedding_column(
52 self.assertIs(categorical_column, embedding_column.categorical_column)
53 self.assertEqual(embedding_dimension, embedding_column.dimension)
54 self.assertEqual('mean', embedding_column.combiner)
55 self.assertEqual('aaa_embedding', embedding_column.name)
56 self.assertEqual('aaa_embedding', embedding_column._var_scope_name)
57 self.assertEqual((embedding_dimension,), embedding_column._variable_shape)
60 }, embedding_column._parse_example_spec)
68 tpu_fc.embedding_column(categorical_column, dimension=embedding_dimension)
76 embedding_column = tpu_fc.embedding_column(
[all …]
|
| D | feature_column_v2_test.py | 89 embedding_column = tpu_fc.embedding_column_v2(
92 self.assertIs(categorical_column, embedding_column.categorical_column)
93 self.assertEqual(embedding_dimension, embedding_column.dimension)
94 self.assertEqual('mean', embedding_column.combiner)
95 self.assertEqual('aaa_embedding', embedding_column.name)
96 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
102 embedding_column = tpu_fc.embedding_column_v2(
107 self.assertIs(categorical_column, embedding_column.categorical_column)
108 self.assertEqual(embedding_dimension, embedding_column.dimension)
109 self.assertEqual('my_combiner', embedding_column.combiner)
[all …]
|
| D | feature_column.py | 55 def embedding_column(categorical_column, function
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | dense_features_test.py | 83 embedding_column = fc.embedding_column(
88 dense_features = df.DenseFeatures([embedding_column])
132 embedding_column = fc.embedding_column(
138 [embedding_column],
180 embedding_column = fc.embedding_column(
185 dense_features = df.DenseFeatures([embedding_column])
436 some_embedding_column = fc.embedding_column(
566 embedded_country = fc.embedding_column(
623 embedded_country = fc.embedding_column(
776 embedding_column = fc.embedding_column(
[all …]
|
| D | dense_features_v2_test.py | 78 embedding_column = fc.embedding_column(
83 dense_features = df.DenseFeatures([embedding_column])
123 embedding_column = fc.embedding_column(
128 dense_features = df.DenseFeatures([embedding_column])
416 some_embedding_column = fc.embedding_column(
545 embedded_country = fc.embedding_column(
602 embedded_country = fc.embedding_column(
|
| D | sequence_feature_column_integration_test.py | 64 fc.embedding_column(col, dimension=10),
73 fc.embedding_column(identity_col, dimension=10),
74 fc.embedding_column(bucket_col, dimension=20)
|
| D | sequence_feature_column_test.py | 124 embedding_column_a = fc.embedding_column(
130 embedding_column_b = fc.embedding_column(
165 embedding_column_a = fc.embedding_column(
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2_test.py | 1813 dragonfruit_embedding_column = fc.embedding_column(
2141 some_embedding_column = fc.embedding_column(
2181 some_embedding_column = fc.embedding_column(
2221 some_embedding_column = fc.embedding_column(
2261 some_embedding_column = fc.embedding_column(
2323 embedding_column = fc.embedding_column(
2328 input_layer = fc_old.InputLayer([embedding_column])
2368 embedding_column = fc.embedding_column(
2373 input_layer = fc_old.InputLayer([embedding_column])
2507 some_embedding_column = fc.embedding_column(
[all …]
|
| D | sequence_feature_column_test.py | 442 embedding_column = fc.embedding_column(
447 embedding_column, {'aaa': inputs})
480 embedding_column = fc.embedding_column(
484 embedding_column, {'aaa': inputs})
507 embedding_column = fc.embedding_column(
511 embedding_column, {'aaa': sparse_input})
|
| D | feature_column_test.py | 2636 embedding_column = fc._embedding_column(
2641 input_layer = InputLayer([embedding_column])
2681 embedding_column = fc._embedding_column(
2686 input_layer = InputLayer([embedding_column])
4364 embedding_column = fc._embedding_column(
4370 embedding_lookup = embedding_column._get_dense_tensor(
4401 embedding_column = fc._embedding_column(
4407 embedding_lookup = embedding_column._get_dense_tensor(
4780 embedding_column = fc._embedding_column(
4782 self.assertIs(categorical_column, embedding_column.categorical_column)
[all …]
|
| D | feature_column_v2.py | 491 def embedding_column(categorical_column, function
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.pbtxt | 36 name: "embedding_column"
|
| D | tensorflow.feature_column.pbtxt | 28 name: "embedding_column"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.feature_column.pbtxt | 28 name: "embedding_column"
|
| /external/tensorflow/tensorflow/python/keras/premade/ |
| D | wide_deep_test.py | 233 emb_column = fc.embedding_column(cat_column, dimension=5)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 1067 emb_vocab_column = fc.embedding_column(
1137 emb_vocab = fc.embedding_column(
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | saved_model_test.py | 610 embedding = fc.embedding_column(cat_vocab, dimension=8)
|
| /external/tensorflow/ |
| D | RELEASE.md | 1342 * Support `embedding_column` with variable-length input features for
|