6 #include <boost/test/unit_test.hpp> 14 explicit Pooling2dFixture(
const char* poolingtype, std::string dataLayout, std::string paddingOption)
17 " name: \"Placeholder\"\n" 18 " op: \"Placeholder\"\n" 43 " input: \"Placeholder\"\n" 51 " key: \"data_format\"\n" 64 if(dataLayout ==
"NHWC")
102 if(dataLayout ==
"NHWC")
114 struct MaxPoolFixtureNhwcValid : Pooling2dFixture
116 MaxPoolFixtureNhwcValid() : Pooling2dFixture(
"MaxPool",
"NHWC",
"VALID") {}
120 RunTest<4>({1.0f, 2.0f, 3.0f, -4.0f}, {3.0f});
123 struct MaxPoolFixtureNchwValid : Pooling2dFixture
125 MaxPoolFixtureNchwValid() : Pooling2dFixture(
"MaxPool",
"NCHW",
"VALID") {}
129 RunTest<4>({1.0f, 2.0f, 3.0f, -4.0f}, {3.0f});
132 struct MaxPoolFixtureNhwcSame : Pooling2dFixture
134 MaxPoolFixtureNhwcSame() : Pooling2dFixture(
"MaxPool",
"NHWC",
"SAME") {}
138 RunTest<4>({1.0f, 2.0f, 3.0f, -4.0f}, {3.0f, 2.0f, 3.0f, -4.0f});
141 struct MaxPoolFixtureNchwSame : Pooling2dFixture
143 MaxPoolFixtureNchwSame() : Pooling2dFixture(
"MaxPool",
"NCHW",
"SAME") {}
147 RunTest<4>({1.0f, 2.0f, 3.0f, -4.0f}, {3.0f, 2.0f, 3.0f, -4.0f});
150 struct AvgPoolFixtureNhwcValid : Pooling2dFixture
152 AvgPoolFixtureNhwcValid() : Pooling2dFixture(
"AvgPool",
"NHWC",
"VALID") {}
156 RunTest<4>({1.0f, 2.0f, 3.0f, 4.0f}, {2.5f});
159 struct AvgPoolFixtureNchwValid : Pooling2dFixture
161 AvgPoolFixtureNchwValid() : Pooling2dFixture(
"AvgPool",
"NCHW",
"VALID") {}
165 RunTest<4>({1.0f, 2.0f, 3.0f, 4.0f}, {2.5f});
168 struct AvgPoolFixtureNhwcSame : Pooling2dFixture
170 AvgPoolFixtureNhwcSame() : Pooling2dFixture(
"AvgPool",
"NHWC",
"SAME") {}
174 RunTest<4>({1.0f, 2.0f, 3.0f, 4.0f}, {2.5f, 3.0f, 3.5f, 4.0f});
177 struct AvgPoolFixtureNchwSame : Pooling2dFixture
179 AvgPoolFixtureNchwSame() : Pooling2dFixture(
"AvgPool",
"NCHW",
"SAME") {}
183 RunTest<4>({1.0f, 2.0f, 3.0f, 4.0f}, {2.5f, 3.0f, 3.5f, 4.0f});
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
BOOST_FIXTURE_TEST_CASE(ValidMaxPoolTest, MaxPoolValidFixture)
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)