Home
last modified time | relevance | path

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

/external/armnn/src/armnn/optimizations/
DFuseBatchNorm.hpp58 ConstantLayer* weightLayer = PolymorphicDowncast<ConstantLayer*>( in Run() local
61 weightsTensor = ConstTensor(weightLayer->m_LayerOutput->GetTensorInfo(), in Run()
62 weightLayer->m_LayerOutput->Map(true)); in Run()
176 weightLayer->GetOutputSlot(0).Disconnect(base.GetInputSlot(1)); in Run()
177 weightLayer->GetOutputSlot(0).Connect(newConv2dLayer.GetInputSlot(1)); in Run()
178weightLayer->m_LayerOutput = std::make_unique<ScopedTensorHandle>(fusedWeightsTensor); in Run()