summaryrefslogtreecommitdiff
path: root/tests/tools/tflite_run/src/tflite_run.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tests/tools/tflite_run/src/tflite_run.cc')
-rw-r--r--tests/tools/tflite_run/src/tflite_run.cc32
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())