diff options
| author |  Yu-Cheng Ling <ycling@google.com> | 2018-02-20 20:34:05 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-20 20:38:24 -0800 |
| commit | a6735131827a15127a7563bf7190c5a4e3a19bff (patch) | |
| tree | de627df17ebdd2a427bd77318158c8478b729a87 /tensorflow/contrib/lite/interpreter.cc | |
| parent | 395616ff770318bfe19a9722bc8bc9d792779235 (diff) | |
TFLite: Define a DELEGATE op type.
PiperOrigin-RevId: 186405366
Diffstat (limited to 'tensorflow/contrib/lite/interpreter.cc')
| -rw-r--r-- | tensorflow/contrib/lite/interpreter.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/interpreter.cc b/tensorflow/contrib/lite/interpreter.cc index 0c30f1c64f..370e495527 100644 --- a/tensorflow/contrib/lite/interpreter.cc +++ b/tensorflow/contrib/lite/interpreter.cc @@ -116,6 +116,9 @@ TfLiteStatus Interpreter::ReplaceSubgraphsWithDelegateKernels( TfLiteStatus Interpreter::ReplaceSubgraphsWithDelegateKernels( TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace) { + // Annotate the registration as DELEGATE op. + registration.builtin_code = BuiltinOperator_DELEGATE; + // Analyze the graph to find all independent subgraphs that are either // fully not-this-delegate or this-delegate computation. InterpreterInfo info(this); |