#include <stdex/Memory.h>
-namespace moco
-{
-namespace tf
+namespace
{
-/**
- * @brief GraphBuilder for Relu6 node
- */
-class Relu6GraphBuilder final : public Relu6GraphBuilderBase
-{
-public:
- void build(const tensorflow::NodeDef &, GraphBuilderContext *) const override;
-};
+
+using namespace moco::tf;
class ReLU6GraphUpdate final : public GraphUpdate
{
const TensorName _name;
};
+void ReLU6GraphUpdate::input(const SymbolTable *table) const
+{
+ loco::Node *target = table->node(_name);
+ _node->input(target);
+}
+
+void TFRelu6GraphUpdate::input(const SymbolTable *table) const
+{
+ loco::Node *target = table->node(_name);
+ _node->features(target);
+}
+
+} // namespace
+
+namespace moco
+{
+namespace tf
+{
+/**
+ * @brief GraphBuilder for Relu6 node
+ */
+class Relu6GraphBuilder final : public Relu6GraphBuilderBase
+{
+public:
+ void build(const tensorflow::NodeDef &, GraphBuilderContext *) const override;
+};
+
bool Relu6GraphBuilderBase::validate(const tensorflow::NodeDef &node) const
{
// ReLU6 node SHOULD have only one input
updates->enroll(std::move(update));
}
-void ReLU6GraphUpdate::input(const SymbolTable *table) const
-{
- loco::Node *target = table->node(_name);
- _node->input(target);
-}
-
-void TFRelu6GraphUpdate::input(const SymbolTable *table) const
-{
- loco::Node *target = table->node(_name);
- _node->features(target);
-}
-
} // namespace tf
} // namespace moco