5 #include <boost/test/unit_test.hpp> 7 #include "../TfLiteParser.hpp" 18 explicit LoadModelFixture()
23 "operator_codes": [ { "builtin_code": "AVERAGE_POOL_2D" }, { "builtin_code": "CONV_2D" } ], 28 "shape": [ 1, 1, 1, 1 ] , 31 "name": "OutputTensor", 40 "shape": [ 1, 2, 2, 1 ] , 43 "name": "InputTensor", 58 "builtin_options_type": "Pool2DOptions", 66 "fused_activation_function": "NONE" 68 "custom_options_format": "FLEXBUFFERS" 74 "shape": [ 1, 3, 3, 1 ], 77 "name": "ConvInputTensor", 84 "shape": [ 1, 1, 1, 1 ], 87 "name": "ConvOutputTensor", 96 "shape": [ 1, 3, 3, 1 ], 99 "name": "filterTensor", 115 "builtin_options_type": "Conv2DOptions", 120 "fused_activation_function": "NONE" 122 "custom_options_format": "FLEXBUFFERS" 127 "description": "Test loading a model", 128 "buffers" : [ {}, {} ] 134 void CheckModel(
const ModelPtr& model, uint32_t version,
size_t opcodeSize,
135 const std::vector<tflite::BuiltinOperator>& opcodes,
136 size_t subgraphs,
const std::string desc,
size_t buffers)
139 BOOST_CHECK_EQUAL(version, model->version);
140 BOOST_CHECK_EQUAL(opcodeSize, model->operator_codes.size());
141 CheckBuiltinOperators(opcodes, model->operator_codes);
142 BOOST_CHECK_EQUAL(subgraphs, model->subgraphs.size());
143 BOOST_CHECK_EQUAL(desc, model->description);
144 BOOST_CHECK_EQUAL(buffers, model->buffers.size());
147 void CheckBuiltinOperators(
const std::vector<tflite::BuiltinOperator>& expectedOperators,
148 const std::vector<std::unique_ptr<tflite::OperatorCodeT>>& result)
150 BOOST_CHECK_EQUAL(expectedOperators.size(), result.size());
151 for (
size_t i = 0; i < expectedOperators.size(); i++)
153 BOOST_CHECK_EQUAL(expectedOperators[i], result[i]->builtin_code);
157 void CheckSubgraph(
const SubgraphPtr& subgraph,
size_t tensors,
const std::vector<int32_t>& inputs,
158 const std::vector<int32_t>& outputs,
size_t operators,
const std::string& name)
161 BOOST_CHECK_EQUAL(tensors, subgraph->tensors.size());
162 BOOST_CHECK_EQUAL_COLLECTIONS(inputs.begin(), inputs.end(), subgraph->inputs.begin(), subgraph->inputs.end());
163 BOOST_CHECK_EQUAL_COLLECTIONS(outputs.begin(), outputs.end(),
164 subgraph->outputs.begin(), subgraph->outputs.end());
165 BOOST_CHECK_EQUAL(operators, subgraph->operators.size());
166 BOOST_CHECK_EQUAL(name, subgraph->name);
169 void CheckOperator(
const OperatorPtr& operatorPtr, uint32_t opcode,
const std::vector<int32_t>& inputs,
170 const std::vector<int32_t>& outputs, tflite::BuiltinOptions optionType,
171 tflite::CustomOptionsFormat custom_options_format)
174 BOOST_CHECK_EQUAL(opcode, operatorPtr->opcode_index);
175 BOOST_CHECK_EQUAL_COLLECTIONS(inputs.begin(), inputs.end(),
176 operatorPtr->inputs.begin(), operatorPtr->inputs.end());
177 BOOST_CHECK_EQUAL_COLLECTIONS(outputs.begin(), outputs.end(),
178 operatorPtr->outputs.begin(), operatorPtr->outputs.end());
179 BOOST_CHECK_EQUAL(optionType, operatorPtr->builtin_options.type);
180 BOOST_CHECK_EQUAL(custom_options_format, operatorPtr->custom_options_format);
186 TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
187 CheckModel(model, 3, 2, { tflite::BuiltinOperator_AVERAGE_POOL_2D, tflite::BuiltinOperator_CONV_2D },
188 2,
"Test loading a model", 2);
189 CheckSubgraph(model->subgraphs[0], 2, { 1 }, { 0 }, 1,
"");
190 CheckSubgraph(model->subgraphs[1], 3, { 0 }, { 1 }, 1,
"");
191 CheckOperator(model->subgraphs[0]->operators[0], 0, { 1 }, { 0 }, tflite::BuiltinOptions_Pool2DOptions,
192 tflite::CustomOptionsFormat_FLEXBUFFERS);
193 CheckOperator(model->subgraphs[1]->operators[0], 1, { 0, 2 }, { 1 }, tflite::BuiltinOptions_Conv2DOptions,
194 tflite::CustomOptionsFormat_FLEXBUFFERS);
199 using namespace boost::filesystem;
200 std::string fname = unique_path(temp_directory_path() /
"%%%%-%%%%-%%%%.tflite").string();
201 bool saved = flatbuffers::SaveFile(fname.c_str(),
202 reinterpret_cast<char *
>(m_GraphBinary.data()),
203 m_GraphBinary.size(),
true);
204 BOOST_CHECK_MESSAGE(saved,
"Cannot save test file");
207 CheckModel(model, 3, 2, { tflite::BuiltinOperator_AVERAGE_POOL_2D, tflite::BuiltinOperator_CONV_2D },
208 2,
"Test loading a model", 2);
209 CheckSubgraph(model->subgraphs[0], 2, { 1 }, { 0 }, 1,
"");
210 CheckSubgraph(model->subgraphs[1], 3, { 0 }, { 1 }, 1,
"");
211 CheckOperator(model->subgraphs[0]->operators[0], 0, { 1 }, { 0 }, tflite::BuiltinOptions_Pool2DOptions,
212 tflite::CustomOptionsFormat_FLEXBUFFERS);
213 CheckOperator(model->subgraphs[1]->operators[0], 1, { 0, 2 }, { 1 }, tflite::BuiltinOptions_Conv2DOptions,
214 tflite::CustomOptionsFormat_FLEXBUFFERS);
225 std::string testData =
"invalid data";
226 BOOST_CHECK_THROW(TfLiteParser::LoadModelFromBinary(reinterpret_cast<const uint8_t*>(&testData),
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
TfLiteParser::ModelPtr ModelPtr
TfLiteParser::SubgraphPtr SubgraphPtr
bool ReadStringToBinary()
BOOST_CHECK(profilingService.GetCurrentState()==ProfilingState::WaitingForAck)
BOOST_AUTO_TEST_CASE(LoadNullBinary)
TfLiteParser::OperatorPtr OperatorPtr
BOOST_AUTO_TEST_SUITE_END()
BOOST_FIXTURE_TEST_CASE(LoadModelFromBinary, LoadModelFixture)