6 #include <boost/test/unit_test.hpp> 14 ExpandDimsFixture(
const std::string& expandDim)
18 " name: \"graphInput\" \n" 19 " op: \"Placeholder\" \n" 35 " name: \"ExpandDims\" \n" 36 " op: \"ExpandDims\" \n" 37 " input: \"graphInput\" \n" 57 struct ExpandZeroDim : ExpandDimsFixture
59 ExpandZeroDim() : ExpandDimsFixture(
"0") {}
62 struct ExpandTwoDim : ExpandDimsFixture
64 ExpandTwoDim() : ExpandDimsFixture(
"2") {}
67 struct ExpandThreeDim : ExpandDimsFixture
69 ExpandThreeDim() : ExpandDimsFixture(
"3") {}
72 struct ExpandMinusOneDim : ExpandDimsFixture
74 ExpandMinusOneDim() : ExpandDimsFixture(
"-1") {}
77 struct ExpandMinusThreeDim : ExpandDimsFixture
79 ExpandMinusThreeDim() : ExpandDimsFixture(
"-3") {}
84 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"ExpandDims").second.GetShape() ==
90 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"ExpandDims").second.GetShape() ==
96 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"ExpandDims").second.GetShape() ==
102 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"ExpandDims").second.GetShape() ==
108 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"ExpandDims").second.GetShape() ==
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
BOOST_FIXTURE_TEST_CASE(ParseExpandZeroDim, ExpandZeroDim)
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)