5 #include <boost/test/unit_test.hpp> 13 GlobalPoolingFixture()
20 " input_param { shape: { dim: 1 dim: 3 dim: 2 dim: 2 } }\n" 26 " type: \"Pooling\"\n" 29 " global_pooling: true\n" BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(GlobalPooling, GlobalPoolingFixture)
BOOST_AUTO_TEST_SUITE_END()
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
Parses and loads the network defined by the m_Prototext string.