Searched refs:save_fn (Results 1 – 4 of 4) sorted by relevance
94 def save_fn(): function97 save_fn()
257 def save_fn(): function297 save_fn()300 return save_fn()
398 def save_fn(checkpoint_key): function412 concrete_save_fn = save_fn.get_concrete_function()496 def save_fn(name): function503 return factory(save_function=save_fn, restore_function=restore_fn, name=name)
229 for save_fn, restore_fn in saveable_map.values():230 self._add_function_to_graph(save_fn)303 for local_name, (save_fn, restore_fn) in (306 saveable_object_proto.save_function = self._node_ids[save_fn]