21.02
|
#include "../GraphUtils.hpp"
#include "../TestUtils.hpp"
#include <Optimizer.hpp>
#include <boost/test/unit_test.hpp>
Go to the source code of this file.
Functions | |
void | AddBroadcastReshapeLayerOptimizerTest (const TensorInfo &info0, const TensorInfo &info1, const TensorInfo &outputInfo, const std::string &reshapeLayerName, const TensorShape &expectedReshapeShape, const DataType expectedDataType) |
BOOST_AUTO_TEST_CASE (AddBroadcastReshapeLayerSimpleTest) | |
BOOST_AUTO_TEST_CASE (AddBroadcastReshapeLayer1DTest) | |
BOOST_AUTO_TEST_CASE (AddBroadcastReshapeLayer2DTest) | |
BOOST_AUTO_TEST_CASE (AddBroadcastReshapeLayer3DTest) | |
BOOST_AUTO_TEST_CASE (AddBroadcastReshapeLayer3DMergedTest) | |
BOOST_AUTO_TEST_CASE (AddBroadcastReshapeLayerSubtractionTest) | |
BOOST_AUTO_TEST_CASE (AddBroadcastReshapeLayerDivisionTest) | |
BOOST_AUTO_TEST_CASE (AddBroadcastReshapeLayerMultiplicationTest) | |
BOOST_AUTO_TEST_CASE (AddNoBroadcastReshapeLayerTest) | |
void AddBroadcastReshapeLayerOptimizerTest | ( | const TensorInfo & | info0, |
const TensorInfo & | info1, | ||
const TensorInfo & | outputInfo, | ||
const std::string & | reshapeLayerName, | ||
const TensorShape & | expectedReshapeShape, | ||
const DataType | expectedDataType | ||
) |
Definition at line 18 of file AddBroadcastReshapeLayerTests.cpp.
References Graph::AddLayer(), Graph::cbegin(), Graph::cend(), CheckSequence(), OutputSlot::Connect(), GetFirstLayerWithName(), Layer::GetInputSlot(), Layer::GetOutputSlot(), OutputSlot::GetTensorInfo(), armnn::MakeOptimizations(), Optimizer::Pass(), and OutputSlot::SetTensorInfo().
Referenced by BOOST_AUTO_TEST_CASE().
BOOST_AUTO_TEST_CASE | ( | AddBroadcastReshapeLayerSimpleTest | ) |
Definition at line 65 of file AddBroadcastReshapeLayerTests.cpp.
References AddBroadcastReshapeLayerOptimizerTest(), and armnn::Float32.
BOOST_AUTO_TEST_CASE | ( | AddBroadcastReshapeLayer1DTest | ) |
Definition at line 74 of file AddBroadcastReshapeLayerTests.cpp.
References AddBroadcastReshapeLayerOptimizerTest(), and armnn::Float32.
BOOST_AUTO_TEST_CASE | ( | AddBroadcastReshapeLayer2DTest | ) |
Definition at line 84 of file AddBroadcastReshapeLayerTests.cpp.
References AddBroadcastReshapeLayerOptimizerTest(), and armnn::Float32.
BOOST_AUTO_TEST_CASE | ( | AddBroadcastReshapeLayer3DTest | ) |
Definition at line 94 of file AddBroadcastReshapeLayerTests.cpp.
References AddBroadcastReshapeLayerOptimizerTest(), and armnn::Float32.
BOOST_AUTO_TEST_CASE | ( | AddBroadcastReshapeLayer3DMergedTest | ) |
Definition at line 104 of file AddBroadcastReshapeLayerTests.cpp.
References AddBroadcastReshapeLayerOptimizerTest(), and armnn::Float32.
BOOST_AUTO_TEST_CASE | ( | AddBroadcastReshapeLayerSubtractionTest | ) |
Definition at line 114 of file AddBroadcastReshapeLayerTests.cpp.
References Graph::AddLayer(), Graph::cbegin(), Graph::cend(), CheckSequence(), OutputSlot::Connect(), armnn::Float32, GetFirstLayerWithName(), Layer::GetInputSlot(), Layer::GetOutputSlot(), OutputSlot::GetTensorInfo(), armnn::MakeOptimizations(), Optimizer::Pass(), and OutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | AddBroadcastReshapeLayerDivisionTest | ) |
Definition at line 159 of file AddBroadcastReshapeLayerTests.cpp.
References Graph::AddLayer(), Graph::cbegin(), Graph::cend(), CheckSequence(), OutputSlot::Connect(), GetFirstLayerWithName(), Layer::GetInputSlot(), Layer::GetOutputSlot(), OutputSlot::GetTensorInfo(), armnn::MakeOptimizations(), Optimizer::Pass(), armnn::QAsymmS8, and OutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | AddBroadcastReshapeLayerMultiplicationTest | ) |
Definition at line 204 of file AddBroadcastReshapeLayerTests.cpp.
References Graph::AddLayer(), Graph::cbegin(), Graph::cend(), CheckSequence(), OutputSlot::Connect(), GetFirstLayerWithName(), Layer::GetInputSlot(), Layer::GetOutputSlot(), OutputSlot::GetTensorInfo(), armnn::MakeOptimizations(), Optimizer::Pass(), armnn::QAsymmU8, and OutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | AddNoBroadcastReshapeLayerTest | ) |
Definition at line 249 of file AddBroadcastReshapeLayerTests.cpp.
References Graph::AddLayer(), BOOST_AUTO_TEST_SUITE_END(), Graph::cbegin(), Graph::cend(), CheckSequence(), OutputSlot::Connect(), GetFirstLayerWithName(), Layer::GetInputSlot(), Layer::GetOutputSlot(), armnn::MakeOptimizations(), Optimizer::Pass(), armnn::QAsymmU8, and OutputSlot::SetTensorInfo().