Back to Tensorflow

Model.Builder

tensorflow/lite/g3doc/api_docs/java/org/tensorflow/lite/support/model/Model.Builder.html

2.21.02.9 KB
Original Source

public static class Model.Builder

This class is deprecated.
Please use Model.createModel(Context, String, Options).

Builder for Model.

Public Constructors

| | Builder(Context context, String modelPath) Creates a builder which loads tflite model from asset folder using memory-mapped files.

|

Public Methods

| Model | build() | | Model.Builder | setDevice(Model.Device device) Sets running device.

| | Model.Builder | setNumThreads(int numThreads) Sets number of threads.

|

Inherited Methods

From class java.lang.Object

| boolean | equals(Object arg0) | | final Class<?> | getClass() | | int | hashCode() | | final void | notify() | | final void | notifyAll() | | String | toString() | | final void | wait(long arg0, int arg1) | | final void | wait(long arg0) | | final void | wait() |

Public Constructors

public Builder (Context context, String modelPath)

Creates a builder which loads tflite model from asset folder using memory-mapped files.

Parameters

| context | Application context to access assets. | | modelPath | Asset path of the model (.tflite file). |

Throws

| IOException | if an I/O error occurs when loading the tflite model. |

Public Methods

public Model build ()

public Model.Builder setDevice (Model.Device device)

Sets running device. By default, TFLite will run on CPU.

Parameters

| device | |

public Model.Builder setNumThreads (int numThreads)

Sets number of threads. By default it's 1.

Parameters

| numThreads | |