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/testing/tflite_driver.h | |
| parent | 78ee51bb8c41d616c01674e405784a51e4ec98a5 (diff) | |
Automated rollback of commit 82af048bc8c3c044c98a27b1c4c27bb62d4e4a14
PiperOrigin-RevId: 214705311
Diffstat (limited to 'tensorflow/contrib/lite/testing/tflite_driver.h')
| -rw-r--r-- | tensorflow/contrib/lite/testing/tflite_driver.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/testing/tflite_driver.h b/tensorflow/contrib/lite/testing/tflite_driver.h index dc2a4e5877..aed35f877d 100644 --- a/tensorflow/contrib/lite/testing/tflite_driver.h +++ b/tensorflow/contrib/lite/testing/tflite_driver.h @@ -17,7 +17,7 @@ limitations under the License. #include <map> -#include "tensorflow/contrib/lite/delegates/flex/delegate.h" +#include "tensorflow/contrib/lite/delegates/eager/delegate.h" #include "tensorflow/contrib/lite/interpreter.h" #include "tensorflow/contrib/lite/kernels/register.h" #include "tensorflow/contrib/lite/model.h" @@ -53,7 +53,7 @@ class TfLiteDriver : public TestRunner { class Expectation; - std::unique_ptr<FlexDelegate> delegate_; + std::unique_ptr<EagerDelegate> delegate_; bool use_nnapi_ = false; std::unique_ptr<FlatBufferModel> model_; std::unique_ptr<Interpreter> interpreter_; |