Searched refs:feature_to_config_dict (Results 1 – 3 of 3) sorted by relevance
120 table = tpu_embedding.feature_to_config_dict[feature].table_id171 tpu_embedding.feature_to_config_dict[feature].max_sequence_length)
1223 feature_to_config_dict, argument1293 feature_to_config_dict)1294 self._feature_to_config_dict = _create_ordered_dict(feature_to_config_dict)1434 def feature_to_config_dict(self): member in TPUEmbedding1945 feature_to_config_dict):1948 [feature.table_id for feature in feature_to_config_dict.values()])2807 def _create_table_to_features_and_num_features_dicts(feature_to_config_dict): argument2811 for feature, feature_config in six.iteritems(feature_to_config_dict):
19 name: "feature_to_config_dict"