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.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 |
---|