6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 14 explicit EmptyNetworkFixture() {
27 BOOST_TEST(m_Parser->GetSubgraphCount() == 1); 28 BOOST_TEST(m_Parser->GetSubgraphInputTensorNames(0).size() == 0); 29 BOOST_TEST(m_Parser->GetSubgraphOutputTensorNames(0).size() == 0); 34 explicit MissingTensorsFixture()
56 explicit InvalidTensorsFixture()
61 "operator_codes": [ ], 64 "shape": [ 1, 1, 1, 1, 1, 1 ], 69 "shape": [ 1, 1, 1, 1, 1, 1 ], 90 explicit ValidTensorsFixture()
95 "operator_codes": [ { "builtin_code": "AVERAGE_POOL_2D" } ], 98 "shape": [ 1, 1, 1, 1 ], 103 "shape": [ 1, 1, 1, 1 ], 114 "builtin_options_type": "Pool2DOptions", 122 "fused_activation_function": "NONE" 124 "custom_options_format": "FLEXBUFFERS" 134 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0).size(), 1u); 135 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0).size(), 1u); 136 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0)[0], "In");
137 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0)[0],
"Out");
151 explicit Rank0TensorFixture()
156 "operator_codes": [ { "builtin_code": "MINIMUM" } ], 180 "custom_options_format": "FLEXBUFFERS" 191 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0).size(), 2u); 192 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0).size(), 1u); 193 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0)[0], "In0");
194 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0)[1],
"In1");
195 BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0)[0],
"Out");
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_AUTO_TEST_SUITE_END()
constexpr unsigned int MaxNumOfTensorDimensions