| /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)
66 embedding_column = tpu_fc.embedding_column(
71 self.assertIs(categorical_column, embedding_column.categorical_column)
[all …]
|
| D | feature_column.py | 43 def embedding_column(categorical_column, function
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2_test.py | 2022 some_embedding_column = fc.embedding_column(
2702 dragonfruit_embedding_column = fc.embedding_column(
3023 some_embedding_column = fc.embedding_column(
3063 some_embedding_column = fc.embedding_column(
3103 some_embedding_column = fc.embedding_column(
3143 some_embedding_column = fc.embedding_column(
3206 embedding_column = fc.embedding_column(
3211 dense_features = fc.DenseFeatures([embedding_column])
3251 embedding_column = fc.embedding_column(
3256 dense_features = fc.DenseFeatures([embedding_column])
[all …]
|
| D | sequence_feature_column_test.py | 116 embedding_column_a = fc.embedding_column(
122 embedding_column_b = fc.embedding_column(
158 embedding_column_a = fc.embedding_column(
651 embedding_column_a = fc.embedding_column(
974 embedding_column = fc.embedding_column(
979 embedding_column, {'aaa': inputs})
1012 embedding_column = fc.embedding_column(
1016 embedding_column, {'aaa': inputs})
1039 embedding_column = fc.embedding_column(
1043 embedding_column, {'aaa': sparse_input})
|
| D | sequence_feature_column_integration_test.py | 58 fc.embedding_column(col, dimension=10),
67 fc.embedding_column(identity_col, dimension=10),
68 fc.embedding_column(bucket_col, dimension=20)
|
| D | feature_column_test.py | 2664 embedding_column = fc._embedding_column(
2669 input_layer = InputLayer([embedding_column])
2709 embedding_column = fc._embedding_column(
2714 input_layer = InputLayer([embedding_column])
4788 embedding_column = fc._embedding_column(
4790 self.assertIs(categorical_column, embedding_column.categorical_column)
4791 self.assertEqual(embedding_dimension, embedding_column.dimension)
4792 self.assertEqual('mean', embedding_column.combiner)
4793 self.assertIsNone(embedding_column.ckpt_to_load_from)
4794 self.assertIsNone(embedding_column.tensor_name_in_ckpt)
[all …]
|
| D | feature_column_v2.py | 818 def embedding_column(categorical_column, function
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dnn_test.py | 83 embedding_language = feature_column.embedding_column(
87 embedding_wire = feature_column.embedding_column(
132 embedding_language = feature_column.embedding_column(
240 embedding_language = feature_column.embedding_column(
268 feature_column.embedding_column(language_column, dimension=1),
385 fc_core.embedding_column(language_column, dimension=1),
429 feature_column.embedding_column(
476 feature_column.embedding_column(
584 feature_column.embedding_column(
723 feature_column.embedding_column(
[all …]
|
| D | dnn_linear_combined_test.py | 142 embedding_language = feature_column.embedding_column(
146 embedding_wire = feature_column.embedding_column(
301 embedding_language = feature_column.embedding_column(
329 feature_column.embedding_column(language_column, dimension=1),
491 feature_column.embedding_column(
573 feature_column.embedding_column(
918 feature_column.embedding_column(
1017 feature_column.embedding_column(language, dimension=1)],
1041 feature_column.embedding_column(language, dimension=1)],
1076 feature_column.embedding_column(language, dimension=1)],
[all …]
|
| D | state_saving_rnn_estimator_test.py | 179 feature_column.embedding_column(
224 wire_cast_embedded = feature_column.embedding_column(
274 seq_feature_name, dimension=1), feature_column.embedding_column(
632 feature_column.embedding_column(
|
| D | dynamic_rnn_estimator_test.py | 121 wire_cast_embedded = feature_column.embedding_column(wire_cast, dimension=8)
|
| D | debug_test.py | 577 feature_column.embedding_column(language, dimension=1)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | feature_column_ops_test.py | 196 wire_embedding = feature_column.embedding_column(hashed_sparse, 10)
590 embedding_column = feature_column.embedding_column(sparse_column, 10)
600 columns = [one_hot_column, embedding_column, real_valued_column]
615 embedding_column = feature_column.embedding_column(sparse_column, 10)
625 columns = [one_hot_column, embedding_column, real_valued_column]
851 embeded_sparse = feature_column.embedding_column(hashed_sparse, 10)
902 embeded_sparse = feature_column.embedding_column(
925 embedded_sparse = feature_column.embedding_column(
930 embedded_sparse_alternate = feature_column.embedding_column(
959 embeded_sparse = feature_column.embedding_column(weighted_ids, 10)
[all …]
|
| D | feature_column_test.py | 149 b = fc.embedding_column(a, dimension=4, combiner="mean")
157 column = fc.embedding_column(a, dimension=4, combiner="mean")
855 embedding_col = fc.embedding_column(
1003 embedding_col = fc.embedding_column(
1088 embedding_col = fc.embedding_column(sparse_col, dimension=16)
1118 embedding_col_initialized = fc.embedding_column(
|
| D | feature_column.py | 1318 def embedding_column(sparse_id_column, function
|
| /external/tensorflow/tensorflow/examples/get_started/regression/ |
| D | dnn_regression.py | 76 tf.feature_column.embedding_column(make, dimension=3),
|
| D | custom_regression.py | 131 tf.feature_column.embedding_column(make, dimension=3),
|
| /external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/ |
| D | sequence_feature_column_test.py | 922 embedding_column = fc._embedding_column(
927 embedding_lookup, _ = embedding_column._get_sequence_dense_tensor(
961 embedding_column = fc._embedding_column(categorical_column, dimension=2)
963 _, sequence_length = embedding_column._get_sequence_dense_tensor(
988 embedding_column = fc._embedding_column(categorical_column, dimension=2)
990 _, sequence_length = embedding_column._get_sequence_dense_tensor(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.feature_column.pbtxt | 28 name: "embedding_column"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.feature_column.pbtxt | 28 name: "embedding_column"
|
| /external/tensorflow/tensorflow/contrib/timeseries/examples/ |
| D | lstm.py | 200 tf.feature_column.embedding_column(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 1065 emb_vocab_column = fc.embedding_column(
1135 emb_vocab = fc.embedding_column(
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | head_test.py | 382 feature_column.embedding_column(
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | saved_model_export_utils_test.py | 691 embedding_col = fc.embedding_column(
|