Lines Matching refs:to_restore
2436 to_restore = embedding_weights
2437 if isinstance(to_restore, variables.PartitionedVariable):
2438 to_restore = to_restore._get_variable_list() # pylint: disable=protected-access
2440 self.tensor_name_in_ckpt: to_restore
2579 to_restore = embedding_weights
2580 if isinstance(to_restore, variables.PartitionedVariable):
2581 to_restore = to_restore._get_variable_list() # pylint: disable=protected-access
2583 self.tensor_name_in_ckpt: to_restore