ArmNN
 20.05
GetInputsOutputs.cpp File Reference
#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"

Go to the source code of this file.

Typedefs

using ModelPtr = TfLiteParser::ModelPtr
 

Functions

 BOOST_FIXTURE_TEST_CASE (GetEmptyInputs, GetEmptyInputsOutputsFixture)
 
 BOOST_FIXTURE_TEST_CASE (GetEmptyOutputs, GetEmptyInputsOutputsFixture)
 
 BOOST_FIXTURE_TEST_CASE (GetInputs, GetInputsOutputsFixture)
 
 BOOST_FIXTURE_TEST_CASE (GetOutputs, GetInputsOutputsFixture)
 
 BOOST_FIXTURE_TEST_CASE (GetInputsMultipleInputs, GetInputsOutputsFixture)
 
 BOOST_FIXTURE_TEST_CASE (GetOutputs2, GetInputsOutputsFixture)
 
 BOOST_AUTO_TEST_CASE (GetInputsNullModel)
 
 BOOST_AUTO_TEST_CASE (GetOutputsNullModel)
 
 BOOST_FIXTURE_TEST_CASE (GetInputsInvalidSubgraph, GetInputsOutputsFixture)
 
 BOOST_FIXTURE_TEST_CASE (GetOutputsInvalidSubgraph, GetInputsOutputsFixture)
 
 BOOST_FIXTURE_TEST_CASE (GetInputsInvalidOperator, GetInputsOutputsFixture)
 
 BOOST_FIXTURE_TEST_CASE (GetOutputsInvalidOperator, GetInputsOutputsFixture)
 

Typedef Documentation

◆ ModelPtr

using ModelPtr = TfLiteParser::ModelPtr

Definition at line 10 of file GetInputsOutputs.cpp.

Function Documentation

◆ BOOST_AUTO_TEST_CASE() [1/2]

BOOST_AUTO_TEST_CASE ( GetInputsNullModel  )

Definition at line 205 of file GetInputsOutputs.cpp.

206 {
207  BOOST_CHECK_THROW(TfLiteParser::GetInputs(nullptr, 0, 0), armnn::ParseException);
208 }

◆ BOOST_AUTO_TEST_CASE() [2/2]

BOOST_AUTO_TEST_CASE ( GetOutputsNullModel  )

Definition at line 210 of file GetInputsOutputs.cpp.

211 {
212  BOOST_CHECK_THROW(TfLiteParser::GetOutputs(nullptr, 0, 0), armnn::ParseException);
213 }

◆ BOOST_FIXTURE_TEST_CASE() [1/10]

BOOST_FIXTURE_TEST_CASE ( GetEmptyInputs  ,
GetEmptyInputsOutputsFixture   
)

Definition at line 153 of file GetInputsOutputs.cpp.

154 {
155  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
156  TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetInputs(model, 0, 0);
157  BOOST_CHECK_EQUAL(0, tensors.size());
158 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [2/10]

BOOST_FIXTURE_TEST_CASE ( GetEmptyOutputs  ,
GetEmptyInputsOutputsFixture   
)

Definition at line 160 of file GetInputsOutputs.cpp.

161 {
162  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
163  TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetOutputs(model, 0, 0);
164  BOOST_CHECK_EQUAL(0, tensors.size());
165 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [3/10]

BOOST_FIXTURE_TEST_CASE ( GetInputs  ,
GetInputsOutputsFixture   
)

Definition at line 167 of file GetInputsOutputs.cpp.

168 {
169  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
170  TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetInputs(model, 0, 0);
171  BOOST_CHECK_EQUAL(1, tensors.size());
172  CheckTensors(tensors[0], 4, { 1, 2, 2, 1 }, tflite::TensorType::TensorType_UINT8, 1,
173  "InputTensor", { -1.2f }, { 25.5f }, { 0.25f }, { 10 });
174 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [4/10]

BOOST_FIXTURE_TEST_CASE ( GetOutputs  ,
GetInputsOutputsFixture   
)

Definition at line 176 of file GetInputsOutputs.cpp.

177 {
178  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
179  TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetOutputs(model, 0, 0);
180  BOOST_CHECK_EQUAL(1, tensors.size());
181  CheckTensors(tensors[0], 4, { 1, 1, 1, 1 }, tflite::TensorType::TensorType_UINT8, 0,
182  "OutputTensor", { 0.0f }, { 255.0f }, { 1.0f }, { 0 });
183 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [5/10]

BOOST_FIXTURE_TEST_CASE ( GetInputsMultipleInputs  ,
GetInputsOutputsFixture   
)

Definition at line 185 of file GetInputsOutputs.cpp.

186 {
187  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
188  TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetInputs(model, 1, 0);
189  BOOST_CHECK_EQUAL(2, tensors.size());
190  CheckTensors(tensors[0], 4, { 1, 3, 3, 1 }, tflite::TensorType::TensorType_UINT8, 0,
191  "ConvInputTensor", { }, { }, { 1.0f }, { 0 });
192  CheckTensors(tensors[1], 4, { 1, 3, 3, 1 }, tflite::TensorType::TensorType_UINT8, 2,
193  "filterTensor", { 0.0f }, { 255.0f }, { 1.0f }, { 0 });
194 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [6/10]

BOOST_FIXTURE_TEST_CASE ( GetOutputs2  ,
GetInputsOutputsFixture   
)

Definition at line 196 of file GetInputsOutputs.cpp.

197 {
198  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
199  TfLiteParser::TensorRawPtrVector tensors = TfLiteParser::GetOutputs(model, 1, 0);
200  BOOST_CHECK_EQUAL(1, tensors.size());
201  CheckTensors(tensors[0], 4, { 1, 1, 1, 1 }, tflite::TensorType::TensorType_UINT8, 1,
202  "ConvOutputTensor", { 0.0f }, { 511.0f }, { 2.0f }, { 0 });
203 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [7/10]

BOOST_FIXTURE_TEST_CASE ( GetInputsInvalidSubgraph  ,
GetInputsOutputsFixture   
)

Definition at line 215 of file GetInputsOutputs.cpp.

216 {
217  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
218  BOOST_CHECK_THROW(TfLiteParser::GetInputs(model, 2, 0), armnn::ParseException);
219 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [8/10]

BOOST_FIXTURE_TEST_CASE ( GetOutputsInvalidSubgraph  ,
GetInputsOutputsFixture   
)

Definition at line 221 of file GetInputsOutputs.cpp.

222 {
223  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
224  BOOST_CHECK_THROW(TfLiteParser::GetOutputs(model, 2, 0), armnn::ParseException);
225 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [9/10]

BOOST_FIXTURE_TEST_CASE ( GetInputsInvalidOperator  ,
GetInputsOutputsFixture   
)

Definition at line 227 of file GetInputsOutputs.cpp.

228 {
229  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
230  BOOST_CHECK_THROW(TfLiteParser::GetInputs(model, 0, 1), armnn::ParseException);
231 }
std::unique_ptr< onnx::ModelProto > ModelPtr

◆ BOOST_FIXTURE_TEST_CASE() [10/10]

BOOST_FIXTURE_TEST_CASE ( GetOutputsInvalidOperator  ,
GetInputsOutputsFixture   
)

Definition at line 233 of file GetInputsOutputs.cpp.

References BOOST_AUTO_TEST_SUITE_END().

234 {
235  TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
236  BOOST_CHECK_THROW(TfLiteParser::GetOutputs(model, 0, 1), armnn::ParseException);
237 }
std::unique_ptr< onnx::ModelProto > ModelPtr