diff options
Diffstat (limited to 'tools/nnapi_quickcheck/tests/softmax_1.cpp')
-rw-r--r-- | tools/nnapi_quickcheck/tests/softmax_1.cpp | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/tools/nnapi_quickcheck/tests/softmax_1.cpp b/tools/nnapi_quickcheck/tests/softmax_1.cpp index b9cc6bf0f..71424755a 100644 --- a/tools/nnapi_quickcheck/tests/softmax_1.cpp +++ b/tools/nnapi_quickcheck/tests/softmax_1.cpp @@ -16,14 +16,14 @@ #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 "util/feature/Shape.h" +#include "misc/environment.h" +#include "misc/feature/Shape.h" #include "tflite/Diff.h" #include "tflite/Quantization.h" @@ -35,15 +35,15 @@ #include <cassert> using namespace tflite; -using namespace tflite::ops::builtin; +using namespace nnfw::tflite; TEST(NNAPI_Quickcheck_softmax_1, simple_test) { 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 "softmax_1.lst" @@ -56,12 +56,12 @@ TEST(NNAPI_Quickcheck_softmax_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); // Initialize random number generator std::minstd_rand random(SEED); - const nnfw::util::feature::Shape ifm_shape{IFM_C, IFM_H, IFM_W}; + const nnfw::misc::feature::Shape ifm_shape{IFM_C, IFM_H, IFM_W}; const int32_t OFM_C = IFM_C; const int32_t OFM_H = IFM_H; @@ -107,7 +107,7 @@ TEST(NNAPI_Quickcheck_softmax_1, simple_test) interp.SetOutputs({0}); }; - const nnfw::support::tflite::interp::FunctionBuilder builder(setup); + const nnfw::tflite::FunctionBuilder builder(setup); RandomTestParam param; |