com.spotify.zoltar.tf

Interface TensorFlowGraphLoader

All Superinterfaces:

ModelLoader<TensorFlowGraphModel>

Functional Interface:

This is a functional interface and can therefore be used as the assignment target for a lambda expression or method reference.

@FunctionalInterface
public interface TensorFlowGraphLoader
extends ModelLoader<TensorFlowGraphModel>

TensorFlowGraphModel loader. This loader is composed with ModelMemoizer and Preloader.

Nested Class Summary

Nested classes/interfaces inherited from interface com.spotify.zoltar.ModelLoader

ModelLoader.ThrowableSupplier<M extends Model<?>>

Method Summary

Modifier and Type	Method and Description
static TensorFlowGraphLoader	create(byte[] graphDef, org.tensorflow.framework.ConfigProto config, String prefix) Returns a TensorFlow model loader based on a serialized TensorFlow Graph.
static TensorFlowGraphLoader	create(Model.Id id, byte[] graphDef, org.tensorflow.framework.ConfigProto config, String prefix) Returns a TensorFlow model loader based on a serialized TensorFlow Graph.
static TensorFlowGraphLoader	create(Model.Id id, String modelUri, org.tensorflow.framework.ConfigProto config, String prefix) Returns a TensorFlow model loader based on a serialized TensorFlow Graph.
static TensorFlowGraphLoader	create(ModelLoader.ThrowableSupplier<TensorFlowGraphModel> supplier) Returns a TensorFlow model loader based on a serialized TensorFlow Graph.
static TensorFlowGraphLoader	create(String modelUri, org.tensorflow.framework.ConfigProto config, String prefix) Returns a TensorFlow model loader based on a serialized TensorFlow Graph.

Methods inherited from interface com.spotify.zoltar.ModelLoader

get, get, lift, with

Method Detail

create

static TensorFlowGraphLoader create(String modelUri,
                                    @Nullable
                                    org.tensorflow.framework.ConfigProto config,
                                    @Nullable
                                    String prefix)

Returns a TensorFlow model loader based on a serialized TensorFlow Graph.

Parameters:

modelUri - should point to a serialized TensorFlow Graph file on local filesystem, resource, GCS etc.

config - optional TensorFlow ConfigProto config.

prefix - optional prefix that will be prepended to names in the graph.

create

static TensorFlowGraphLoader create(Model.Id id,
                                    String modelUri,
                                    @Nullable
                                    org.tensorflow.framework.ConfigProto config,
                                    @Nullable
                                    String prefix)

Returns a TensorFlow model loader based on a serialized TensorFlow Graph.

Parameters:

id - model id @{link Model.Id}.

modelUri - should point to a serialized TensorFlow Graph file on local filesystem, resource, GCS etc.

config - optional TensorFlow ConfigProto config.

prefix - optional prefix that will be prepended to names in the graph.

create

static TensorFlowGraphLoader create(byte[] graphDef,
                                    @Nullable
                                    org.tensorflow.framework.ConfigProto config,
                                    @Nullable
                                    String prefix)

Returns a TensorFlow model loader based on a serialized TensorFlow Graph.

Parameters:

graphDef - byte array representing the TensorFlow Graph definition.

config - optional TensorFlow ConfigProto config.

prefix - optional prefix that will be prepended to names in the graph.

create

static TensorFlowGraphLoader create(Model.Id id,
                                    byte[] graphDef,
                                    @Nullable
                                    org.tensorflow.framework.ConfigProto config,
                                    @Nullable
                                    String prefix)

Returns a TensorFlow model loader based on a serialized TensorFlow Graph.

Parameters:

id - model id @{link Model.Id}.

graphDef - byte array representing the TensorFlow Graph definition.

config - optional TensorFlow ConfigProto config.

prefix - optional prefix that will be prepended to names in the graph.

create

static TensorFlowGraphLoader create(ModelLoader.ThrowableSupplier<TensorFlowGraphModel> supplier)

Returns a TensorFlow model loader based on a serialized TensorFlow Graph.

Parameters:

supplier - TensorFlowGraphModel supplier.