Struct tensorflow::SavedModelBuilder

pub struct SavedModelBuilder { /* private fields */ }

Builds a SavedModelSaver, which can be used to save models.

Implementations

impl SavedModelBuilder

pub fn new() -> SavedModelBuilder

Creates a new SavedModelBuilder.

pub fn add_collection(&mut self, key: &str, variables: &[Variable]) -> &mut Self

Adds a collection to be saved.

pub fn add_tag(&mut self, tag: &str) -> &mut Self

Adds a tag.

pub fn add_signature( &mut self, key: &str, signature_def: SignatureDef ) -> &mut Self

Adds a signature.

pub fn inject(self, scope: &mut Scope) -> Result<SavedModelSaver>

Adds ops to the graph necessary for saving and restoring models, returning a SavedModelSaver which handles the actual model saving.

Trait Implementations

impl Debug for SavedModelBuilder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for SavedModelBuilder

fn default() -> Self

Returns the “default value” for a type.