15 namespace optimizations
32 RedirectWeightsAndBiases<Convolution2dLayer>(&layer);
35 RedirectWeightsAndBiases<DepthwiseConvolution2dLayer>(&layer);
40 RedirectWeightsAndBiases<FullyConnectedLayer>(&layer);
56 template <
typename LayerT>
57 static LayerT* RedirectWeightsAndBiases(
Layer* layer)
59 LayerT* layerPtr = PolymorphicDowncast<LayerT*>(layer);
63 for (
unsigned int inputSlotIndex = 1; inputSlotIndex != layerPtr->GetNumInputSlots(); ++inputSlotIndex)
65 OutputSlot* outputSlot = layerPtr->GetInputSlot(inputSlotIndex).GetConnectedOutputSlot();
72 if (inputSlotIndex == 1)
76 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.
#define ARMNN_ASSERT(COND)