6 #include "../OnnxParser.hpp" 8 #include <onnx/onnx.pb.h> 9 #include "google/protobuf/stubs/logging.h" 11 using ModelPtr = std::unique_ptr<onnx::ModelProto>;
17 explicit GetInputsOutputsMainFixture()
22 producer_version: "2.5.1" 43 name: "ActivationLayer" 72 CHECK_EQ(1, tensors.size());
73 CHECK_EQ(
"Input", tensors[0]);
81 CHECK_EQ(1, tensors.size());
82 CHECK_EQ(
"Output", tensors[0]);
87 GetEmptyInputsOutputsFixture()
91 producer_name: "CNTK " 92 producer_version: "2.5.1 " 143 CHECK_EQ(0, tensors.size());
146 TEST_CASE(
"GetInputsNullModel")
151 TEST_CASE(
"GetOutputsNullModel")
153 auto silencer = google::protobuf::LogSilencer();
159 GetInputsMultipleFixture() {
163 producer_name: "CNTK" 164 producer_version: "2.5.1" 247 CHECK_EQ(2, tensors.size());
248 CHECK_EQ(
"Input0", tensors[0]);
249 CHECK_EQ(
"Input1", tensors[1]);
static std::vector< std::string > GetInputs(ModelPtr &model)
Retrieve inputs names.
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
static std::vector< std::string > GetOutputs(ModelPtr &model)
Retrieve outputs names.
static ModelPtr LoadModelFromString(const std::string &inputString)