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() ==
114 ExpandDimsAsInputFixture(
const std::string& expandDim,
115 const bool wrongDataType =
false,
116 const std::string& numElements =
"1")
118 std::string dataType = (wrongDataType) ?
"DT_FLOAT" :
"DT_INT32";
119 std::string val = (wrongDataType) ? (
"float_val: " + expandDim +
".0") : (
"int_val: "+ expandDim);
151 type: )" + dataType + R"( 158 dtype: )" + dataType + R"( 161 size: )" + numElements + R"( 193 struct ExpandDimAsInput : ExpandDimsAsInputFixture
195 ExpandDimAsInput() : ExpandDimsAsInputFixture(
"0")
197 Setup({{
"a", {1,4}} ,{
"b",{1,1}}}, {
"ExpandDims" });
205 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(
"ExpandDims").second.GetShape() ==
209 struct ExpandDimAsInputWrongDataType : ExpandDimsAsInputFixture
211 ExpandDimAsInputWrongDataType() : ExpandDimsAsInputFixture(
"0",
true,
"1") {}
221 struct ExpandDimAsInputWrongShape : ExpandDimsAsInputFixture
223 ExpandDimAsInputWrongShape() : ExpandDimsAsInputFixture(
"0",
false,
"2") {}
235 ExpandDimsAsNotConstInputFixture()
309 BOOST_REQUIRE_THROW(Setup({{
"a", {1,4}} ,{
"b",{1,1}}}, {
"ExpandDims" }),
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
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.
BOOST_FIXTURE_TEST_CASE(ParseExpandZeroDim, ExpandZeroDim)