Home
last modified time | relevance | path

Searched refs:restore_fn (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/checkpoint/
Dfunctional_saver.py163 def restore_fn(merged_prefix): function
166 return restore_fn
168 tf_fn = def_function.function(restore_fn, autograph=False)
213 restore_fn = saveable_object_util.saveable_object_to_restore_fn(
229 self._keys_to_restore_fn[(checkpoint_key, slice_spec)] = restore_fn
230 self._restore_fn_to_keys.setdefault(restore_fn, []).append(
247 restore_fn = _get_mapped_registered_restore_fn(
250 self._registered_savers[registered_name] = (save_fn, restore_fn)
409 def restore_fn(): function
426 restore_fn = self._keys_to_restore_fn[(checkpoint_key,
[all …]
Dsave_util_v1_test.py34 restore_fn=lambda trackables, merged_prefix: None)
/external/tensorflow/tensorflow/python/saved_model/registration/
Dregistration_test.py149 restore_fn=lambda: None)
155 restore_fn=lambda: None)
163 restore_fn=lambda: None)
171 restore_fn=lambda: None)
180 restore_fn=lambda: None)
186 restore_fn=lambda: None)
192 restore_fn=None)
200 restore_fn=lambda: "restore")
D__init__.py40 restore_fn=None, argument
48 restore_fn=restore_fn,
Dregistration_saving_test.py115 restore_fn=restore_stacks_and_parts)
287 restore_fn=lambda **kwargs: None,
305 restore_fn=lambda **kwargs: None,
328 def restore_fn(trackables, merged_prefix): function
337 restore_fn=restore_fn)
379 def restore_fn(trackables, merged_prefix): function
400 restore_fn=restore_fn,
Dregistration.py202 restore_fn=None, argument
324 if not callable(restore_fn):
327 _saver_registry.register(package, name, predicate, (save_fn, restore_fn,
DREADME.md43 predicate, save_fn=None, restore_fn=None):
47 be saved using the registered `save_fn` or `restore_fn`.
50 * `restore_fn`: A `tf.function` or `None`. If `None`, run the default
76 **`restore_fn` details**
80 def restore_fn(trackables, file_prefix): -> None
172 restore_fn=restore_stacks_and_parts)
Dtf_registration_test.py44 restore_fn=lambda: None)
50 restore_fn=lambda: None)
/external/tensorflow/tensorflow/python/saved_model/
Dtracing_utils_test.py45 save_fn, restore_fn = tracing_utils.trace_save_and_restore(t)
47 restore_fn({"a": constant_op.constant(2), "b": constant_op.constant(3)})
58 save_fn, restore_fn = tracing_utils.trace_save_and_restore(t)
60 self.assertIs(t._restore_from_tensors, restore_fn)
Dtracing_utils.py54 def restore_fn(restored_tensors): function
61 concrete_restore = restore_fn.get_concrete_function(
Dsave.py320 for save_fn, restore_fn in self._saveable_objects_map[obj].values():
324 self.node_ids[restore_fn] = len(self.nodes)
325 self.nodes.append(restore_fn)
360 for local_name, (save_fn, restore_fn) in (
364 saveable_object_proto.restore_function = self.node_ids[restore_fn]
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py447 def restore_fn(*restored_tensors): function
455 concrete_restore = restore_fn.get_concrete_function()
511 for name, (save_fn, restore_fn) in saveable_fn_by_name.items():
516 restore_function=restore_fn)
550 def restore_fn(*restored_tensors): function
553 return factory(save_function=save_fn, restore_function=restore_fn,
622 restore_fn = self._trackable._restore_from_tensors # pylint: disable=protected-access
624 isinstance(restore_fn, core.ConcreteFunction)):
625 self._call_with_mapped_captures(restore_fn, [restored_tensor_dict])
627 restore_fn(restored_tensor_dict)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2.py1589 restore_fn=_restore_callback,