diff options
Diffstat (limited to 'tools/nnapi_quickcheck/tests/gather_2.cpp')
-rw-r--r-- | tools/nnapi_quickcheck/tests/gather_2.cpp | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/tools/nnapi_quickcheck/tests/gather_2.cpp b/tools/nnapi_quickcheck/tests/gather_2.cpp index 4b115685d..ac9ec8b3b 100644 --- a/tools/nnapi_quickcheck/tests/gather_2.cpp +++ b/tools/nnapi_quickcheck/tests/gather_2.cpp @@ -16,13 +16,13 @@ #include "gtest/gtest.h" -#include "tflite/kernels/register.h" -#include "tensorflow/contrib/lite/model.h" -#include "tensorflow/contrib/lite/builtin_op_data.h" +#include "tflite/ext/kernels/register.h" +#include "tensorflow/lite/model.h" +#include "tensorflow/lite/builtin_op_data.h" #include "env.h" #include "memory.h" -#include "util/environment.h" +#include "misc/environment.h" #include "tflite/Diff.h" #include "tflite/interp/FunctionBuilder.h" @@ -31,21 +31,21 @@ #include <iostream> using namespace tflite; -using namespace tflite::ops::builtin; +using namespace nnfw::tflite; TEST(NNAPI_Quickcheck_gather_2, simple_test) { // Set random seed int SEED = std::chrono::system_clock::now().time_since_epoch().count(); - nnfw::util::env::IntAccessor("SEED").access(SEED); + nnfw::misc::env::IntAccessor("SEED").access(SEED); // Set random test parameters int verbose = 0; int tolerance = 1; - nnfw::util::env::IntAccessor("VERBOSE").access(verbose); - nnfw::util::env::IntAccessor("TOLERANCE").access(tolerance); + nnfw::misc::env::IntAccessor("VERBOSE").access(verbose); + nnfw::misc::env::IntAccessor("TOLERANCE").access(tolerance); #define INT_VALUE(NAME, VALUE) IntVar NAME##_Value(#NAME, VALUE); #include "gather_2.lst" @@ -123,7 +123,7 @@ TEST(NNAPI_Quickcheck_gather_2, simple_test) interp.SetOutputs({2}); }; - const nnfw::support::tflite::interp::FunctionBuilder builder(setup); + const nnfw::tflite::FunctionBuilder builder(setup); RandomTestParam param; |