12 #include <boost/test/unit_test.hpp> 98 RunTest<4>({ 1.0f, 2.0f, 3.0f, 4.0f }, { 2.0f, 4.0f, 6.0f, 8.0f });
105 auto optimizedNetwork = armnn::PolymorphicDowncast<armnn::OptimizedNetwork*>(optimized.get());
106 auto graph = optimizedNetwork->GetGraph();
108 BOOST_TEST((graph.GetNumInputs() == 1));
109 BOOST_TEST((graph.GetNumOutputs() == 1));
110 BOOST_TEST((graph.GetNumLayers() == 3));
123 BOOST_TEST(
IsConnected(inputLayer, addLayer, 0, 0, tensorInfo));
124 BOOST_TEST(
IsConnected(inputLayer, addLayer, 0, 1, tensorInfo));
126 for (
auto&& outputLayer : graph.GetOutputLayers())
128 BOOST_TEST(
IsConnected(addLayer, const_cast<armnn::OutputLayer*>(outputLayer), 0, 0, tensorInfo));
241 Setup({ { "Input0", { 1, 1, 2, 2 } },
242 {
"Input1", { 1, 1, 2, 2 } } },
245 RunTest<4>({ {
"Input0", { 4.0f, 3.0f,
248 {
"Input1", { 1.0f, 2.0f,
251 { {
"Output", { 7.0f, 4.0f,
257 auto optimized = SetupOptimizedNetwork({ {
"Input0", { 1, 1, 2, 2 } },
258 {
"Input1", { 1, 1, 2, 2 } } },
261 auto optimizedNetwork = armnn::PolymorphicDowncast<armnn::OptimizedNetwork*>(optimized.get());
262 auto graph = optimizedNetwork->GetGraph();
264 BOOST_TEST((graph.GetNumInputs() == 2));
265 BOOST_TEST((graph.GetNumOutputs() == 1));
266 BOOST_TEST((graph.GetNumLayers() == 5));
289 BOOST_TEST(
IsConnected(inputLayer0, subLayer, 0, 0, tensorInfo));
290 BOOST_TEST(
IsConnected(inputLayer1, subLayer, 0, 1, tensorInfo));
291 BOOST_TEST(
IsConnected(inputLayer0, addLayer, 0, 0, tensorInfo));
292 BOOST_TEST(
IsConnected(subLayer, addLayer, 0, 1, tensorInfo));
294 for (
auto&& outputLayer : graph.GetOutputLayers())
296 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)