Searched refs:batchNormLayer (Results 1 – 2 of 2) sorted by relevance
41 auto batchNormLayer = PolymorphicDowncast<BatchNormalizationLayer*>(&child); in Run() local44 BatchNormalizationDescriptor batchNormDescriptor = batchNormLayer->GetParameters(); in Run()48 …ConstTensor betaTensor(batchNormLayer->m_Beta->GetTensorInfo(), batchNormLayer->m_Beta->Map(true)); in Run()49 …ConstTensor gammaTensor(batchNormLayer->m_Gamma->GetTensorInfo(), batchNormLayer->m_Gamma->Map(tru… in Run()50 …ConstTensor meanTensor(batchNormLayer->m_Mean->GetTensorInfo(), batchNormLayer->m_Mean->Map(true)); in Run()51 …ConstTensor varTensor(batchNormLayer->m_Variance->GetTensorInfo(), batchNormLayer->m_Variance->Map… in Run()
167 IConnectableLayer* batchNormLayer = network->AddBatchNormalizationLayer(batchNormDescriptor, in CreateNetwork() local196 batchNormLayer->GetOutputSlot(0).SetTensorInfo(outputInfo); in CreateNetwork()200 convLayer ->GetOutputSlot(0).Connect(batchNormLayer->GetInputSlot(0)); in CreateNetwork()201 batchNormLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0)); in CreateNetwork()