5 #include <boost/test/unit_test.hpp> 6 #include "../OnnxParser.hpp" 8 #include <onnx/onnx.pb.h> 9 #include "google/protobuf/stubs/logging.h" 12 using ModelPtr = std::unique_ptr<onnx::ModelProto>;
18 explicit GetInputsOutputsMainFixture()
23 producer_version: "2.5.1" 44 name: "ActivationLayer" 73 BOOST_CHECK_EQUAL(1, tensors.size()); 74 BOOST_CHECK_EQUAL("Input", tensors[0]);
82 BOOST_CHECK_EQUAL(1, tensors.size());
83 BOOST_CHECK_EQUAL(
"Output", tensors[0]);
88 GetEmptyInputsOutputsFixture()
92 producer_name: "CNTK " 93 producer_version: "2.5.1 " 144 BOOST_CHECK_EQUAL(0, tensors.size()); 154 auto silencer = google::protobuf::LogSilencer();
160 GetInputsMultipleFixture() {
164 producer_name: "CNTK" 165 producer_version: "2.5.1" 248 BOOST_CHECK_EQUAL(2, tensors.size()); 249 BOOST_CHECK_EQUAL("Input0", tensors[0]);
250 BOOST_CHECK_EQUAL(
"Input1", tensors[1]);
static std::vector< std::string > GetOutputs(ModelPtr &model)
Retrieve outputs names.
static ModelPtr LoadModelFromString(const std::string &inputString)
static std::vector< std::string > GetInputs(ModelPtr &model)
Retrieve inputs names.
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)