| // |
| // Copyright © 2017 Arm Ltd. All rights reserved. |
| // SPDX-License-Identifier: MIT |
| // |
| |
| #include "armnnOnnxParser/IOnnxParser.hpp" |
| #include "ParserPrototxtFixture.hpp" |
| |
| TEST_SUITE("OnnxParser_PrototxtFixture") |
| { |
| struct ProtoxtTestFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParser::IOnnxParser> |
| { |
| ProtoxtTestFixture() |
| { |
| m_Prototext = R"( |
| ir_version: 3 |
| producer_name: "CNTK " |
| producer_version: "2.5.1 " |
| domain: "ai.cntk " |
| model_version: 1 |
| graph { |
| name: "CNTKGraph " |
| node { |
| input: "Input" |
| output: "Output" |
| name: "Plus112" |
| op_type: "Add " |
| } |
| input { |
| name: "Input" |
| type { |
| tensor_type { |
| elem_type: 1 |
| shape { |
| dim { |
| dim_value: 2 |
| } |
| } |
| } |
| } |
| } |
| output { |
| name: "Output" |
| type { |
| tensor_type { |
| elem_type: 1 |
| shape { |
| dim { |
| dim_value: 1 |
| } |
| dim { |
| dim_value: 10 |
| } |
| } |
| } |
| } |
| } |
| } |
| opset_import { |
| version: 7 |
| })"; |
| // Setup(); |
| } |
| }; |
| |
| |
| TEST_CASE_FIXTURE(ProtoxtTestFixture, "ProtoxtTest") |
| { |
| //TODO : add a test to check if the inputs and outputs are correctly inferred. |
| } |
| |
| TEST_CASE_FIXTURE(ProtoxtTestFixture, "ProtoxtTestWithBadInputs") |
| { |
| |
| // CHECK_THROWS_AS(RunTest<4>({{ "InexistantInput" , {0.0, 1.0, 2.0, 3.0}}}, |
| // {{ "InexistantOutput" , {0.0, 1.0, 2.0, 3.0}}}), |
| // armnn::InvalidArgumentException ); |
| } |
| |
| } |