Document exceptions for Interpreter in the base java API.

PiperOrigin-RevId: 288342138
Change-Id: I5532be5c598806f874eb777ee3e659c737646814
This commit is contained in:
Lu Wang 2020-01-06 11:37:22 -08:00 committed by TensorFlower Gardener
parent 31c3789692
commit 30a3ac9cf3
2 changed files with 32 additions and 13 deletions

View File

@ -98,7 +98,8 @@ public final class Interpreter implements AutoCloseable {
/** /**
* Sets whether to allow float16 precision for FP32 calculation when possible. Defaults to false * Sets whether to allow float16 precision for FP32 calculation when possible. Defaults to false
* (disallow). * (disallow).
* WARNING: This is an experimental API and subject to change. *
* <p>WARNING: This is an experimental API and subject to change.
*/ */
public Options setAllowFp16PrecisionForFp32(boolean allow) { public Options setAllowFp16PrecisionForFp32(boolean allow) {
this.allowFp16PrecisionForFp32 = allow; this.allowFp16PrecisionForFp32 = allow;
@ -142,6 +143,8 @@ public final class Interpreter implements AutoCloseable {
* Initializes a {@code Interpreter} * Initializes a {@code Interpreter}
* *
* @param modelFile: a File of a pre-trained TF Lite model. * @param modelFile: a File of a pre-trained TF Lite model.
* @throws IllegalArgumentException if {@code modelFile} does not encode a valid TensorFlow Lite
* model.
*/ */
public Interpreter(@NonNull File modelFile) { public Interpreter(@NonNull File modelFile) {
this(modelFile, /*options = */ null); this(modelFile, /*options = */ null);
@ -165,6 +168,8 @@ public final class Interpreter implements AutoCloseable {
* *
* @param modelFile: a file of a pre-trained TF Lite model * @param modelFile: a file of a pre-trained TF Lite model
* @param options: a set of options for customizing interpreter behavior * @param options: a set of options for customizing interpreter behavior
* @throws IllegalArgumentException if {@code modelFile} does not encode a valid TensorFlow Lite
* model.
*/ */
public Interpreter(@NonNull File modelFile, Options options) { public Interpreter(@NonNull File modelFile, Options options) {
wrapper = new NativeInterpreterWrapper(modelFile.getAbsolutePath(), options); wrapper = new NativeInterpreterWrapper(modelFile.getAbsolutePath(), options);
@ -176,6 +181,9 @@ public final class Interpreter implements AutoCloseable {
* <p>The ByteBuffer should not be modified after the construction of a {@code Interpreter}. The * <p>The ByteBuffer should not be modified after the construction of a {@code Interpreter}. The
* {@code ByteBuffer} can be either a {@code MappedByteBuffer} that memory-maps a model file, or a * {@code ByteBuffer} can be either a {@code MappedByteBuffer} that memory-maps a model file, or a
* direct {@code ByteBuffer} of nativeOrder() that contains the bytes content of a model. * direct {@code ByteBuffer} of nativeOrder() that contains the bytes content of a model.
*
* @throws IllegalArgumentException if {@code byteBuffer} is not a {@link MappedByteBuffer} nor a
* direct {@link Bytebuffer} of nativeOrder.
*/ */
public Interpreter(@NonNull ByteBuffer byteBuffer) { public Interpreter(@NonNull ByteBuffer byteBuffer) {
this(byteBuffer, /* options= */ null); this(byteBuffer, /* options= */ null);
@ -216,8 +224,11 @@ public final class Interpreter implements AutoCloseable {
* {@link #Options}. * {@link #Options}.
* *
* <p>The ByteBuffer should not be modified after the construction of a {@code Interpreter}. The * <p>The ByteBuffer should not be modified after the construction of a {@code Interpreter}. The
* {@code ByteBuffer} can be either a {@code MappedByteBuffer} that memory-maps a model file, or a * {@code ByteBuffer} can be either a {@link MappedByteBuffer} that memory-maps a model file, or a
* direct {@code ByteBuffer} of nativeOrder() that contains the bytes content of a model. * direct {@link ByteBuffer} of nativeOrder() that contains the bytes content of a model.
*
* @throws IllegalArgumentException if {@code byteBuffer} is not a {@link MappedByteBuffer} nor a
* direct {@link Bytebuffer} of nativeOrder.
*/ */
public Interpreter(@NonNull ByteBuffer byteBuffer, Options options) { public Interpreter(@NonNull ByteBuffer byteBuffer, Options options) {
wrapper = new NativeInterpreterWrapper(byteBuffer, options); wrapper = new NativeInterpreterWrapper(byteBuffer, options);
@ -251,6 +262,8 @@ public final class Interpreter implements AutoCloseable {
* that it is set the appropriate write position. A null value is allowed only if the caller * that it is set the appropriate write position. A null value is allowed only if the caller
* is using a {@link Delegate} that allows buffer handle interop, and such a buffer has been * is using a {@link Delegate} that allows buffer handle interop, and such a buffer has been
* bound to the output {@link Tensor}. See {@link Options#setAllowBufferHandleOutput()}. * bound to the output {@link Tensor}. See {@link Options#setAllowBufferHandleOutput()}.
* @throws IllegalArgumentException if {@code input} or {@code output} is null or empty, or if
* error occurs when running the inference.
*/ */
public void run(Object input, Object output) { public void run(Object input, Object output) {
Object[] inputs = {input}; Object[] inputs = {input};
@ -289,6 +302,8 @@ public final class Interpreter implements AutoCloseable {
* Buffer}s of primitive types including int, float, long, and byte. It only needs to keep * Buffer}s of primitive types including int, float, long, and byte. It only needs to keep
* entries for the outputs to be used. When a {@link Buffer} is used, the caller must ensure * entries for the outputs to be used. When a {@link Buffer} is used, the caller must ensure
* that it is set the appropriate write position. * that it is set the appropriate write position.
* @throws IllegalArgumentException if {@code inputs} or {@code outputs} is null or empty, or if
* error occurs when running the inference.
*/ */
public void runForMultipleInputsOutputs( public void runForMultipleInputsOutputs(
@NonNull Object[] inputs, @NonNull Map<Integer, Object> outputs) { @NonNull Object[] inputs, @NonNull Map<Integer, Object> outputs) {
@ -299,7 +314,8 @@ public final class Interpreter implements AutoCloseable {
/** /**
* Resizes idx-th input of the native model to the given dims. * Resizes idx-th input of the native model to the given dims.
* *
* <p>IllegalArgumentException will be thrown if it fails to resize. * @throws IllegalArgumentException if {@code idx} is negtive or is not smaller than the number of
* model inputs; or if error occurs when resizing the idx-th input.
*/ */
public void resizeInput(int idx, @NonNull int[] dims) { public void resizeInput(int idx, @NonNull int[] dims) {
checkNotClosed(); checkNotClosed();
@ -315,8 +331,8 @@ public final class Interpreter implements AutoCloseable {
/** /**
* Gets index of an input given the op name of the input. * Gets index of an input given the op name of the input.
* *
* <p>IllegalArgumentException will be thrown if the op name does not exist in the model file used * @throws IllegalArgumentException if {@code opName} does not match any input in the model used
* to initialize the {@link Interpreter}. * to initialize the {@link Interpreter}.
*/ */
public int getInputIndex(String opName) { public int getInputIndex(String opName) {
checkNotClosed(); checkNotClosed();
@ -326,7 +342,8 @@ public final class Interpreter implements AutoCloseable {
/** /**
* Gets the Tensor associated with the provdied input index. * Gets the Tensor associated with the provdied input index.
* *
* <p>IllegalArgumentException will be thrown if the provided index is invalid. * @throws IllegalArgumentException if {@code inputIndex} is negtive or is not smaller than the
* number of model inputs.
*/ */
public Tensor getInputTensor(int inputIndex) { public Tensor getInputTensor(int inputIndex) {
checkNotClosed(); checkNotClosed();
@ -342,8 +359,8 @@ public final class Interpreter implements AutoCloseable {
/** /**
* Gets index of an output given the op name of the output. * Gets index of an output given the op name of the output.
* *
* <p>IllegalArgumentException will be thrown if the op name does not exist in the model file used * @throws IllegalArgumentException if {@code opName} does not match any output in the model used
* to initialize the {@link Interpreter}. * to initialize the {@link Interpreter}.
*/ */
public int getOutputIndex(String opName) { public int getOutputIndex(String opName) {
checkNotClosed(); checkNotClosed();
@ -353,7 +370,8 @@ public final class Interpreter implements AutoCloseable {
/** /**
* Gets the Tensor associated with the provdied output index. * Gets the Tensor associated with the provdied output index.
* *
* <p>IllegalArgumentException will be thrown if the provided index is invalid. * @throws IllegalArgumentException if {@code outputIndex} is negtive or is not smaller than the
* number of model outputs.
*/ */
public Tensor getOutputTensor(int outputIndex) { public Tensor getOutputTensor(int outputIndex) {
checkNotClosed(); checkNotClosed();
@ -363,8 +381,7 @@ public final class Interpreter implements AutoCloseable {
/** /**
* Returns native inference timing. * Returns native inference timing.
* *
* <p>IllegalArgumentException will be thrown if the model is not initialized by the {@link * @throws IllegalArgumentException if the model is not initialized by the {@link Interpreter}.
* Interpreter}.
*/ */
public Long getLastNativeInferenceDurationNanoseconds() { public Long getLastNativeInferenceDurationNanoseconds() {
checkNotClosed(); checkNotClosed();
@ -403,6 +420,8 @@ public final class Interpreter implements AutoCloseable {
* interaction between Interpeter creation and delegate application. * interaction between Interpeter creation and delegate application.
* *
* <p>WARNING: This is an experimental API and subject to change. * <p>WARNING: This is an experimental API and subject to change.
*
* @throws IllegalArgumentException if error occurs when modifying graph with {@code delegate}.
*/ */
public void modifyGraphWithDelegate(Delegate delegate) { public void modifyGraphWithDelegate(Delegate delegate) {
checkNotClosed(); checkNotClosed();

View File

@ -349,7 +349,7 @@ Java_org_tensorflow_lite_NativeInterpreterWrapper_createModel(
if (!model) { if (!model) {
ThrowException(env, kIllegalArgumentException, ThrowException(env, kIllegalArgumentException,
"Contents of %s does not encode a valid " "Contents of %s does not encode a valid "
"TensorFlowLite model: %s", "TensorFlow Lite model: %s",
path, error_reporter->CachedErrorMessage()); path, error_reporter->CachedErrorMessage());
env->ReleaseStringUTFChars(model_file, path); env->ReleaseStringUTFChars(model_file, path);
return 0; return 0;