6 #include "../TestUtils.hpp" 10 #include <boost/test/unit_test.hpp> 26 std::string transposeLayerName =
"original_transpose";
31 transposeLayerName.c_str());
70 BOOST_TEST(
CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>,
71 &IsLayerOfType<armnn::InputLayer>, &IsLayerOfType<armnn::InputLayer>,
72 &IsLayerOfType<armnn::MultiplicationLayer>, &IsLayerOfType<armnn::MemCopyLayer>,
73 &IsLayerOfType<armnn::FloorLayer>, &IsLayerOfType<armnn::FakeQuantizationLayer>,
74 &IsLayerOfType<armnn::AdditionLayer>, &IsLayerOfType<armnn::ActivationLayer>,
75 &IsLayerOfType<armnn::TransposeLayer>, &IsLayerOfType<armnn::OutputLayer>));
80 BOOST_TEST(
CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>,
81 &IsLayerOfType<armnn::InputLayer>, &IsLayerOfType<armnn::InputLayer>,
82 &IsLayerOfType<armnn::TransposeLayer>, &IsLayerOfType<armnn::TransposeLayer>,
83 &IsLayerOfType<armnn::TransposeLayer>, &IsLayerOfType<armnn::MultiplicationLayer>,
84 &IsLayerOfType<armnn::MemCopyLayer>, &IsLayerOfType<armnn::FloorLayer>,
85 &IsLayerOfType<armnn::FakeQuantizationLayer>, &IsLayerOfType<armnn::AdditionLayer>,
86 &IsLayerOfType<armnn::ActivationLayer>, &IsLayerOfType<armnn::OutputLayer>));
88 std::list<std::string> testRelatedLayers = { transposeLayerName };
90 BOOST_TEST(CheckRelatedLayers<armnn::TransposeLayer>(graph, testRelatedLayers));
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
Optimizer::Optimizations MakeOptimizations(Args &&... args)
static void Pass(Graph &graph, const Optimizations &optimizations)
This layer represents an activation operation with the specified activation function.
OptimizeForConnection< Layer, TransposeLayer, MoveTransposeUpImpl > MoveTransposeUp
int LayerBindingId
Type of identifiers for bindable layers (inputs, outputs).
const InputSlot & GetInputSlot(unsigned int index) const override
Get a const input slot handle by slot index.
A layer user-provided data can be bound to (e.g. inputs, outputs).
A FakeQuantizationDescriptor for the FakeQuantizationLayer.
This layer represents a memory copy operation.
BOOST_AUTO_TEST_CASE(CheckConvolution2dLayer)
An ActivationDescriptor for the ActivationLayer.
This layer represents a fake quantization operation.
This layer represents a floor operation.
This layer represents a transpose operation.
This layer represents an addition operation.
void SetTensorInfo(const TensorInfo &tensorInfo)
Sets the TensorInfo used by this output handler.
BOOST_AUTO_TEST_SUITE_END()
const OutputHandler & GetOutputHandler(unsigned int i=0) const
bool CheckSequence(const armnn::Graph::ConstIterator first, const armnn::Graph::ConstIterator last)
A TransposeDescriptor for the TransposeLayer.
This layer represents a multiplication operation.