#![allow(unknown_lints)]
#![allow(clippy::all)]
#![cfg_attr(rustfmt, rustfmt_skip)]
#![allow(box_pointers)]
#![allow(dead_code)]
#![allow(missing_docs)]
#![allow(non_camel_case_types)]
#![allow(non_snake_case)]
#![allow(non_upper_case_globals)]
#![allow(trivial_casts)]
#![allow(unsafe_code)]
#![allow(unused_imports)]
#![allow(unused_results)]
use protobuf::Message as Message_imported_for_functions;
use protobuf::ProtobufEnum as ProtobufEnum_imported_for_functions;
#[derive(PartialEq,Clone,Default)]
pub struct SaverDef {
pub filename_tensor_name: ::std::string::String,
pub save_tensor_name: ::std::string::String,
pub restore_op_name: ::std::string::String,
pub max_to_keep: i32,
pub sharded: bool,
pub keep_checkpoint_every_n_hours: f32,
pub version: SaverDef_CheckpointFormatVersion,
pub unknown_fields: ::protobuf::UnknownFields,
pub cached_size: ::protobuf::CachedSize,
}
impl<'a> ::std::default::Default for &'a SaverDef {
fn default() -> &'a SaverDef {
<SaverDef as ::protobuf::Message>::default_instance()
}
}
impl SaverDef {
pub fn new() -> SaverDef {
::std::default::Default::default()
}
pub fn get_filename_tensor_name(&self) -> &str {
&self.filename_tensor_name
}
pub fn clear_filename_tensor_name(&mut self) {
self.filename_tensor_name.clear();
}
pub fn set_filename_tensor_name(&mut self, v: ::std::string::String) {
self.filename_tensor_name = v;
}
pub fn mut_filename_tensor_name(&mut self) -> &mut ::std::string::String {
&mut self.filename_tensor_name
}
pub fn take_filename_tensor_name(&mut self) -> ::std::string::String {
::std::mem::replace(&mut self.filename_tensor_name, ::std::string::String::new())
}
pub fn get_save_tensor_name(&self) -> &str {
&self.save_tensor_name
}
pub fn clear_save_tensor_name(&mut self) {
self.save_tensor_name.clear();
}
pub fn set_save_tensor_name(&mut self, v: ::std::string::String) {
self.save_tensor_name = v;
}
pub fn mut_save_tensor_name(&mut self) -> &mut ::std::string::String {
&mut self.save_tensor_name
}
pub fn take_save_tensor_name(&mut self) -> ::std::string::String {
::std::mem::replace(&mut self.save_tensor_name, ::std::string::String::new())
}
pub fn get_restore_op_name(&self) -> &str {
&self.restore_op_name
}
pub fn clear_restore_op_name(&mut self) {
self.restore_op_name.clear();
}
pub fn set_restore_op_name(&mut self, v: ::std::string::String) {
self.restore_op_name = v;
}
pub fn mut_restore_op_name(&mut self) -> &mut ::std::string::String {
&mut self.restore_op_name
}
pub fn take_restore_op_name(&mut self) -> ::std::string::String {
::std::mem::replace(&mut self.restore_op_name, ::std::string::String::new())
}
pub fn get_max_to_keep(&self) -> i32 {
self.max_to_keep
}
pub fn clear_max_to_keep(&mut self) {
self.max_to_keep = 0;
}
pub fn set_max_to_keep(&mut self, v: i32) {
self.max_to_keep = v;
}
pub fn get_sharded(&self) -> bool {
self.sharded
}
pub fn clear_sharded(&mut self) {
self.sharded = false;
}
pub fn set_sharded(&mut self, v: bool) {
self.sharded = v;
}
pub fn get_keep_checkpoint_every_n_hours(&self) -> f32 {
self.keep_checkpoint_every_n_hours
}
pub fn clear_keep_checkpoint_every_n_hours(&mut self) {
self.keep_checkpoint_every_n_hours = 0.;
}
pub fn set_keep_checkpoint_every_n_hours(&mut self, v: f32) {
self.keep_checkpoint_every_n_hours = v;
}
pub fn get_version(&self) -> SaverDef_CheckpointFormatVersion {
self.version
}
pub fn clear_version(&mut self) {
self.version = SaverDef_CheckpointFormatVersion::LEGACY;
}
pub fn set_version(&mut self, v: SaverDef_CheckpointFormatVersion) {
self.version = v;
}
}
impl ::protobuf::Message for SaverDef {
fn is_initialized(&self) -> bool {
true
}
fn merge_from(&mut self, is: &mut ::protobuf::CodedInputStream<'_>) -> ::protobuf::ProtobufResult<()> {
while !is.eof()? {
let (field_number, wire_type) = is.read_tag_unpack()?;
match field_number {
1 => {
::protobuf::rt::read_singular_proto3_string_into(wire_type, is, &mut self.filename_tensor_name)?;
},
2 => {
::protobuf::rt::read_singular_proto3_string_into(wire_type, is, &mut self.save_tensor_name)?;
},
3 => {
::protobuf::rt::read_singular_proto3_string_into(wire_type, is, &mut self.restore_op_name)?;
},
4 => {
if wire_type != ::protobuf::wire_format::WireTypeVarint {
return ::std::result::Result::Err(::protobuf::rt::unexpected_wire_type(wire_type));
}
let tmp = is.read_int32()?;
self.max_to_keep = tmp;
},
5 => {
if wire_type != ::protobuf::wire_format::WireTypeVarint {
return ::std::result::Result::Err(::protobuf::rt::unexpected_wire_type(wire_type));
}
let tmp = is.read_bool()?;
self.sharded = tmp;
},
6 => {
if wire_type != ::protobuf::wire_format::WireTypeFixed32 {
return ::std::result::Result::Err(::protobuf::rt::unexpected_wire_type(wire_type));
}
let tmp = is.read_float()?;
self.keep_checkpoint_every_n_hours = tmp;
},
7 => {
::protobuf::rt::read_proto3_enum_with_unknown_fields_into(wire_type, is, &mut self.version, 7, &mut self.unknown_fields)?
},
_ => {
::protobuf::rt::read_unknown_or_skip_group(field_number, wire_type, is, self.mut_unknown_fields())?;
},
};
}
::std::result::Result::Ok(())
}
#[allow(unused_variables)]
fn compute_size(&self) -> u32 {
let mut my_size = 0;
if !self.filename_tensor_name.is_empty() {
my_size += ::protobuf::rt::string_size(1, &self.filename_tensor_name);
}
if !self.save_tensor_name.is_empty() {
my_size += ::protobuf::rt::string_size(2, &self.save_tensor_name);
}
if !self.restore_op_name.is_empty() {
my_size += ::protobuf::rt::string_size(3, &self.restore_op_name);
}
if self.max_to_keep != 0 {
my_size += ::protobuf::rt::value_size(4, self.max_to_keep, ::protobuf::wire_format::WireTypeVarint);
}
if self.sharded != false {
my_size += 2;
}
if self.keep_checkpoint_every_n_hours != 0. {
my_size += 5;
}
if self.version != SaverDef_CheckpointFormatVersion::LEGACY {
my_size += ::protobuf::rt::enum_size(7, self.version);
}
my_size += ::protobuf::rt::unknown_fields_size(self.get_unknown_fields());
self.cached_size.set(my_size);
my_size
}
fn write_to_with_cached_sizes(&self, os: &mut ::protobuf::CodedOutputStream<'_>) -> ::protobuf::ProtobufResult<()> {
if !self.filename_tensor_name.is_empty() {
os.write_string(1, &self.filename_tensor_name)?;
}
if !self.save_tensor_name.is_empty() {
os.write_string(2, &self.save_tensor_name)?;
}
if !self.restore_op_name.is_empty() {
os.write_string(3, &self.restore_op_name)?;
}
if self.max_to_keep != 0 {
os.write_int32(4, self.max_to_keep)?;
}
if self.sharded != false {
os.write_bool(5, self.sharded)?;
}
if self.keep_checkpoint_every_n_hours != 0. {
os.write_float(6, self.keep_checkpoint_every_n_hours)?;
}
if self.version != SaverDef_CheckpointFormatVersion::LEGACY {
os.write_enum(7, self.version.value())?;
}
os.write_unknown_fields(self.get_unknown_fields())?;
::std::result::Result::Ok(())
}
fn get_cached_size(&self) -> u32 {
self.cached_size.get()
}
fn get_unknown_fields(&self) -> &::protobuf::UnknownFields {
&self.unknown_fields
}
fn mut_unknown_fields(&mut self) -> &mut ::protobuf::UnknownFields {
&mut self.unknown_fields
}
fn as_any(&self) -> &dyn (::std::any::Any) {
self as &dyn (::std::any::Any)
}
fn as_any_mut(&mut self) -> &mut dyn (::std::any::Any) {
self as &mut dyn (::std::any::Any)
}
fn into_any(self: Box<Self>) -> ::std::boxed::Box<dyn (::std::any::Any)> {
self
}
fn descriptor(&self) -> &'static ::protobuf::reflect::MessageDescriptor {
Self::descriptor_static()
}
fn new() -> SaverDef {
SaverDef::new()
}
fn descriptor_static() -> &'static ::protobuf::reflect::MessageDescriptor {
static mut descriptor: ::protobuf::lazy::Lazy<::protobuf::reflect::MessageDescriptor> = ::protobuf::lazy::Lazy::INIT;
unsafe {
descriptor.get(|| {
let mut fields = ::std::vec::Vec::new();
fields.push(::protobuf::reflect::accessor::make_simple_field_accessor::<_, ::protobuf::types::ProtobufTypeString>(
"filename_tensor_name",
|m: &SaverDef| { &m.filename_tensor_name },
|m: &mut SaverDef| { &mut m.filename_tensor_name },
));
fields.push(::protobuf::reflect::accessor::make_simple_field_accessor::<_, ::protobuf::types::ProtobufTypeString>(
"save_tensor_name",
|m: &SaverDef| { &m.save_tensor_name },
|m: &mut SaverDef| { &mut m.save_tensor_name },
));
fields.push(::protobuf::reflect::accessor::make_simple_field_accessor::<_, ::protobuf::types::ProtobufTypeString>(
"restore_op_name",
|m: &SaverDef| { &m.restore_op_name },
|m: &mut SaverDef| { &mut m.restore_op_name },
));
fields.push(::protobuf::reflect::accessor::make_simple_field_accessor::<_, ::protobuf::types::ProtobufTypeInt32>(
"max_to_keep",
|m: &SaverDef| { &m.max_to_keep },
|m: &mut SaverDef| { &mut m.max_to_keep },
));
fields.push(::protobuf::reflect::accessor::make_simple_field_accessor::<_, ::protobuf::types::ProtobufTypeBool>(
"sharded",
|m: &SaverDef| { &m.sharded },
|m: &mut SaverDef| { &mut m.sharded },
));
fields.push(::protobuf::reflect::accessor::make_simple_field_accessor::<_, ::protobuf::types::ProtobufTypeFloat>(
"keep_checkpoint_every_n_hours",
|m: &SaverDef| { &m.keep_checkpoint_every_n_hours },
|m: &mut SaverDef| { &mut m.keep_checkpoint_every_n_hours },
));
fields.push(::protobuf::reflect::accessor::make_simple_field_accessor::<_, ::protobuf::types::ProtobufTypeEnum<SaverDef_CheckpointFormatVersion>>(
"version",
|m: &SaverDef| { &m.version },
|m: &mut SaverDef| { &mut m.version },
));
::protobuf::reflect::MessageDescriptor::new_pb_name::<SaverDef>(
"SaverDef",
fields,
file_descriptor_proto()
)
})
}
}
fn default_instance() -> &'static SaverDef {
static mut instance: ::protobuf::lazy::Lazy<SaverDef> = ::protobuf::lazy::Lazy::INIT;
unsafe {
instance.get(SaverDef::new)
}
}
}
impl ::protobuf::Clear for SaverDef {
fn clear(&mut self) {
self.filename_tensor_name.clear();
self.save_tensor_name.clear();
self.restore_op_name.clear();
self.max_to_keep = 0;
self.sharded = false;
self.keep_checkpoint_every_n_hours = 0.;
self.version = SaverDef_CheckpointFormatVersion::LEGACY;
self.unknown_fields.clear();
}
}
impl ::std::fmt::Debug for SaverDef {
fn fmt(&self, f: &mut ::std::fmt::Formatter<'_>) -> ::std::fmt::Result {
::protobuf::text_format::fmt(self, f)
}
}
impl ::protobuf::reflect::ProtobufValue for SaverDef {
fn as_ref(&self) -> ::protobuf::reflect::ReflectValueRef {
::protobuf::reflect::ReflectValueRef::Message(self)
}
}
#[derive(Clone,PartialEq,Eq,Debug,Hash)]
pub enum SaverDef_CheckpointFormatVersion {
LEGACY = 0,
V1 = 1,
V2 = 2,
}
impl ::protobuf::ProtobufEnum for SaverDef_CheckpointFormatVersion {
fn value(&self) -> i32 {
*self as i32
}
fn from_i32(value: i32) -> ::std::option::Option<SaverDef_CheckpointFormatVersion> {
match value {
0 => ::std::option::Option::Some(SaverDef_CheckpointFormatVersion::LEGACY),
1 => ::std::option::Option::Some(SaverDef_CheckpointFormatVersion::V1),
2 => ::std::option::Option::Some(SaverDef_CheckpointFormatVersion::V2),
_ => ::std::option::Option::None
}
}
fn values() -> &'static [Self] {
static values: &'static [SaverDef_CheckpointFormatVersion] = &[
SaverDef_CheckpointFormatVersion::LEGACY,
SaverDef_CheckpointFormatVersion::V1,
SaverDef_CheckpointFormatVersion::V2,
];
values
}
fn enum_descriptor_static() -> &'static ::protobuf::reflect::EnumDescriptor {
static mut descriptor: ::protobuf::lazy::Lazy<::protobuf::reflect::EnumDescriptor> = ::protobuf::lazy::Lazy::INIT;
unsafe {
descriptor.get(|| {
::protobuf::reflect::EnumDescriptor::new_pb_name::<SaverDef_CheckpointFormatVersion>("SaverDef.CheckpointFormatVersion", file_descriptor_proto())
})
}
}
}
impl ::std::marker::Copy for SaverDef_CheckpointFormatVersion {
}
impl ::std::default::Default for SaverDef_CheckpointFormatVersion {
fn default() -> Self {
SaverDef_CheckpointFormatVersion::LEGACY
}
}
impl ::protobuf::reflect::ProtobufValue for SaverDef_CheckpointFormatVersion {
fn as_ref(&self) -> ::protobuf::reflect::ReflectValueRef {
::protobuf::reflect::ReflectValueRef::Enum(self.descriptor())
}
}
static file_descriptor_proto_data: &'static [u8] = b"\
\n$tensorflow/core/protobuf/saver.proto\x12\ntensorflow\"\x89\x03\n\x08S\
averDef\x120\n\x14filename_tensor_name\x18\x01\x20\x01(\tR\x12filenameTe\
nsorName\x12(\n\x10save_tensor_name\x18\x02\x20\x01(\tR\x0esaveTensorNam\
e\x12&\n\x0frestore_op_name\x18\x03\x20\x01(\tR\rrestoreOpName\x12\x1e\n\
\x0bmax_to_keep\x18\x04\x20\x01(\x05R\tmaxToKeep\x12\x18\n\x07sharded\
\x18\x05\x20\x01(\x08R\x07sharded\x12@\n\x1dkeep_checkpoint_every_n_hour\
s\x18\x06\x20\x01(\x02R\x19keepCheckpointEveryNHours\x12F\n\x07version\
\x18\x07\x20\x01(\x0e2,.tensorflow.SaverDef.CheckpointFormatVersionR\x07\
version\"5\n\x17CheckpointFormatVersion\x12\n\n\x06LEGACY\x10\0\x12\x06\
\n\x02V1\x10\x01\x12\x06\n\x02V2\x10\x02Bq\n\x13org.tensorflow.utilB\x0b\
SaverProtosP\x01ZHgithub.com/tensorflow/tensorflow/tensorflow/go/core/co\
re_protos_go_proto\xf8\x01\x01J\x95\x0e\n\x06\x12\x04\0\0/\x01\n\x08\n\
\x01\x0c\x12\x03\0\0\x12\n\x08\n\x01\x02\x12\x03\x02\0\x13\n\x08\n\x01\
\x08\x12\x03\x04\0\x1f\n\t\n\x02\x08\x1f\x12\x03\x04\0\x1f\n\x08\n\x01\
\x08\x12\x03\x05\0,\n\t\n\x02\x08\x08\x12\x03\x05\0,\n\x08\n\x01\x08\x12\
\x03\x06\0\"\n\t\n\x02\x08\n\x12\x03\x06\0\"\n\x08\n\x01\x08\x12\x03\x07\
\0,\n\t\n\x02\x08\x01\x12\x03\x07\0,\n\x08\n\x01\x08\x12\x03\x08\0_\n\t\
\n\x02\x08\x0b\x12\x03\x08\0_\nH\n\x02\x04\0\x12\x04\x0b\0/\x01\x1a<\x20\
Protocol\x20buffer\x20representing\x20the\x20configuration\x20of\x20a\
\x20Saver.\n\n\n\n\x03\x04\0\x01\x12\x03\x0b\x08\x10\nt\n\x04\x04\0\x02\
\0\x12\x03\x0e\x02\"\x1ag\x20The\x20name\x20of\x20the\x20tensor\x20in\
\x20which\x20to\x20specify\x20the\x20filename\x20when\x20saving\x20or\n\
\x20restoring\x20a\x20model\x20checkpoint.\n\n\x0c\n\x05\x04\0\x02\0\x05\
\x12\x03\x0e\x02\x08\n\x0c\n\x05\x04\0\x02\0\x01\x12\x03\x0e\t\x1d\n\x0c\
\n\x05\x04\0\x02\0\x03\x12\x03\x0e\x20!\nC\n\x04\x04\0\x02\x01\x12\x03\
\x11\x02\x1e\x1a6\x20The\x20operation\x20to\x20run\x20when\x20saving\x20\
a\x20model\x20checkpoint.\n\n\x0c\n\x05\x04\0\x02\x01\x05\x12\x03\x11\
\x02\x08\n\x0c\n\x05\x04\0\x02\x01\x01\x12\x03\x11\t\x19\n\x0c\n\x05\x04\
\0\x02\x01\x03\x12\x03\x11\x1c\x1d\nF\n\x04\x04\0\x02\x02\x12\x03\x14\
\x02\x1d\x1a9\x20The\x20operation\x20to\x20run\x20when\x20restoring\x20a\
\x20model\x20checkpoint.\n\n\x0c\n\x05\x04\0\x02\x02\x05\x12\x03\x14\x02\
\x08\n\x0c\n\x05\x04\0\x02\x02\x01\x12\x03\x14\t\x18\n\x0c\n\x05\x04\0\
\x02\x02\x03\x12\x03\x14\x1b\x1c\nX\n\x04\x04\0\x02\x03\x12\x03\x17\x02\
\x18\x1aK\x20Maximum\x20number\x20of\x20checkpoints\x20to\x20keep.\x20\
\x20If\x200,\x20no\x20checkpoints\x20are\x20deleted.\n\n\x0c\n\x05\x04\0\
\x02\x03\x05\x12\x03\x17\x02\x07\n\x0c\n\x05\x04\0\x02\x03\x01\x12\x03\
\x17\x08\x13\n\x0c\n\x05\x04\0\x02\x03\x03\x12\x03\x17\x16\x17\nL\n\x04\
\x04\0\x02\x04\x12\x03\x1a\x02\x13\x1a?\x20Shard\x20the\x20save\x20files\
,\x20one\x20per\x20device\x20that\x20has\x20Variable\x20nodes.\n\n\x0c\n\
\x05\x04\0\x02\x04\x05\x12\x03\x1a\x02\x06\n\x0c\n\x05\x04\0\x02\x04\x01\
\x12\x03\x1a\x07\x0e\n\x0c\n\x05\x04\0\x02\x04\x03\x12\x03\x1a\x11\x12\n\
\x90\x02\n\x04\x04\0\x02\x05\x12\x03\x20\x02*\x1a\x82\x02\x20How\x20ofte\
n\x20to\x20keep\x20an\x20additional\x20checkpoint.\x20If\x20not\x20speci\
fied,\x20only\x20the\x20last\n\x20\"max_to_keep\"\x20checkpoints\x20are\
\x20kept;\x20if\x20specified,\x20in\x20addition\x20to\x20keeping\n\x20th\
e\x20last\x20\"max_to_keep\"\x20checkpoints,\x20an\x20additional\x20chec\
kpoint\x20will\x20be\x20kept\n\x20for\x20every\x20n\x20hours\x20of\x20tr\
aining.\n\n\x0c\n\x05\x04\0\x02\x05\x05\x12\x03\x20\x02\x07\n\x0c\n\x05\
\x04\0\x02\x05\x01\x12\x03\x20\x08%\n\x0c\n\x05\x04\0\x02\x05\x03\x12\
\x03\x20()\n\xa1\x02\n\x04\x04\0\x04\0\x12\x04&\x02-\x03\x1a\x92\x02\x20\
A\x20version\x20number\x20that\x20identifies\x20a\x20different\x20on-dis\
k\x20checkpoint\x20format.\n\x20Usually,\x20each\x20subclass\x20of\x20Ba\
seSaverBuilder\x20works\x20with\x20a\x20particular\n\x20version/format.\
\x20\x20However,\x20it\x20is\x20possible\x20that\x20the\x20same\x20build\
er\x20may\x20be\n\x20upgraded\x20to\x20support\x20a\x20newer\x20checkpoi\
nt\x20format\x20in\x20the\x20future.\n\n\x0c\n\x05\x04\0\x04\0\x01\x12\
\x03&\x07\x1e\n(\n\x06\x04\0\x04\0\x02\0\x12\x03(\x04\x0f\x1a\x19\x20Int\
ernal\x20legacy\x20format.\n\n\x0e\n\x07\x04\0\x04\0\x02\0\x01\x12\x03(\
\x04\n\n\x0e\n\x07\x04\0\x04\0\x02\0\x02\x12\x03(\r\x0e\nY\n\x06\x04\0\
\x04\0\x02\x01\x12\x03*\x04\x0b\x1aJ\x20Deprecated\x20format:\x20tf.Save\
r()\x20which\x20works\x20with\x20tensorflow::table::Table.\n\n\x0e\n\x07\
\x04\0\x04\0\x02\x01\x01\x12\x03*\x04\x06\n\x0e\n\x07\x04\0\x04\0\x02\
\x01\x02\x12\x03*\t\n\n0\n\x06\x04\0\x04\0\x02\x02\x12\x03,\x04\x0b\x1a!\
\x20Current\x20format:\x20more\x20efficient.\n\n\x0e\n\x07\x04\0\x04\0\
\x02\x02\x01\x12\x03,\x04\x06\n\x0e\n\x07\x04\0\x04\0\x02\x02\x02\x12\
\x03,\t\n\n\x0b\n\x04\x04\0\x02\x06\x12\x03.\x02&\n\x0c\n\x05\x04\0\x02\
\x06\x06\x12\x03.\x02\x19\n\x0c\n\x05\x04\0\x02\x06\x01\x12\x03.\x1a!\n\
\x0c\n\x05\x04\0\x02\x06\x03\x12\x03.$%b\x06proto3\
";
static mut file_descriptor_proto_lazy: ::protobuf::lazy::Lazy<::protobuf::descriptor::FileDescriptorProto> = ::protobuf::lazy::Lazy::INIT;
fn parse_descriptor_proto() -> ::protobuf::descriptor::FileDescriptorProto {
::protobuf::parse_from_bytes(file_descriptor_proto_data).unwrap()
}
pub fn file_descriptor_proto() -> &'static ::protobuf::descriptor::FileDescriptorProto {
unsafe {
file_descriptor_proto_lazy.get(|| {
parse_descriptor_proto()
})
}
}