// Test "MaxPoolGraphBuilderImpl<ImportTarget::Canonical>"
{
- // TODO: fix indentation
- // clang-format off
-
- // what to test:
- // - there should exist MaxPool2D
- // - ifm node should be FeatureEncode
- // - following node should be FeatureDecode
- // - stride values should match
- // - window values should match
+ // what to test:
+ // - there should exist MaxPool2D
+ // - ifm node should be FeatureEncode
+ // - following node should be FeatureDecode
+ // - stride values should match
+ // - window values should match
- using MaxPoolGraphBuilder = MaxPoolGraphBuilderImpl<ImportTarget::Canonical>;
+ using MaxPoolGraphBuilder = MaxPoolGraphBuilderImpl<ImportTarget::Canonical>;
- moco::tf::GraphBuilderRegistry r{&moco::tf::GraphBuilderRegistry::get()};
- r.add("MaxPool", stdex::make_unique<MaxPoolGraphBuilder>());
- moco::tf::Importer importer{&r};
+ moco::tf::GraphBuilderRegistry r{&moco::tf::GraphBuilderRegistry::get()};
+ r.add("MaxPool", stdex::make_unique<MaxPoolGraphBuilder>());
+ moco::tf::Importer importer{&r};
- std::unique_ptr<loco::Graph> graph = importer.import(signature, graph_def);
+ std::unique_ptr<loco::Graph> graph = importer.import(signature, graph_def);
- loco::MaxPool2D *maxpool2d_node =
- moco::tf::test::find_first_node_bytype<loco::MaxPool2D>(graph.get());
- ASSERT_NE(maxpool2d_node, nullptr);
+ loco::MaxPool2D *maxpool2d_node =
+ moco::tf::test::find_first_node_bytype<loco::MaxPool2D>(graph.get());
+ ASSERT_NE(maxpool2d_node, nullptr);
- loco::Node *previous_node = maxpool2d_node->ifm();
- auto following_nodes = loco::succs(maxpool2d_node);
- ASSERT_EQ(following_nodes.size(), 1);
- loco::Node *following_node = *following_nodes.begin();
- ASSERT_NE(following_node, nullptr);
+ loco::Node *previous_node = maxpool2d_node->ifm();
+ auto following_nodes = loco::succs(maxpool2d_node);
+ ASSERT_EQ(following_nodes.size(), 1);
+ loco::Node *following_node = *following_nodes.begin();
+ ASSERT_NE(following_node, nullptr);
- loco::FeatureEncode *enc_node = dynamic_cast<loco::FeatureEncode *>(previous_node);
- loco::FeatureDecode *dec_node = dynamic_cast<loco::FeatureDecode *>(following_node);
+ loco::FeatureEncode *enc_node = dynamic_cast<loco::FeatureEncode *>(previous_node);
+ loco::FeatureDecode *dec_node = dynamic_cast<loco::FeatureDecode *>(following_node);
- ASSERT_NE(enc_node, nullptr);
- ASSERT_NE(dec_node, nullptr);
+ ASSERT_NE(enc_node, nullptr);
+ ASSERT_NE(dec_node, nullptr);
- // attrs inside MaxPool2D
- auto maxpool2d = maxpool2d_node; // TODO remove this new variable
+ // attrs inside MaxPool2D
+ auto maxpool2d = maxpool2d_node; // TODO remove this new variable
- // stride
- ASSERT_EQ(maxpool2d->stride()->vertical(), 3);
- ASSERT_EQ(maxpool2d->stride()->horizontal(), 2);
+ // stride
+ ASSERT_EQ(maxpool2d->stride()->vertical(), 3);
+ ASSERT_EQ(maxpool2d->stride()->horizontal(), 2);
- // window
- ASSERT_EQ(maxpool2d->window()->vertical(), 2);
- ASSERT_EQ(maxpool2d->window()->horizontal(), 3);
- // clang-format on
+ // window
+ ASSERT_EQ(maxpool2d->window()->vertical(), 2);
+ ASSERT_EQ(maxpool2d->window()->horizontal(), 3);
}
// Test "MaxPoolGraphBuilderImpl<ImportTarget::TensorFlow>"