diff options
Diffstat (limited to 'tests/tools/tflite_run/src/tflite_run.cc')
-rw-r--r-- | tests/tools/tflite_run/src/tflite_run.cc | 32 |
1 files changed, 6 insertions, 26 deletions
diff --git a/tests/tools/tflite_run/src/tflite_run.cc b/tests/tools/tflite_run/src/tflite_run.cc index 107aed757..cedcff82a 100644 --- a/tests/tools/tflite_run/src/tflite_run.cc +++ b/tests/tools/tflite_run/src/tflite_run.cc @@ -37,8 +37,6 @@ #include <algorithm> #include <vector> -#include <boost/filesystem.hpp> - using namespace tflite; using namespace nnfw::tflite; using namespace std::placeholders; // for _1, _2 ... @@ -58,12 +56,7 @@ static const char *default_backend_cand = "tflite_cpu"; int main(const int argc, char **argv) { - bool use_nnapi = false; - - if (std::getenv("USE_NNAPI") != nullptr) - { - use_nnapi = true; - } + const bool use_nnapi = nnfw::misc::EnvVar("USE_NNAPI").asBool(false); StderrReporter error_reporter; @@ -327,27 +320,14 @@ int main(const int argc, char **argv) if (args.getWriteReport()) { // prepare csv task - std::string exec_name; - std::string model_name; + std::string exec_basename; + std::string model_basename; std::string backend_name = default_backend_cand; { - namespace fs = boost::filesystem; - - fs::path model_path(args.getTFLiteFilename()); - model_name = model_path.stem().string(); - - fs::path exec_path(argv[0]); - exec_name = exec_path.stem().string(); + model_basename = basename(args.getTFLiteFilename().c_str()); + exec_basename = basename(argv[0]); } - benchmark::writeResult(result, exec_name, model_name, backend_name); - } - - if (!args.getDumpFilename().empty()) - { - const std::string &dump_filename = args.getDumpFilename(); - tensor_dumper.dump(dump_filename); - std::cout << "Input/output tensors have been dumped to file \"" << dump_filename << "\"." - << std::endl; + benchmark::writeResult(result, exec_basename, model_basename, backend_name); } if (!args.getDumpFilename().empty()) |