compiler/mio-tflite260/example.cpp

   1 /*
   2  * Copyright (c) 2021 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 //
  18 // This example shows how to include and use "mio-tflite260"
  19 //
  20 #include <mio/tflite/schema_generated.h>
  21
  22 #include <fstream>
  23 #include <iostream>
  24 #include <vector>
  25
  26 int main(int argc, char **argv)
  27 {
  28   std::ifstream ifs(argv[1], std::ios_base::binary);
  29   std::vector<char> buf(std::istreambuf_iterator<char>{ifs}, std::istreambuf_iterator<char>{});
  30
  31   flatbuffers::Verifier verifier{reinterpret_cast<uint8_t *>(buf.data()), buf.size()};
  32
  33   if (!tflite::VerifyModelBuffer(verifier))
  34   {
  35     std::cout << "Fail" << std::endl;
  36     return 255;
  37   }
  38
  39   std::cout << "Pass" << std::endl;
  40   return 0;
  41 }