Searched refs:save_function (Results 1 – 4 of 4) sorted by relevance
449 def __init__(self, save_function, restore_function, name): argument450 self.save_function = save_function458 tensors = save_function(name_tensor)469 def restored_saved_object_factory(save_function, restore_function): argument471 save_function=save_function,503 return factory(save_function=save_fn, restore_function=restore_fn, name=name)
204 int32 save_function = 2; field
274 self.get(saveable_object_proto.save_function),
306 saveable_object_proto.save_function = self._node_ids[save_fn]