diff options
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; |