Searched refs:key_name_fn (Results 1 – 1 of 1) sorted by relevance
109 Status Save(std::function<std::string(std::string)> key_name_fn, in Save() argument112 writer->WriteScalar(key_name_fn(kNext), counter_.Peek())); in Save()116 Status Restore(std::function<std::string(std::string)> key_name_fn, in Restore() argument119 TF_RETURN_IF_ERROR(reader->ReadScalar(key_name_fn(kNext), &next)); in Restore()