org.tensorflow

Class SavedModelBundle.Loader

java.lang.Object

org.tensorflow.SavedModelBundle.Loader

Enclosing class:

SavedModelBundle

public static final class SavedModelBundle.Loader
extends Object

Options for loading a SavedModel.

Method Summary

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.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

load

public SavedModelBundle load()

Load a SavedModelBundle with the configured options.

withRunOptions

public SavedModelBundle.Loader withRunOptions(byte[] options)

Sets options to use when executing model initialization operations.

Parameters:

options - Serialized RunOptions protocol buffer.

withConfigProto

public SavedModelBundle.Loader withConfigProto(byte[] configProto)

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

Parameters:

configProto - Serialized ConfigProto protocol buffer.

withTags

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.