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