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(ConfigProto configProto)
Set configuration of the Session object created when loading the model.
SavedModelBundle.Loader
withRunOptions(RunOptions 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 (ConfigProto configProto)

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

Parameters
configProto A ConfigProto protocol buffer.
Returns
  • this object

public SavedModelBundle.Loader withRunOptions (RunOptions options)

Sets options to use when executing model initialization operations.

Parameters
options A RunOptions protocol buffer.
Returns
  • this object

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

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

Has no effect if tags is null or empty

Parameters
tags the tags identifying the specific MetaGraphDef to load.
Returns
  • this object
Throws
IllegalArgumentException if tags are invalid