[moco] Revise AvgPool Conv2D input names (#7989)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Tue, 8 Oct 2019 08:59:14 +0000 (17:59 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Tue, 8 Oct 2019 08:59:14 +0000 (17:59 +0900)
This will fix AvgPool and Conv2D input name to follow that from TensorFlow REGISTER_OP input names

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
compiler/moco/lang/include/moco/IR/Nodes/TFAvgPool.h
compiler/moco/lang/include/moco/IR/Nodes/TFConv2D.h
compiler/moco/lang/src/IR/Nodes/TFAvgPool.test.cpp
compiler/moco/lang/src/IR/Nodes/TFConv2D.test.cpp

index 65486a6..74c91b5 100644 (file)
@@ -73,8 +73,8 @@ public:
   TFAvgPool() = default;
 
 public:
-  Node *input(void) const { return at(0)->node(); }
-  void input(Node *node) { return at(0)->node(node); }
+  Node *value(void) const { return at(0)->node(); }
+  void value(Node *node) { return at(0)->node(node); }
 
 public:
   const TFDataLayout &data_layout(void) const { return _data_layout; }
index 8d7bd71..0d5a178 100644 (file)
@@ -30,8 +30,8 @@ public:
   loco::Node *input(void) const { return at(0)->node(); }
   void input(Node *node) { at(0)->node(node); }
 
-  loco::Node *filters(void) const { return at(1)->node(); }
-  void filters(Node *node) { at(1)->node(node); }
+  loco::Node *filter(void) const { return at(1)->node(); }
+  void filter(Node *node) { at(1)->node(node); }
 
 public:
   const TFPadding &padding(void) const { return _padding; }
index 3059855..32a27ff 100644 (file)
@@ -26,7 +26,7 @@ TEST(TFAvgPoolTest, constructor)
   ASSERT_EQ(avgpool.dialect(), moco::TFDialect::get());
   ASSERT_EQ(avgpool.opcode(), moco::TFOpcode::AvgPool);
 
-  ASSERT_EQ(avgpool.input(), nullptr);
+  ASSERT_EQ(avgpool.value(), nullptr);
   ASSERT_EQ(avgpool.data_layout(), "");
   ASSERT_EQ(avgpool.padding(), "");
   ASSERT_EQ(avgpool.ksize(), std::vector<int64_t>({}));
index 28a8133..3e3453d 100644 (file)
@@ -27,7 +27,7 @@ TEST(TFConv2DTest, constructor)
   ASSERT_EQ(conv2d_node.opcode(), moco::TFOpcode::Conv2D);
 
   ASSERT_EQ(conv2d_node.input(), nullptr);
-  ASSERT_EQ(conv2d_node.filters(), nullptr);
+  ASSERT_EQ(conv2d_node.filter(), nullptr);
   ASSERT_EQ(conv2d_node.padding(), "");
   ASSERT_EQ(conv2d_node.data_layout(), "");
   ASSERT_EQ(conv2d_node.strides().size(), 0);