compiler/tfldump/driver/Driver.cpp

   1 /*
   2  * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *    http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <arser/arser.h>
  18 #include <tflread/Model.h>
  19 #include <tfldump/Dump.h>
  20
  21 #include <iostream>
  22
  23 int entry(int argc, char **argv)
  24 {
  25   arser::Arser arser;
  26   arser.add_argument("tflite").type(arser::DataType::STR).help("TFLite file to dump");
  27
  28   try
  29   {
  30     arser.parse(argc, argv);
  31   }
  32   catch (const std::runtime_error &err)
  33   {
  34     std::cout << err.what() << '\n';
  35     std::cout << arser;
  36     return 255;
  37   }
  38
  39   std::string tflite_path = arser.get<std::string>("tflite");
  40   // Load TF lite model from a tflite file
  41   std::unique_ptr<tflread::Model> model = tflread::load_tflite(tflite_path);
  42   if (model == nullptr)
  43   {
  44     std::cerr << "ERROR: Failed to load tflite '" << tflite_path << "'" << std::endl;
  45     return 255;
  46   }
  47
  48   const tflite::Model *tflmodel = model->model();
  49   if (tflmodel == nullptr)
  50   {
  51     std::cerr << "ERROR: Failed to load tflite '" << tflite_path << "'" << std::endl;
  52     return 255;
  53   }
  54
  55   std::cout << "Dump: " << tflite_path << std::endl << std::endl;
  56
  57   std::cout << tflmodel << std::endl;
  58
  59   return 0;
  60 }