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>
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; }
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; }
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>({}));
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);