6 #include "../TestUtils.hpp" 10 #include <boost/test/unit_test.hpp> 32 floor->GetOutputSlot().Connect(output->GetInputSlot(0));
35 &IsLayerOfType<armnn::FloorLayer>, &IsLayerOfType<armnn::OutputLayer>));
41 &IsLayerOfType<armnn::DebugLayer>, &IsLayerOfType<armnn::FloorLayer>,
42 &IsLayerOfType<armnn::DebugLayer>, &IsLayerOfType<armnn::OutputLayer>));
BOOST_AUTO_TEST_CASE(CheckConvolution2dLayer)
LayerT * AddLayer(Args &&... args)
Adds a new layer, of type LayerType, to the graph constructed with the arguments passed.
static void Pass(Graph &graph, const Optimizations &optimizations)
int Connect(InputSlot &destination)
This layer represents a floor operation.
bool CheckSequence(const armnn::Graph::ConstIterator first, const armnn::Graph::ConstIterator last)
void SetTensorInfo(const TensorInfo &tensorInfo) override
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()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
ConstIterator cend() const
Returns const iterator pointing to the end of the list. Lowercase for range-based for loops...
OptimizeForType< Layer, AddDebugImpl > InsertDebugLayer
const OutputSlot & GetOutputSlot(unsigned int index=0) const override