15 namespace optimizations
38 RedirectWeightsAndBiases<FullyConnectedLayer>(&layer);
54 template <
typename LayerT>
55 static LayerT* RedirectWeightsAndBiases(
Layer* layer)
57 LayerT* layerPtr = PolymorphicDowncast<LayerT*>(layer);
61 for (
unsigned int inputSlotIndex = 1; inputSlotIndex != layerPtr->GetNumInputSlots(); ++inputSlotIndex)
63 OutputSlot* outputSlot = layerPtr->GetInputSlot(inputSlotIndex).GetConnectedOutputSlot();
68 if (inputSlotIndex == 1)
72 else if (inputSlotIndex == 2)
A layer that the constant data can be bound to.
std::shared_ptr< ConstTensorHandle > m_LayerOutput
Layer & GetOwningLayer() const
Copyright (c) 2021 ARM Limited and Contributors.
void IgnoreUnused(Ts &&...)
LayerType GetType() const override
Returns the armnn::LayerType of this layer.