Searched refs:to_restore (Results 1 – 3 of 3) sorted by relevance
2436 to_restore = embedding_weights2437 if isinstance(to_restore, variables.PartitionedVariable):2438 to_restore = to_restore._get_variable_list() # pylint: disable=protected-access2440 self.tensor_name_in_ckpt: to_restore2579 to_restore = embedding_weights2580 if isinstance(to_restore, variables.PartitionedVariable):2581 to_restore = to_restore._get_variable_list() # pylint: disable=protected-access2583 self.tensor_name_in_ckpt: to_restore
3055 to_restore = embedding_weights3056 if isinstance(to_restore, variables.PartitionedVariable):3057 to_restore = to_restore._get_variable_list() # pylint: disable=protected-access3059 self.tensor_name_in_ckpt: to_restore3257 to_restore = var3258 if isinstance(to_restore, variables.PartitionedVariable):3259 to_restore = to_restore._get_variable_list() # pylint: disable=protected-access3261 self._ckpt_to_load_from, {self._tensor_name_in_ckpt: to_restore})
233 unsigned to_restore = 0; in sanitize_hash() local255 samplers_to_restore[to_restore++] = sampler; in sanitize_hash()278 while (to_restore--) { in sanitize_hash()279 struct cso_sampler *sampler = samplers_to_restore[to_restore]; in sanitize_hash()