5 #include <boost/test/unit_test.hpp> 7 #include "../TfLiteParser.hpp" 20 explicit LoadModelFixture()
25 "operator_codes": [ { "builtin_code": "AVERAGE_POOL_2D" }, { "builtin_code": "CONV_2D" } ], 30 "shape": [ 1, 1, 1, 1 ] , 33 "name": "OutputTensor", 42 "shape": [ 1, 2, 2, 1 ] , 45 "name": "InputTensor", 60 "builtin_options_type": "Pool2DOptions", 68 "fused_activation_function": "NONE" 70 "custom_options_format": "FLEXBUFFERS" 76 "shape": [ 1, 3, 3, 1 ], 79 "name": "ConvInputTensor", 86 "shape": [ 1, 1, 1, 1 ], 89 "name": "ConvOutputTensor", 98 "shape": [ 1, 3, 3, 1 ], 101 "name": "filterTensor", 117 "builtin_options_type": "Conv2DOptions", 122 "fused_activation_function": "NONE" 124 "custom_options_format": "FLEXBUFFERS" 129 "description": "Test loading a model", 130 "buffers" : [ {}, {} ] 136 void CheckModel(
const ModelPtr& model, uint32_t version,
size_t opcodeSize,
137 const std::vector<tflite::BuiltinOperator>& opcodes,
138 size_t subgraphs,
const std::string desc,
size_t buffers)
141 BOOST_CHECK_EQUAL(version, model->version);
142 BOOST_CHECK_EQUAL(opcodeSize, model->operator_codes.size());
143 CheckBuiltinOperators(opcodes, model->operator_codes);
144 BOOST_CHECK_EQUAL(subgraphs, model->subgraphs.size());
145 BOOST_CHECK_EQUAL(desc, model->description);
146 BOOST_CHECK_EQUAL(buffers, model->buffers.size());
149 void CheckBuiltinOperators(
const std::vector<tflite::BuiltinOperator>& expectedOperators,
150 const std::vector<std::unique_ptr<tflite::OperatorCodeT>>& result)
152 BOOST_CHECK_EQUAL(expectedOperators.size(), result.size());
153 for (
size_t i = 0; i < expectedOperators.size(); i++)
155 BOOST_CHECK_EQUAL(expectedOperators[i], result[i]->builtin_code);
159 void CheckSubgraph(
const SubgraphPtr& subgraph,
size_t tensors,
const std::vector<int32_t>& inputs,
160 const std::vector<int32_t>& outputs,
size_t operators,
const std::string& name)
162 BOOST_CHECK(subgraph);
163 BOOST_CHECK_EQUAL(tensors, subgraph->tensors.size());
164 BOOST_CHECK_EQUAL_COLLECTIONS(inputs.begin(), inputs.end(), subgraph->inputs.begin(), subgraph->inputs.end());
165 BOOST_CHECK_EQUAL_COLLECTIONS(outputs.begin(), outputs.end(),
166 subgraph->outputs.begin(), subgraph->outputs.end());
167 BOOST_CHECK_EQUAL(operators, subgraph->operators.size());
168 BOOST_CHECK_EQUAL(name, subgraph->name);
171 void CheckOperator(
const OperatorPtr& operatorPtr, uint32_t opcode,
const std::vector<int32_t>& inputs,
172 const std::vector<int32_t>& outputs, tflite::BuiltinOptions optionType,
173 tflite::CustomOptionsFormat custom_options_format)
175 BOOST_CHECK(operatorPtr);
176 BOOST_CHECK_EQUAL(opcode, operatorPtr->opcode_index);
177 BOOST_CHECK_EQUAL_COLLECTIONS(inputs.begin(), inputs.end(),
178 operatorPtr->inputs.begin(), operatorPtr->inputs.end());
179 BOOST_CHECK_EQUAL_COLLECTIONS(outputs.begin(), outputs.end(),
180 operatorPtr->outputs.begin(), operatorPtr->outputs.end());
181 BOOST_CHECK_EQUAL(optionType, operatorPtr->builtin_options.type);
182 BOOST_CHECK_EQUAL(custom_options_format, operatorPtr->custom_options_format);
189 m_GraphBinary.size());
190 CheckModel(model, 3, 2, { tflite::BuiltinOperator_AVERAGE_POOL_2D, tflite::BuiltinOperator_CONV_2D },
191 2,
"Test loading a model", 2);
192 CheckSubgraph(model->subgraphs[0], 2, { 1 }, { 0 }, 1,
"");
193 CheckSubgraph(model->subgraphs[1], 3, { 0 }, { 1 }, 1,
"");
194 CheckOperator(model->subgraphs[0]->operators[0], 0, { 1 }, { 0 }, tflite::BuiltinOptions_Pool2DOptions,
195 tflite::CustomOptionsFormat_FLEXBUFFERS);
196 CheckOperator(model->subgraphs[1]->operators[0], 1, { 0, 2 }, { 1 }, tflite::BuiltinOptions_Conv2DOptions,
197 tflite::CustomOptionsFormat_FLEXBUFFERS);
204 bool saved = flatbuffers::SaveFile(fname.c_str(),
205 reinterpret_cast<char *
>(m_GraphBinary.data()),
206 m_GraphBinary.size(),
true);
207 BOOST_CHECK_MESSAGE(saved,
"Cannot save test file");
210 CheckModel(model, 3, 2, { tflite::BuiltinOperator_AVERAGE_POOL_2D, tflite::BuiltinOperator_CONV_2D },
211 2,
"Test loading a model", 2);
212 CheckSubgraph(model->subgraphs[0], 2, { 1 }, { 0 }, 1,
"");
213 CheckSubgraph(model->subgraphs[1], 3, { 0 }, { 1 }, 1,
"");
214 CheckOperator(model->subgraphs[0]->operators[0], 0, { 1 }, { 0 }, tflite::BuiltinOptions_Pool2DOptions,
215 tflite::CustomOptionsFormat_FLEXBUFFERS);
216 CheckOperator(model->subgraphs[1]->operators[0], 1, { 0, 2 }, { 1 }, tflite::BuiltinOptions_Conv2DOptions,
217 tflite::CustomOptionsFormat_FLEXBUFFERS);
228 std::string testData =
"invalid data";
229 BOOST_CHECK_THROW(TfLiteParserImpl::LoadModelFromBinary(reinterpret_cast<const uint8_t*>(&testData),
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
TfLiteParserImpl::OperatorPtr OperatorPtr
bool ReadStringToBinary()
TfLiteParserImpl::SubgraphPtr SubgraphPtr
BOOST_AUTO_TEST_CASE(LoadNullBinary)
BOOST_AUTO_TEST_SUITE_END()
fs::path NamedTempFile(const char *fileName)
Construct a temporary file name.
BOOST_FIXTURE_TEST_CASE(LoadModelFromBinary, LoadModelFixture)
TfLiteParserImpl::ModelPtr ModelPtr