Searched refs:model_proto (Results 1 – 3 of 3) sorted by relevance
1932 Status Model::ToProto(ModelProto* model_proto) { in ToProto() argument1933 ModelProto::Node* output_proto = model_proto->mutable_output(); in ToProto()1936 model_proto->set_id_counter(id_counter_); in ToProto()1937 model_proto->set_collect_resource_usage(collect_resource_usage_); in ToProto()1941 Status Model::FromProto(ModelProto model_proto, std::unique_ptr<Model>* model) { in FromProto() argument1945 Node::FromProto(model_proto.output(), /*output=*/nullptr, &output)); in FromProto()1948 restored_model->id_counter_ = model_proto.id_counter(); in FromProto()1950 model_proto.collect_resource_usage()); in FromProto()1957 ModelProto model_proto; in Save() local1965 TF_RETURN_IF_ERROR(model_snapshot->ToProto(&model_proto)); in Save()[all …]
690 Status ToProto(ModelProto* model_proto);693 static Status FromProto(ModelProto model_proto,
1658 name = "model_proto",1686 ":model_proto",