6 #include "../TestUtils.hpp" 10 #include <boost/test/unit_test.hpp> 31 std::string reshape1Name =
"reshape1";
32 std::string reshape2Name =
"reshape2";
43 &IsLayerOfType<armnn::ReshapeLayer>, &IsLayerOfType<armnn::ReshapeLayer>,
44 &IsLayerOfType<armnn::OutputLayer>));
48 auto checkReshape = [&info2](
const armnn::Layer*
const layer) ->
bool {
50 return IsLayerOfType<armnn::ReshapeLayer>(layer) &&
51 (reshapeLayer->GetParameters().m_TargetShape == info2.GetShape()) &&
52 (reshapeLayer->GetOutputHandler().GetTensorInfo().GetShape() == info2.GetShape());
57 &IsLayerOfType<armnn::OutputLayer>));
60 std::list<std::string> testRelatedLayers = { reshape2Name, reshape1Name };
62 BOOST_TEST(CheckRelatedLayers<armnn::ReshapeLayer>(graph, testRelatedLayers));
76 &IsLayerOfType<armnn::OutputLayer>));
const OutputHandler & GetOutputHandler(unsigned int i=0) const
BOOST_AUTO_TEST_CASE(CheckConvolution2dLayer)
OptimizeForConnection< ReshapeLayer, ReshapeLayer, OptimizeConsecutiveReshapesImpl > OptimizeConsecutiveReshapes
LayerT * AddLayer(Args &&... args)
Adds a new layer, of type LayerType, to the graph constructed with the arguments passed.
This layer represents a reshape operation.
static void Pass(Graph &graph, const Optimizations &optimizations)
bool CheckSequence(const armnn::Graph::ConstIterator first, const armnn::Graph::ConstIterator last)
A ReshapeDescriptor for the ReshapeLayer.
A layer user-provided data can be bound to (e.g. inputs, outputs).
ConstIterator cbegin() const
Returns const iterator pointing to the beginning of the list. Lowercase for range-based for loops...
Optimizer::Optimizations MakeOptimizations(Args &&... args)
BOOST_AUTO_TEST_SUITE_END()
void SetTensorInfo(const TensorInfo &tensorInfo)
Sets the TensorInfo used by this output handler.
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
ConstIterator cend() const
Returns const iterator pointing to the end of the list. Lowercase for range-based for loops...
LayerT * InsertNewLayer(InputSlot &insertBefore, Args &&... args)