From 0ca19802f78255cd9d77a2608daa647c6630d7f9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=9F=B3=E6=99=93=E4=BC=9F?= <39303645+Shixiaowei02@users.noreply.github.com> Date: Wed, 19 Aug 2020 15:05:42 +0800 Subject: [PATCH] input_vars of flatbuffers impl, test=develop (#4147) * input_vars of flatbuffers impl, test=develop * add comments in build.sh, test=develop --- lite/model_parser/flatbuffers/op_desc.h | 11 +++++++++-- lite/tools/build.sh | 1 + 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/lite/model_parser/flatbuffers/op_desc.h b/lite/model_parser/flatbuffers/op_desc.h index 32974f34d1..23bd00e8c0 100644 --- a/lite/model_parser/flatbuffers/op_desc.h +++ b/lite/model_parser/flatbuffers/op_desc.h @@ -171,8 +171,15 @@ class OpDescView : public OpDescAPI { } std::vector input_vars() const { - NotImplemented(); - return std::vector(); + VLOG(5) << "This function call is expensive."; + std::vector res; + for (const auto& var : *(desc_->inputs())) { + if (var && var->arguments()) { + res.emplace_back(var->arguments()->begin()->c_str(), + var->arguments()->end()->c_str()); + } + } + return res; } std::vector output_vars() const { diff --git a/lite/tools/build.sh b/lite/tools/build.sh index 4729440c00..6fc38180e7 100755 --- a/lite/tools/build.sh +++ b/lite/tools/build.sh @@ -436,6 +436,7 @@ function print_usage { echo -e "--build_python: (OFF|ON); controls whether to publish python api lib (ANDROID and IOS is not supported)" echo -e "--build_java: (OFF|ON); controls whether to publish java api lib (Only ANDROID is supported)" echo -e "--build_dir: directory for building" + echo -e "--enable_flatbuffers_view: (OFF|ON); Use the flatbuffers read-only view to load the model. If ON, the naive buffer will no longer be supported." echo echo -e "argument choices:" echo -e "--arm_os:\t android|ios|ios64" -- GitLab