diff options
Diffstat (limited to 'src/armnnTfLiteParser/test/InputOutputTensorNames.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/InputOutputTensorNames.cpp | 67 |
1 files changed, 63 insertions, 4 deletions
diff --git a/src/armnnTfLiteParser/test/InputOutputTensorNames.cpp b/src/armnnTfLiteParser/test/InputOutputTensorNames.cpp index 6f5016265a..d42ae2e438 100644 --- a/src/armnnTfLiteParser/test/InputOutputTensorNames.cpp +++ b/src/armnnTfLiteParser/test/InputOutputTensorNames.cpp @@ -60,9 +60,19 @@ struct InvalidTensorsFixture : public ParserFlatbuffersFixture "version": 3, "operator_codes": [ ], "subgraphs": [{ - "tensors": [ {}, {}, {}, {} ], - "inputs" : [ 0, 1 ], - "outputs" : [ 2, 3 ], + "tensors": [ { + "shape": [ 1, 1, 1, 1, 1 ], + "type": "FLOAT32", + "name": "In", + "buffer": 0 + }, { + "shape": [ 1, 1, 1, 1, 1 ], + "type": "FLOAT32", + "name": "Out", + "buffer": 1 + }], + "inputs" : [ 0 ], + "outputs" : [ 1 ], }] })"; } @@ -70,7 +80,7 @@ struct InvalidTensorsFixture : public ParserFlatbuffersFixture BOOST_FIXTURE_TEST_CASE(InvalidTensorsThrowException, InvalidTensorsFixture) { - // this throws because it cannot do the input output tensor connections + // Tensor numDimensions must be less than or equal to MaxNumOfTensorDimensions BOOST_CHECK_THROW(Setup(), armnn::InvalidArgumentException); } @@ -135,4 +145,53 @@ BOOST_FIXTURE_TEST_CASE(ThrowIfSubgraphIdInvalidForInOutNames, ValidTensorsFixtu BOOST_CHECK_THROW(m_Parser->GetSubgraphOutputTensorNames(1), armnn::ParseException); } +struct Rank0TensorFixture : public ParserFlatbuffersFixture +{ + explicit Rank0TensorFixture() + { + m_JsonString = R"( + { + "version": 3, + "operator_codes": [ { "builtin_code": "MINIMUM" } ], + "subgraphs": [{ + "tensors": [ { + "shape": [ ], + "type": "FLOAT32", + "name": "In0", + "buffer": 0, + }, { + "shape": [ ], + "type": "FLOAT32", + "name": "In1", + "buffer": 1, + }, { + "shape": [ ], + "type": "FLOAT32", + "name": "Out", + "buffer": 2, + }], + "inputs" : [ 0, 1 ], + "outputs" : [ 2 ], + "operators": [{ + "opcode_index": 0, + "inputs": [ 0, 1 ], + "outputs": [ 2 ], + "custom_options_format": "FLEXBUFFERS" + }] + }] + } + )"; + } +}; + +BOOST_FIXTURE_TEST_CASE(Rank0Tensor, Rank0TensorFixture) +{ + Setup(); + BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0).size(), 2u); + BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0).size(), 1u); + BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0)[0], "In0"); + BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0)[1], "In1"); + BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0)[0], "Out"); +} + BOOST_AUTO_TEST_SUITE_END() |