From 3bbdcfe663098dfefce2ea1d6638f9bd1f5cce5f Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Tue, 3 Jul 2018 11:33:08 +0900 Subject: [PATCH] Remove 'tflitekit' (#463) Now 'nnkit' includes 'tflite' backend which is identical to 'tflitekit'. Signed-off-by: Jonghyun Park --- contrib/tflitekit/CMakeLists.txt | 12 ------- contrib/tflitekit/src/tflitekit.cpp | 62 ------------------------------------- 2 files changed, 74 deletions(-) delete mode 100644 contrib/tflitekit/CMakeLists.txt delete mode 100644 contrib/tflitekit/src/tflitekit.cpp diff --git a/contrib/tflitekit/CMakeLists.txt b/contrib/tflitekit/CMakeLists.txt deleted file mode 100644 index e990643..0000000 --- a/contrib/tflitekit/CMakeLists.txt +++ /dev/null @@ -1,12 +0,0 @@ -nncc_find_package(TensorFlowLite QUIET) - -if(NOT TensorFlowLite_FOUND) - return() -endif(NOT TensorFlowLite_FOUND) - -file(GLOB_RECURSE SOURCES "src/*.cpp") - -add_executable(tflitekit ${SOURCES}) -target_link_libraries(tflitekit nncc_foundation) -target_link_libraries(tflitekit cli) -target_link_libraries(tflitekit tensorflowlite) diff --git a/contrib/tflitekit/src/tflitekit.cpp b/contrib/tflitekit/src/tflitekit.cpp deleted file mode 100644 index f29cf48..0000000 --- a/contrib/tflitekit/src/tflitekit.cpp +++ /dev/null @@ -1,62 +0,0 @@ -#include "tensorflow/contrib/lite/kernels/register.h" -#include "tensorflow/contrib/lite/model.h" - -#include -#include - -using namespace tflite; -using namespace tflite::ops::builtin; - -class RunCommand final : public cli::Command -{ -public: - int run(int argc, const char * const *argv) const override; -}; - -int RunCommand::run(int argc, const char * const *argv) const -{ - // USAGE: HEADER run [.tflite] - const auto filename = argv[0]; - - StderrReporter error_reporter; - - auto model = FlatBufferModel::BuildFromFile(filename, &error_reporter); - - std::unique_ptr interpreter; - - TfLiteStatus status = kTfLiteError; - - BuiltinOpResolver resolver; - InterpreterBuilder builder(*model, resolver); - - status = builder(&interpreter); - assert(status == kTfLiteOk); - - interpreter->SetNumThreads(1); - - status = interpreter->AllocateTensors(); - assert(status == kTfLiteOk); - - status = interpreter->Invoke(); - assert(status == kTfLiteOk); - - std::cout << "# of outputs: " << interpreter->outputs().size() << std::endl; - - return 0; -} - -#include -#include - -#include -#include -#include - -int main(int argc, char **argv) -{ - cli::App app{argv[0]}; - - app.insert("run", nncc::foundation::make_unique()); - - return app.run(argc - 1, argv + 1); -} -- 2.7.4