[moco/tf] Rename FusedBatchNorm Graph builder (#4119)

This will rename GraphBuilder for FusedBatchNorm with 'TF' prefix to match other GraphBuilders

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/contrib/moco-tf/src/Op/TFFusedBatchNorm.cpp b/contrib/moco-tf/src/Op/FusedBatchNorm.cpp
similarity index 76%
rename from contrib/moco-tf/src/Op/TFFusedBatchNorm.cpp
rename to contrib/moco-tf/src/Op/FusedBatchNorm.cpp
index 55cfadd..ece31ed 100644 (file)
--- a/contrib/moco-tf/src/Op/TFFusedBatchNorm.cpp
+++ b/contrib/moco-tf/src/Op/FusedBatchNorm.cpp
@@ -31,9 +31,9 @@ namespace tf
 {
 
 /**
- * @brief GraphBuilder for TF FusedBatchNorm node
+ * @brief GraphBuilder for FusedBatchNorm node
  */
-class TFFusedBatchNormGraphBuilder final : public GraphBuilder
+class FusedBatchNormGraphBuilder final : public GraphBuilder
 {
 public:
   bool validate(const tensorflow::NodeDef &) const override;
@@ -41,12 +41,12 @@ public:
 };
 
 /**
- * @brief GraphUpdate for TF FusedBatchNorm node
+ * @brief GraphUpdate for FusedBatchNorm node
  */
-class TFFusedBatchNormGraphUpdate final : public GraphUpdate
+class FusedBatchNormGraphUpdate final : public GraphUpdate
 {
 public:
-  TFFusedBatchNormGraphUpdate(TFFusedBatchNorm *node, std::vector<TensorName> names)
+  FusedBatchNormGraphUpdate(TFFusedBatchNorm *node, std::vector<TensorName> names)
       : _node(node), _names(names)
   {
   }
@@ -58,15 +58,15 @@ private:
   std::vector<TensorName> _names;
 };
 
-bool TFFusedBatchNormGraphBuilder::validate(const tensorflow::NodeDef &node) const
+bool FusedBatchNormGraphBuilder::validate(const tensorflow::NodeDef &node) const
 {
   assert(node.input_size() == 5);
 
   return has_attrs(node, {"epsilon"});
 }
 
-void TFFusedBatchNormGraphBuilder::build(const tensorflow::NodeDef &node,
-                                         GraphBuilderContext *context) const
+void FusedBatchNormGraphBuilder::build(const tensorflow::NodeDef &node,
+                                       GraphBuilderContext *context) const
 {
   assert(context != nullptr);
 
@@ -90,11 +90,11 @@ void TFFusedBatchNormGraphBuilder::build(const tensorflow::NodeDef &node,
   fbn_input_names.push_back(TensorName(node.input(3))); // mean
   fbn_input_names.push_back(TensorName(node.input(4))); // variance
 
-  auto tf_fbn_update = stdex::make_unique<TFFusedBatchNormGraphUpdate>(tf_fbn, fbn_input_names);
+  auto tf_fbn_update = stdex::make_unique<FusedBatchNormGraphUpdate>(tf_fbn, fbn_input_names);
   updates->enroll(std::move(tf_fbn_update));
 }
 
-void TFFusedBatchNormGraphUpdate::input(const SymbolTable *tensor_names) const
+void FusedBatchNormGraphUpdate::input(const SymbolTable *tensor_names) const
 {
   int num_inputs = _names.size();
   assert(num_inputs == 5);
@@ -111,4 +111,4 @@ void TFFusedBatchNormGraphUpdate::input(const SymbolTable *tensor_names) const
 
 #include "GraphBuilderRegistry.h"
 
-REGISTER_OP_BUILDER(FusedBatchNorm, TFFusedBatchNormGraphBuilder)
+REGISTER_OP_BUILDER(FusedBatchNorm, FusedBatchNormGraphBuilder)

diff --git a/contrib/moco-tf/src/Op/TFFusedBatchNorm.test.cpp b/contrib/moco-tf/src/Op/FusedBatchNorm.test.cpp
similarity index 100%
rename from contrib/moco-tf/src/Op/TFFusedBatchNorm.test.cpp
rename to contrib/moco-tf/src/Op/FusedBatchNorm.test.cpp