SavedModelBundle.Loader

public static final class SavedModelBundle.Loader

Options for loading a SavedModel.

Public Methods

SavedModelBundle
load ()
Load a SavedModelBundle with the configured options.
SavedModelBundle.Loader
withConfigProto (byte[] configProto)
Set configuration of the Session object created when loading the model.
SavedModelBundle.Loader
withRunOptions (byte[] options)
Sets options to use when executing model initialization operations.
SavedModelBundle.Loader
withTags (String... tags)
Sets the set of tags that identify the specific graph in the saved model to load.

Inherited Methods

Public Methods

public SavedModelBundle load ()

Load a SavedModelBundle with the configured options.

public SavedModelBundle.Loader withConfigProto (byte[] configProto)

Set configuration of the Session object created when loading the model.

Parameters
configProto Serialized ConfigProto protocol buffer .

public SavedModelBundle.Loader withRunOptions (byte[] options)

Sets options to use when executing model initialization operations.

Parameters
options Serialized RunOptions protocol buffer .

public SavedModelBundle.Loader withTags (String... tags)

Sets the set of tags that identify the specific graph in the saved model to load.

Parameters
tags the tags identifying the specific MetaGraphDef to load.