Searched refs:table_to_features_dict (Results 1 – 2 of 2) sorted by relevance
72 for table_id, table in enumerate(tpu_embedding.table_to_features_dict):95 [None] * len(tpu_embedding.table_to_features_dict[table]))125 lookup_id=tpu_embedding.table_to_features_dict[table].index(feature))163 for feature, gradient in zip(tpu_embedding.table_to_features_dict[table],
1438 def table_to_features_dict(self): member in TPUEmbedding2798 def _create_combiners(table_to_config_dict, table_to_features_dict): argument2803 combiners.extend([combiner] * len(table_to_features_dict[table]))2825 table_to_features_dict = collections.OrderedDict()2828 table_to_features_dict[table] = sorted(table_to_features_dict_tmp[table])2830 return table_to_features_dict, table_to_num_features_dict