10 for (
auto&& layer : graph)
12 if (layer->GetName() == name)
22 for (
auto&& layer : graph)
24 if (layer->GetNameStr() == name)
43 unsigned int srcSlot,
unsigned int destSlot,
48 if (expectedTensorInfo != tensorInfo)
53 for (
unsigned int c = 0; c < numConnections; ++c)
55 auto inputSlot = boost::polymorphic_downcast<const armnn::InputSlot*>(outputSlot.
GetConnection(c));
56 if (inputSlot->GetOwningLayer().GetNameStr() == destLayer->
GetNameStr() &&
57 inputSlot->GetSlotIndex() == destSlot)
unsigned int GetNumInputSlots() const override
Returns the number of connectable input slots.
armnn::Layer * GetFirstLayerWithName(armnn::Graph &graph, const std::string &name)
bool GraphHasNamedLayer(const armnn::Graph &graph, const std::string &name)
virtual const IInputSlot * GetConnection(unsigned int index) const =0
unsigned int GetNumOutputSlots() const override
Returns the number of connectable output slots.
An output connection slot for a layer.
const std::string & GetNameStr() const
bool IsConnected(armnn::Layer *srcLayer, armnn::Layer *destLayer, unsigned int srcSlot, unsigned int destSlot, const armnn::TensorInfo &expectedTensorInfo)
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
Get the const output slot handle by slot index.
virtual const TensorInfo & GetTensorInfo() const =0
bool CheckNumberOfInputSlot(armnn::Layer *layer, unsigned int num)
virtual unsigned int GetNumConnections() const =0
bool CheckNumberOfOutputSlot(armnn::Layer *layer, unsigned int num)