Home
last modified time | relevance | path

Searched defs:meanTensor (Results 1 – 6 of 6) sorted by relevance

/external/armnn/src/armnn/layers/
DBatchNormalizationLayer.cpp75 ConstTensor meanTensor(m_Mean->GetTensorInfo(), m_Mean->Map(true)); in Accept() local
/external/armnn/src/backends/backendsCommon/test/layerTests/
DBatchNormalizationTestImpl.cpp77 armnn::ScopedCpuTensorHandle meanTensor(tensorInfo); in BatchNormTestImpl() local
163 armnn::ScopedCpuTensorHandle meanTensor(tensorInfo); in BatchNormTestNhwcImpl() local
647 armnn::ScopedCpuTensorHandle meanTensor(tensorInfo); in CompareBatchNormTest() local
/external/armnn/src/backends/cl/test/
DOpenClTimerTest.cpp82 ScopedCpuTensorHandle meanTensor(tensorInfo); in BOOST_AUTO_TEST_CASE() local
/external/armnn/src/armnn/optimizations/
DFuseBatchNorm.hpp50 …ConstTensor meanTensor(batchNormLayer->m_Mean->GetTensorInfo(), batchNormLayer->m_Mean->Map(true)); in Run() local
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1229 auto meanTensor = CreateConstTensor(node.input(3)); in ParseBatchNormalization() local
/external/armnn/src/armnnTfParser/
DTfParser.cpp1642 ConstTensor meanTensor = meanNode->GetConstTensor(meanTensorData); in ParseFusedBatchNorm() local