Searched refs:save_function (Results 1 – 4 of 4) sorted by relevance
476 def __init__(self, names_and_slices, save_function, restore_function, name): argument477 self.save_function = save_function485 tensors = save_function(name_tensor)515 save_function=save_fn,553 return factory(save_function=save_fn, restore_function=restore_fn,
303 save_fn_id = saveable_object_proto.save_function311 save_fn_id = saveable_object_proto.save_function
363 saveable_object_proto.save_function = self.node_ids[save_fn]
249 int32 save_function = 2; field