16 TEST_CASE(
"MoveTransposeUpTest")
27 std::string transposeLayerName =
"original_transpose";
32 transposeLayerName.c_str());
71 CHECK(
CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>,
72 &IsLayerOfType<armnn::InputLayer>, &IsLayerOfType<armnn::InputLayer>,
73 &IsLayerOfType<armnn::MultiplicationLayer>, &IsLayerOfType<armnn::MemCopyLayer>,
74 &IsLayerOfType<armnn::FloorLayer>, &IsLayerOfType<armnn::FakeQuantizationLayer>,
75 &IsLayerOfType<armnn::AdditionLayer>, &IsLayerOfType<armnn::ActivationLayer>,
76 &IsLayerOfType<armnn::TransposeLayer>, &IsLayerOfType<armnn::OutputLayer>));
81 CHECK(
CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>,
82 &IsLayerOfType<armnn::InputLayer>, &IsLayerOfType<armnn::InputLayer>,
83 &IsLayerOfType<armnn::TransposeLayer>, &IsLayerOfType<armnn::TransposeLayer>,
84 &IsLayerOfType<armnn::TransposeLayer>, &IsLayerOfType<armnn::MultiplicationLayer>,
85 &IsLayerOfType<armnn::MemCopyLayer>, &IsLayerOfType<armnn::FloorLayer>,
86 &IsLayerOfType<armnn::FakeQuantizationLayer>, &IsLayerOfType<armnn::AdditionLayer>,
87 &IsLayerOfType<armnn::ActivationLayer>, &IsLayerOfType<armnn::OutputLayer>));
89 std::list<std::string> testRelatedLayers = { transposeLayerName };
91 CHECK(CheckRelatedLayers<armnn::TransposeLayer>(graph, testRelatedLayers));
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.
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.
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.