diff options
author | Nupur Garg <nupurgarg@google.com> | 2018-09-26 18:26:51 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-26 18:31:32 -0700 |
commit | 51a6118e5bd85935b1d9ec0e68b92f1f98d14982 (patch) | |
tree | 5b3d7b469bebd510b678a2054c04311e4879fd09 /tensorflow/contrib/lite/model.cc | |
parent | 78ee51bb8c41d616c01674e405784a51e4ec98a5 (diff) |
Automated rollback of commit 82af048bc8c3c044c98a27b1c4c27bb62d4e4a14
PiperOrigin-RevId: 214705311
Diffstat (limited to 'tensorflow/contrib/lite/model.cc')
-rw-r--r-- | tensorflow/contrib/lite/model.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/model.cc b/tensorflow/contrib/lite/model.cc index eff6181a61..ea2817beec 100644 --- a/tensorflow/contrib/lite/model.cc +++ b/tensorflow/contrib/lite/model.cc @@ -28,7 +28,7 @@ limitations under the License. #include "tensorflow/contrib/lite/nnapi_delegate.h" #endif #if defined(TFLITE_EXTENDED) -#include "tensorflow/contrib/lite/delegates/flex/delegate.h" +#include "tensorflow/contrib/lite/delegates/eager/delegate.h" #endif #include "tensorflow/contrib/lite/version.h" @@ -451,7 +451,7 @@ TfLiteStatus InterpreterBuilder::operator()( (**interpreter).SetVariables(std::move(variables)); #if defined(TFLITE_EXTENDED) - if (auto delegate = FlexDelegate::Create()) { + if (auto delegate = EagerDelegate::Create()) { (**interpreter) .ModifyGraphWithDelegate(std::move(delegate), /*allow_dynamic_tensors=*/true); |