public static final class SavedModelBundle.Loader extends Object
Modifier and Type | Method and Description |
---|---|
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.
|
public SavedModelBundle load()
SavedModelBundle
with the configured options.public SavedModelBundle.Loader withRunOptions(byte[] options)
options
- Serialized RunOptions
protocol buffer.public SavedModelBundle.Loader withConfigProto(byte[] configProto)
Session
object created when loading the model.configProto
- Serialized ConfigProto
protocol buffer.public SavedModelBundle.Loader withTags(String... tags)
tags
- the tags identifying the specific MetaGraphDef to load.Copyright © 2022. All rights reserved.