6 #include <boost/test/unit_test.hpp> 12 template <
bool withDimZero,
bool withDimOne>
19 " name: \"graphInput\" \n" 20 " op: \"Placeholder\" \n" 36 " name: \"Squeeze\" \n" 38 " input: \"graphInput\" \n" 46 " key: \"squeeze_dims\" \n" 77 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"Squeeze").second.GetShape() ==
79 RunTest<2>({ 1.0f, 2.0f, 3.0f, 4.0f },
80 { 1.0f, 2.0f, 3.0f, 4.0f });
85 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"Squeeze").second.GetShape() ==
87 RunTest<3>({ 1.0f, 2.0f, 3.0f, 4.0f },
88 { 1.0f, 2.0f, 3.0f, 4.0f });
93 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"Squeeze").second.GetShape() ==
95 RunTest<3>({ 1.0f, 2.0f, 3.0f, 4.0f },
96 { 1.0f, 2.0f, 3.0f, 4.0f });
101 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"Squeeze").second.GetShape() ==
103 RunTest<2>({ 1.0f, 2.0f, 3.0f, 4.0f },
104 { 1.0f, 2.0f, 3.0f, 4.0f });
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
SqueezeFixture< true, true > ExplicitDimensionsSqueezeFixture
BOOST_FIXTURE_TEST_CASE(ParseSqueezeWithSqueezeDims, SqueezeFixtureWithSqueezeDims)
SqueezeFixture< false, false > ImpliedDimensionsSqueezeFixture
SqueezeFixture< true, false > ExplicitDimensionZeroSqueezeFixture
SqueezeFixture< false, true > ExplicitDimensionOneSqueezeFixture
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.