Searched refs:restore_fn (Results 1 – 3 of 3) sorted by relevance
315 def restore_fn(): function333 restore_ops = restore_fn()345 restore_ops = restore_fn()
434 def restore_fn(*restored_tensors): function442 concrete_restore_fn = restore_fn.get_concrete_function()500 def restore_fn(*restored_tensors): function503 return factory(save_function=save_fn, restore_function=restore_fn, name=name)
229 for save_fn, restore_fn in saveable_map.values():231 self._add_function_to_graph(restore_fn)303 for local_name, (save_fn, restore_fn) in (307 saveable_object_proto.restore_function = self._node_ids[restore_fn]