20.02
|
#include "BatchNormalizationTestImpl.hpp"
#include <QuantizeHelper.hpp>
#include <ResolveType.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <armnnUtils/DataLayoutIndexed.hpp>
#include <backendsCommon/CpuTensorHandle.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <backendsCommon/WorkloadFactory.hpp>
#include <backendsCommon/test/TensorCopyUtils.hpp>
#include <backendsCommon/test/WorkloadTestUtils.hpp>
#include <test/TensorHelpers.hpp>
Go to the source code of this file.
LayerTestResult<armnn::Half, 4> BatchNormFloat16NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager | ||
) |
Definition at line 349 of file BatchNormalizationTestImpl.cpp.
References armnn::NHWC.
LayerTestResult<armnn::Half, 4> BatchNormFloat16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager | ||
) |
Definition at line 303 of file BatchNormalizationTestImpl.cpp.
References armnn::NCHW.
LayerTestResult<float, 4> BatchNormFloat32NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager | ||
) |
Definition at line 253 of file BatchNormalizationTestImpl.cpp.
References armnn::NHWC.
LayerTestResult<float, 4> BatchNormFloat32Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager | ||
) |
Definition at line 207 of file BatchNormalizationTestImpl.cpp.
References armnn::NCHW.
LayerTestResult<int16_t, 4> BatchNormInt16NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager | ||
) |
Definition at line 537 of file BatchNormalizationTestImpl.cpp.
References armnn::NHWC.
LayerTestResult<int16_t, 4> BatchNormInt16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager | ||
) |
Definition at line 491 of file BatchNormalizationTestImpl.cpp.
References armnn::NCHW.
LayerTestResult<uint8_t, 4> BatchNormUint8NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager | ||
) |
Definition at line 445 of file BatchNormalizationTestImpl.cpp.
References armnn::NHWC.
LayerTestResult<uint8_t, 4> BatchNormUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager | ||
) |
Definition at line 399 of file BatchNormalizationTestImpl.cpp.
References armnn::NCHW.
LayerTestResult<float,4> CompareBatchNormTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory | ||
) |
Definition at line 587 of file BatchNormalizationTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), CopyDataFromITensorHandle(), CopyDataToITensorHandle(), IWorkloadFactory::CreateBatchNormalization(), IWorkloadFactory::CreateTensorHandle(), armnn::Float32, armnn::IgnoreUnused(), BatchNormalizationQueueDescriptor::m_Beta, BatchNormalizationDescriptor::m_Eps, BatchNormalizationQueueDescriptor::m_Gamma, BatchNormalizationQueueDescriptor::m_Mean, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, and BatchNormalizationQueueDescriptor::m_Variance.