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 });
107 BOOST_TEST((graph.GetNumInputs() == 1));
108 BOOST_TEST((graph.GetNumOutputs() == 1));
109 BOOST_TEST((graph.GetNumLayers() == 3));
122 BOOST_TEST(
IsConnected(inputLayer, addLayer, 0, 0, tensorInfo));
123 BOOST_TEST(
IsConnected(inputLayer, addLayer, 0, 1, tensorInfo));
125 for (
auto&& outputLayer : graph.GetOutputLayers())
127 BOOST_TEST(
IsConnected(addLayer, const_cast<armnn::OutputLayer*>(outputLayer), 0, 0, tensorInfo));
240 Setup({ { "Input0", { 1, 1, 2, 2 } },
241 {
"Input1", { 1, 1, 2, 2 } } },
244 RunTest<4>({ {
"Input0", { 4.0f, 3.0f,
247 {
"Input1", { 1.0f, 2.0f,
250 { {
"Output", { 7.0f, 4.0f,
256 auto optimized = SetupOptimizedNetwork({ {
"Input0", { 1, 1, 2, 2 } },
257 {
"Input1", { 1, 1, 2, 2 } } },
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)
LayerType GetType() const override
Returns the armnn::LayerType of this layer.
bool IsConnected(armnn::Layer *srcLayer, armnn::Layer *destLayer, unsigned int srcSlot, unsigned int destSlot, const armnn::TensorInfo &expectedTensorInfo)
Graph & GetGraphForTesting(IOptimizedNetwork *optNet)
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.
bool CheckNumberOfInputSlot(armnn::Layer *layer, unsigned int num)
bool CheckNumberOfOutputSlot(armnn::Layer *layer, unsigned int num)