Home
last modified time | relevance | path

Searched refs:newConv2dLayer (Results 1 – 2 of 2) sorted by relevance

/external/armnn/src/armnn/optimizations/
DFoldPadIntoLayer2d.hpp190 const auto newConv2dLayer = FoldPadIntoLayer2dImpl<Convolution2dLayer>(graph, connection); in Run() local
192 if (newConv2dLayer != nullptr) in Run()
195 ARMNN_ASSERT_MSG(newConv2dLayer->GetInputSlot(1).GetConnection() != nullptr, in Run()
200 ARMNN_ASSERT_MSG(newConv2dLayer->GetInputSlot(2).GetConnection() != nullptr, in Run()
217 …const auto newConv2dLayer = FoldPadIntoLayer2dImpl<DepthwiseConvolution2dLayer>(graph, connection); in Run() local
219 if (newConv2dLayer != nullptr) in Run()
222 ARMNN_ASSERT_MSG(newConv2dLayer->GetInputSlot(1).GetConnection() != nullptr, in Run()
228 ARMNN_ASSERT_MSG(newConv2dLayer->GetInputSlot(2).GetConnection() != nullptr, in Run()
DFuseBatchNorm.hpp167 auto& newConv2dLayer = *graph.InsertNewLayer<ConvLayer>(base.GetInputSlot(0), in Run() local
173 if (newConv2dLayer.GetNumInputSlots() > 1) in Run()
177 weightLayer->GetOutputSlot(0).Connect(newConv2dLayer.GetInputSlot(1)); in Run()
188 biasLayer->GetOutputSlot(0).Connect(newConv2dLayer.GetInputSlot(2)); in Run()
197 biasLayer->GetOutputSlot(0).Connect(newConv2dLayer.GetInputSlot(2)); in Run()
204 newConv2dLayer.GetOutputSlot().MoveAllConnections(*parentOut); in Run()
206 parentOut = &newConv2dLayer.GetOutputSlot(); in Run()