22.05
|
#include <armnnTestUtils/LayerTestResult.hpp>
#include <Half.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <armnn/backends/WorkloadFactory.hpp>
Go to the source code of this file.
LayerTestResult<armnn::Half, 4> BatchNormFloat16NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 364 of file BatchNormalizationTestImpl.cpp.
References armnn::NHWC.
Referenced by TEST_SUITE().
LayerTestResult<armnn::Half, 4> BatchNormFloat16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 316 of file BatchNormalizationTestImpl.cpp.
References armnn::NCHW.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> BatchNormFloat32NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 264 of file BatchNormalizationTestImpl.cpp.
References armnn::NHWC.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> BatchNormFloat32Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 216 of file BatchNormalizationTestImpl.cpp.
References armnn::NCHW.
Referenced by TEST_SUITE().
LayerTestResult<int16_t, 4> BatchNormInt16NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 560 of file BatchNormalizationTestImpl.cpp.
References armnn::NHWC.
Referenced by TEST_SUITE().
LayerTestResult<int16_t, 4> BatchNormInt16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 512 of file BatchNormalizationTestImpl.cpp.
References armnn::NCHW.
Referenced by TEST_SUITE().
LayerTestResult<uint8_t, 4> BatchNormUint8NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 464 of file BatchNormalizationTestImpl.cpp.
References armnn::NHWC.
Referenced by TEST_SUITE().
LayerTestResult<uint8_t, 4> BatchNormUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 416 of file BatchNormalizationTestImpl.cpp.
References armnn::NCHW.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> CompareBatchNormTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::ITensorHandleFactory & | refTensorHandleFactory | ||
) |
Definition at line 612 of file BatchNormalizationTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), armnn::BatchNormalization, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::Float32, TensorInfo::GetNumElements(), TensorInfo::GetShape(), armnn::IgnoreUnused(), BatchNormalizationQueueDescriptor::m_Beta, BatchNormalizationDescriptor::m_Eps, BatchNormalizationQueueDescriptor::m_Gamma, BatchNormalizationQueueDescriptor::m_Mean, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, and BatchNormalizationQueueDescriptor::m_Variance.
Referenced by TEST_SUITE().