From fae3f4f1579ff14dc94f0da84b3e72da436af0bd Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Thu, 8 Aug 2019 15:56:18 +0900 Subject: [PATCH] [moco-tf] relocate MaxPool local class into anonymous (#6385) This will relocate local class of MaxPool import into anonymous namespace Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Op/MaxPool.cpp | 53 +++++++++++++++++++++---------------- 1 file changed, 30 insertions(+), 23 deletions(-) diff --git a/compiler/moco-tf/src/Op/MaxPool.cpp b/compiler/moco-tf/src/Op/MaxPool.cpp index e2ba665..079d914 100644 --- a/compiler/moco-tf/src/Op/MaxPool.cpp +++ b/compiler/moco-tf/src/Op/MaxPool.cpp @@ -36,19 +36,10 @@ #include #include -namespace moco -{ -namespace tf +namespace { -/** - * @brief GraphBuilder for MaxPool node - */ -class MaxPoolGraphBuilder final : public MaxPoolGraphBuilderBase -{ -public: - void build(const tensorflow::NodeDef &, GraphBuilderContext *) const override; -}; +using namespace moco::tf; class MaxPoolGraphUpdate final : public GraphUpdate { @@ -80,6 +71,34 @@ private: const TensorName _value_name; }; +void MaxPoolGraphUpdate::input(const SymbolTable *tensor_names) const +{ + loco::Node *input_node = tensor_names->node(_input_name); + _encode_node->input(input_node); +} + +void TFMaxPoolGraphUpdate::input(const SymbolTable *node_table) const +{ + loco::Node *value_node = node_table->node(_value_name); + _maxpool_node->value(value_node); +} + +} // namespace + +namespace moco +{ +namespace tf +{ + +/** + * @brief GraphBuilder for MaxPool node + */ +class MaxPoolGraphBuilder final : public MaxPoolGraphBuilderBase +{ +public: + void build(const tensorflow::NodeDef &, GraphBuilderContext *) const override; +}; + bool MaxPoolGraphBuilderBase::validate(const tensorflow::NodeDef &node) const { // note: even though "data_format" is not entered when a model is written, @@ -216,12 +235,6 @@ void MaxPoolGraphBuilderImpl::build(const tensorflow::N updates->enroll(std::move(update)); } -void MaxPoolGraphUpdate::input(const SymbolTable *tensor_names) const -{ - loco::Node *input_node = tensor_names->node(_input_name); - _encode_node->input(input_node); -} - void MaxPoolGraphBuilderImpl::build(const tensorflow::NodeDef &node, GraphBuilderContext *context) const { @@ -271,12 +284,6 @@ void MaxPoolGraphBuilderImpl::build(const tensorflow:: updates->enroll(std::move(update)); } -void TFMaxPoolGraphUpdate::input(const SymbolTable *node_table) const -{ - loco::Node *value_node = node_table->node(_value_name); - _maxpool_node->value(value_node); -} - } // namespace tf } // namespace moco -- 2.7.4