diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2022-09-07 19:04:21 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2022-09-07 19:04:21 +0900 |
commit | c690d52bdd137ed6a17353aa7af35e8141ece77b (patch) | |
tree | dbb7dd99133132dfbffcb8c9e9af4f1ffc2f4808 /compiler/tfldump/driver/Driver.cpp | |
parent | 3ad689f0803519e343c36d5700646e86059df961 (diff) | |
download | nnfw-tizen_7.0_hotfix.tar.gz nnfw-tizen_7.0_hotfix.tar.bz2 nnfw-tizen_7.0_hotfix.zip |
Imported Upstream version 1.21.0upstream/1.21.0tizen_7.0_m2_releaseaccepted/tizen/unified/20220912.170817accepted/tizen/unified/20220912.164738accepted/tizen/7.0/unified/hotfix/20221116.105341accepted/tizen/7.0/unified/20221110.060236tizen_7.0_hotfixtizen_7.0accepted/tizen_7.0_unified_hotfixaccepted/tizen_7.0_unified
Diffstat (limited to 'compiler/tfldump/driver/Driver.cpp')
-rw-r--r-- | compiler/tfldump/driver/Driver.cpp | 15 |
1 files changed, 5 insertions, 10 deletions
diff --git a/compiler/tfldump/driver/Driver.cpp b/compiler/tfldump/driver/Driver.cpp index 38c9c062f..a3e748be1 100644 --- a/compiler/tfldump/driver/Driver.cpp +++ b/compiler/tfldump/driver/Driver.cpp @@ -15,7 +15,7 @@ */ #include <arser/arser.h> -#include <tflread/Model.h> +#include <foder/FileLoader.h> #include <tfldump/Dump.h> #include <iostream> @@ -23,7 +23,7 @@ int entry(int argc, char **argv) { arser::Arser arser; - arser.add_argument("tflite").type(arser::DataType::STR).help("TFLite file to dump"); + arser.add_argument("tflite").help("TFLite file to dump"); try { @@ -38,14 +38,9 @@ int entry(int argc, char **argv) std::string tflite_path = arser.get<std::string>("tflite"); // Load TF lite model from a tflite file - std::unique_ptr<tflread::Model> model = tflread::load_tflite(tflite_path); - if (model == nullptr) - { - std::cerr << "ERROR: Failed to load tflite '" << tflite_path << "'" << std::endl; - return 255; - } - - const tflite::Model *tflmodel = model->model(); + foder::FileLoader fileLoader{tflite_path}; + std::vector<char> modelData = fileLoader.load(); + const tflite::Model *tflmodel = tflite::GetModel(modelData.data()); if (tflmodel == nullptr) { std::cerr << "ERROR: Failed to load tflite '" << tflite_path << "'" << std::endl; |