diff options
Diffstat (limited to 'tools/nnapi_quickcheck/tests/avg_pool_1.cpp')
-rw-r--r-- | tools/nnapi_quickcheck/tests/avg_pool_1.cpp | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/tools/nnapi_quickcheck/tests/avg_pool_1.cpp b/tools/nnapi_quickcheck/tests/avg_pool_1.cpp index e68cfc5b3..052c68915 100644 --- a/tools/nnapi_quickcheck/tests/avg_pool_1.cpp +++ b/tools/nnapi_quickcheck/tests/avg_pool_1.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/Quantization.h" @@ -32,21 +32,21 @@ #include <iostream> using namespace tflite; -using namespace tflite::ops::builtin; +using namespace nnfw::tflite; TEST(NNAPI_Quickcheck_avg_pool_1, 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 "avg_pool_1.lst" @@ -137,7 +137,7 @@ TEST(NNAPI_Quickcheck_avg_pool_1, simple_test) interp.SetOutputs({0}); }; - const nnfw::support::tflite::interp::FunctionBuilder builder(setup); + const nnfw::tflite::FunctionBuilder builder(setup); RandomTestParam param; |