Searched refs:table_config (Results 1 – 3 of 3) sorted by relevance
1452 table_config = self._table_to_config_dict[table]1455 table_descriptor.vocabulary_size = max(table_config.vocabulary_size,1457 table_descriptor.dimension = table_config.dimension1465 if table_config.learning_rate:1466 parameters.learning_rate.constant = table_config.learning_rate1467 elif table_config.learning_rate_fn:1469 self._learning_rate_fn_to_tag[table_config.learning_rate_fn])1497 if table_config.hot_id_replication:1926 for table, table_config in self.table_to_config_dict.items():1927 if table_config.optimization_parameters is not None:[all …]
1370 table_config: tpu_embedding_v2_utils.TableConfig):1396 for table in table_config:1415 table_config: tpu_embedding_v2_utils.TableConfig):1427 for table in table_config:
1146 table_config = tpu_embedding_v2_utils.TableConfig(1154 table=table_config, name='feature')},1204 goldens = [[[-cumsum[i]] * table_config.dim] * num_replicas