10 #include <boost/test/unit_test.hpp> 96 RunTest<4>({ 1.0f, 2.0f, 3.0f, 4.0f }, { 2.0f, 4.0f, 6.0f, 8.0f });
103 auto optimizedNetwork = boost::polymorphic_downcast<armnn::OptimizedNetwork*>(optimized.get());
104 auto graph = optimizedNetwork->GetGraph();
106 BOOST_TEST((graph.GetNumInputs() == 1));
107 BOOST_TEST((graph.GetNumOutputs() == 1));
108 BOOST_TEST((graph.GetNumLayers() == 3));
121 BOOST_TEST(
IsConnected(inputLayer, addLayer, 0, 0, tensorInfo));
122 BOOST_TEST(
IsConnected(inputLayer, addLayer, 0, 1, tensorInfo));
124 for (
auto&& outputLayer : graph.GetOutputLayers())
126 BOOST_TEST(
IsConnected(addLayer, const_cast<armnn::OutputLayer*>(outputLayer), 0, 0, tensorInfo));
239 Setup({ { "Input0", { 1, 1, 2, 2 } },
240 {
"Input1", { 1, 1, 2, 2 } } },
243 RunTest<4>({ {
"Input0", { 4.0f, 3.0f,
246 {
"Input1", { 1.0f, 2.0f,
249 { {
"Output", { 7.0f, 4.0f,
255 auto optimized = SetupOptimizedNetwork({ {
"Input0", { 1, 1, 2, 2 } },
256 {
"Input1", { 1, 1, 2, 2 } } },
259 auto optimizedNetwork = boost::polymorphic_downcast<armnn::OptimizedNetwork*>(optimized.get());
260 auto graph = optimizedNetwork->GetGraph();
262 BOOST_TEST((graph.GetNumInputs() == 2));
263 BOOST_TEST((graph.GetNumOutputs() == 1));
264 BOOST_TEST((graph.GetNumLayers() == 5));
287 BOOST_TEST(
IsConnected(inputLayer0, subLayer, 0, 0, tensorInfo));
288 BOOST_TEST(
IsConnected(inputLayer1, subLayer, 0, 1, tensorInfo));
289 BOOST_TEST(
IsConnected(inputLayer0, addLayer, 0, 0, tensorInfo));
290 BOOST_TEST(
IsConnected(subLayer, addLayer, 0, 1, tensorInfo));
292 for (
auto&& outputLayer : graph.GetOutputLayers())
294 BOOST_TEST(
IsConnected(addLayer, const_cast<armnn::OutputLayer*>(outputLayer), 0, 0, tensorInfo));
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
armnn::Layer * GetFirstLayerWithName(armnn::Graph &graph, const std::string &name)
BOOST_FIXTURE_TEST_CASE(AssertSimpleTest, AssertSimpleFixture)
armnn::IOptimizedNetworkPtr SetupOptimizedNetwork(const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs)
bool IsConnected(armnn::Layer *srcLayer, armnn::Layer *destLayer, unsigned int srcSlot, unsigned int destSlot, const armnn::TensorInfo &expectedTensorInfo)
BOOST_AUTO_TEST_SUITE_END()
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
Parses and loads the network defined by the m_Prototext string.
LayerType GetType() const
bool CheckNumberOfInputSlot(armnn::Layer *layer, unsigned int num)
bool CheckNumberOfOutputSlot(armnn::Layer *layer, unsigned int num)