Update documentation for GPU delegate to recommend using
InterpreterBuilder::AddDelegate rather than
Interpreter::ModifyGraphWithDelegate.
PiperOrigin-RevId: 408364472
Change-Id: Ib8c16abf5a238fd6d3980589359be1dbfef7fe66
diff --git a/tensorflow/lite/delegates/gpu/README.md b/tensorflow/lite/delegates/gpu/README.md
index c37ee90..a217077 100644
--- a/tensorflow/lite/delegates/gpu/README.md
+++ b/tensorflow/lite/delegates/gpu/README.md
@@ -57,21 +57,24 @@
Using TFLite on GPU is as simple as getting the GPU delegate via
`TfLiteGpuDelegateV2Create()` and then passing it to
-`Interpreter::ModifyGraphWithDelegate()` instead of calling
-`Interpreter::AllocateTensors()`:
+`InterpreterBuilder::AddDelegate()`:
```c++
////////
-// Set up interpreter.
+// Set up InterpreterBuilder.
auto model = FlatBufferModel::BuildFromFile(model_path);
ops::builtin::BuiltinOpResolver op_resolver;
-std::unique_ptr<Interpreter> interpreter;
-InterpreterBuilder(*model, op_resolver)(&interpreter);
+InterpreterBuilder interpreter_builder(*model, op_resolver);
////////
// NEW: Prepare GPU delegate.
auto* delegate = TfLiteGpuDelegateV2Create(/*default options=*/nullptr);
-if (interpreter->ModifyGraphWithDelegate(delegate) != kTfLiteOk) return;
+interpreter_builder.AddDelegate(delegate);
+
+////////
+// Set up Interpreter.
+std::unique_ptr<Interpreter> interpreter;
+if (interpreter_builder(&interpreter) != kTfLiteOk) return;
////////
// Run inference.
@@ -85,11 +88,13 @@
```
*IMPORTANT:* When calling `Interpreter::ModifyGraphWithDelegate()` or
+`InterpreterBuilder::operator()` or
`Interpreter::Invoke()`, the caller must have a `EGLContext` in the current
thread and `Interpreter::Invoke()` must be called from the same `EGLContext`.
If such `EGLContext` does not exist, the delegate will internally create one,
but then the developer must ensure that `Interpreter::Invoke()` is always called
-from the same thread `Interpreter::ModifyGraphWithDelegate()` was called.
+from the same thread `InterpreterBuilder::operator()` or
+`Interpreter::ModifyGraphWithDelegate()` was called.
## Building and Runtime
