22.08
|
#include "Conv2dTestImpl.hpp"
#include <armnnUtils/QuantizeHelper.hpp>
#include <armnnUtils/TensorUtils.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <armnn/utility/NumericCast.hpp>
#include <armnnUtils/DataLayoutIndexed.hpp>
#include <armnnUtils/Permute.hpp>
#include <armnn/backends/TensorHandle.hpp>
#include <armnnTestUtils/DataLayoutUtils.hpp>
#include <armnnTestUtils/TensorCopyUtils.hpp>
#include <armnnTestUtils/WorkloadTestUtils.hpp>
#include <armnnTestUtils/TensorHelpers.hpp>
#include <string>
Go to the source code of this file.
void ApplyBias | ( | std::vector< T > & | v, |
float | vScale, | ||
int32_t | vOffset, | ||
const std::vector< B > & | bias, | ||
float | bScale, | ||
int32_t | bOffset, | ||
uint32_t | w, | ||
uint32_t | h | ||
) |
Definition at line 162 of file Conv2dTestImpl.cpp.
References ARMNN_ASSERT, ARMNN_ASSERT_MSG, B, and armnnUtils::SelectiveDequantize().
Referenced by Convolution1dTestImpl(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), and SimpleConvolution2dTestImpl().
LayerTestResult<float,4> CompareConvolution2dTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::ITensorHandleFactory & | refTensorHandleFactory | ||
) |
Definition at line 3747 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T,4> CompareConvolution2dTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::ITensorHandleFactory & | refTensorHandleFactory | ||
) |
Definition at line 1369 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), armnn::Convolution2d, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), TensorInfo::GetNumElements(), TensorInfo::GetShape(), armnn::info, Convolution2dQueueDescriptor::m_Bias, Convolution2dDescriptor::m_BiasEnabled, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, and Convolution2dQueueDescriptor::m_Weight.
LayerTestResult<float, 4> CompareDepthwiseConvolution2dFloatTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::ITensorHandleFactory & | refTensorHandleFactory, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 4039 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> CompareDepthwiseConvolution2dTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::ITensorHandleFactory & | refTensorHandleFactory, | ||
const armnnUtils::DataLayoutIndexed & | layout | ||
) |
Definition at line 3090 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::DepthwiseConvolution2d, IWorkloadFactory::GetBackendId(), armnn::GetBiasDataType(), DataLayoutIndexed::GetDataLayout(), TensorInfo::GetNumElements(), TensorInfo::GetShape(), armnn::info, DepthwiseConvolution2dQueueDescriptor::m_Bias, DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideY, DepthwiseConvolution2dQueueDescriptor::m_Weight, armnn::NCHW, armnn::NHWC, armnnUtils::Permute(), armnnUtils::Permuted(), and TensorInfo::SetShape().
LayerTestResult<uint8_t, 4> CompareDepthwiseConvolution2dUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::ITensorHandleFactory & | refTensorHandleFactory, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 4051 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> Convolution1dTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled | ||
) |
Definition at line 3611 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T,4> Convolution1dTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled | ||
) |
Definition at line 501 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), ApplyBias(), armnn::Convolution2d, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::info, Convolution2dQueueDescriptor::m_Bias, Convolution2dDescriptor::m_BiasEnabled, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, Convolution2dQueueDescriptor::m_Weight, and TensorInfo::SetQuantizationScale().
LayerTestResult<uint8_t, 4> Convolution1dUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled | ||
) |
Definition at line 3621 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 1300 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
template LayerTestResult<armnn::ResolveType<armnn::DataType::BFloat16>, 4> Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test< armnn::DataType::BFloat16, armnn::DataType::BFloat16 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4> Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test< armnn::DataType::Float32, armnn::DataType::Float32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmS8>, 4> Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test< armnn::DataType::QAsymmS8, armnn::DataType::Signed32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmU8>, 4> Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test< armnn::DataType::QAsymmU8, armnn::DataType::Signed32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 4> Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test< armnn::DataType::QSymmS16, armnn::DataType::Signed32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
LayerTestResult<T, 4> Convolution2d2x3x3Dilation3x3Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 1227 of file Conv2dTestImpl.cpp.
LayerTestResult<T, 4> Convolution2d3x3Dilation3x3Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 1169 of file Conv2dTestImpl.cpp.
LayerTestResult<T, 4> Convolution2d3x3DilationTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const std::vector< float > & | inputNoQuantizedValues, | ||
armnn::TensorInfo & | inputTensorInfo, | ||
const std::vector< float > & | kernelNoQuantizedValues, | ||
armnn::TensorInfo & | kernelTensorInfo, | ||
const std::vector< float > & | outputExpectedNoQuantizedValues, | ||
armnn::TensorInfo & | outputTensorInfo, | ||
uint32_t | dilationX, | ||
uint32_t | dilationY, | ||
armnn::DataLayout | layout = armnn::DataLayout::NCHW , |
||
uint32_t | padLeft = 0 , |
||
uint32_t | padTop = 0 , |
||
uint32_t | padRight = 0 , |
||
uint32_t | padBottom = 0 , |
||
uint32_t | strideX = 1 , |
||
uint32_t | strideY = 1 , |
||
bool | biasEnabled = false |
||
) |
Definition at line 1079 of file Conv2dTestImpl.cpp.
References armnn::Float32, TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
LayerTestResult<float, 4> Convolution2d3x3Stride2x2BFloat16SmallValueTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout & | dataLayout | ||
) |
Definition at line 1602 of file Conv2dTestImpl.cpp.
References armnn::BFloat16, armnn::Float32, armnn::IgnoreUnused(), and SimpleConvolution2dNhwcTestImpl().
Referenced by TEST_SUITE().
LayerTestResult<float, 4> Convolution2d3x3Stride2x2BFloat16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout & | dataLayout | ||
) |
Definition at line 1495 of file Conv2dTestImpl.cpp.
References armnn::BFloat16, armnn::Float32, armnn::IgnoreUnused(), and SimpleConvolution2dNhwcTestImpl().
Referenced by TEST_SUITE().
LayerTestResult<float, 4> Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
armnn::DataLayout | layout | ||
) |
Definition at line 3600 of file Conv2dTestImpl.cpp.
References Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTestCommon(), and armnn::Float32.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::DataLayout | layout, | ||
float | qScale, | ||
int32_t | qOffset | ||
) |
Definition at line 944 of file Conv2dTestImpl.cpp.
Referenced by Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest().
LayerTestResult<float, 4> Convolution2dAsymmetricPaddingTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
armnn::DataLayout | layout | ||
) |
Definition at line 3590 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<uint8_t, 4> Convolution2dPerAxisQuantTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3631 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), armnn::Convolution2d, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, Convolution2dDescriptor::m_StrideX, armnn::NCHW, and PermuteTensorNhwcToNchw().
Referenced by TEST_SUITE().
LayerTestResult<T, 4> DepthwiseConvolution2d2x3x3Dilation3x3Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 2879 of file Conv2dTestImpl.cpp.
LayerTestResult<T, 4> DepthwiseConvolution2d3x3Dilation3x3Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 2821 of file Conv2dTestImpl.cpp.
LayerTestResult<T, 4> DepthwiseConvolution2d3x3DilationTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const std::vector< float > & | inputNoQuantizedValues, | ||
armnn::TensorInfo & | inputTensorInfo, | ||
const std::vector< float > & | kernelNoQuantizedValues, | ||
armnn::TensorInfo & | kernelTensorInfo, | ||
const std::vector< float > & | outputExpectedNoQuantizedValues, | ||
armnn::TensorInfo & | outputTensorInfo, | ||
uint32_t | dilationX, | ||
uint32_t | dilationY, | ||
armnn::DataLayout | layout = armnn::DataLayout::NCHW , |
||
bool | biasEnabled = false |
||
) |
Definition at line 2731 of file Conv2dTestImpl.cpp.
References armnn::Float32, TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
LayerTestResult<float, 4> DepthwiseConvolution2dAsymmetricTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3833 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> DepthwiseConvolution2dAsymmetricTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 2484 of file Conv2dTestImpl.cpp.
LayerTestResult<T, 4> DepthwiseConvolution2dAsymmetricTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const std::vector< T > & | input, | ||
const std::vector< T > & | kernel, | ||
const std::vector< B > & | bias, | ||
const std::vector< T > & | outputExpected, | ||
const armnn::TensorShape & | inputShape, | ||
const armnn::TensorShape & | kernelShape, | ||
const armnn::TensorShape & | outputExpectedShape, | ||
float | qScale, | ||
int32_t | qOffset, | ||
const armnn::DataLayout | layout, | ||
uint32_t | padLeft = 0 , |
||
uint32_t | padTop = 0 , |
||
uint32_t | padRight = 0 , |
||
uint32_t | padBottom = 0 , |
||
uint32_t | strideX = 1 , |
||
uint32_t | strideY = 1 |
||
) |
Definition at line 1715 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), ApplyBias(), ARMNN_ASSERT, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::DepthwiseConvolution2d, IWorkloadFactory::GetBackendId(), TensorInfo::GetNumElements(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), armnn::info, DepthwiseConvolution2dQueueDescriptor::m_Bias, DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideY, DepthwiseConvolution2dQueueDescriptor::m_Weight, armnn::NCHW, armnn::NHWC, armnn::numeric_cast(), armnnUtils::Permute(), armnnUtils::Permuted(), TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
LayerTestResult<int16_t, 4> DepthwiseConvolution2dDepthMul1Int16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3891 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> DepthwiseConvolution2dDepthMul1Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3779 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> DepthwiseConvolution2dDepthMul1TestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 1885 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), ApplyBias(), CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::DepthwiseConvolution2d, IWorkloadFactory::GetBackendId(), TensorInfo::GetNumElements(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), armnn::info, DepthwiseConvolution2dQueueDescriptor::m_Bias, DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideY, DepthwiseConvolution2dQueueDescriptor::m_Weight, armnn::NCHW, armnn::NHWC, armnnUtils::Permute(), armnnUtils::Permuted(), TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
LayerTestResult<uint8_t, 4> DepthwiseConvolution2dDepthMul1Uint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3855 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> DepthwiseConvolution2dDepthMul64Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 3790 of file Conv2dTestImpl.cpp.
References armnn::Float32, armnn::GetDataTypeSize(), armnn::NCHW, armnnUtils::Permute(), and armnnUtils::Permuted().
Referenced by TEST_SUITE().
LayerTestResult<float, 4> DepthwiseConvolution2dDepthNhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled | ||
) |
Definition at line 3769 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<int16_t, 4> DepthwiseConvolution2dInt16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3880 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> DepthwiseConvolution2dMult2Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3026 of file Conv2dTestImpl.cpp.
template LayerTestResult<armnn::ResolveType<armnn::DataType::BFloat16>, 4> DepthwiseConvolution2dMult2Test< armnn::DataType::BFloat16, armnn::DataType::BFloat16 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4> DepthwiseConvolution2dMult2Test< armnn::DataType::Float32, armnn::DataType::Float32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
LayerTestResult<T, 4> DepthwiseConvolution2dMult4Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 2951 of file Conv2dTestImpl.cpp.
template LayerTestResult<armnn::ResolveType<armnn::DataType::BFloat16>, 4> DepthwiseConvolution2dMult4Test< armnn::DataType::BFloat16, armnn::DataType::BFloat16 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4> DepthwiseConvolution2dMult4Test< armnn::DataType::Float32, armnn::DataType::Float32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
LayerTestResult<T, 4> DepthwiseConvolution2dNhwcTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled | ||
) |
Definition at line 2566 of file Conv2dTestImpl.cpp.
References armnn::NHWC.
LayerTestResult<uint8_t, 4> DepthwiseConvolution2dPerAxisQuantTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3902 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::DepthwiseConvolution2d, IWorkloadFactory::GetBackendId(), DepthwiseConvolution2dQueueDescriptor::m_Bias, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dQueueDescriptor::m_Weight, armnn::NCHW, armnnUtils::Permute(), armnnUtils::Permuted(), PermuteTensorNhwcToNchw(), and TensorInfo::SetShape().
Referenced by TEST_SUITE().
LayerTestResult<float, 4> DepthwiseConvolution2dTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3758 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> DepthwiseConvolution2dTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 2064 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), ApplyBias(), CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::DepthwiseConvolution2d, IWorkloadFactory::GetBackendId(), TensorInfo::GetNumElements(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), armnn::info, DepthwiseConvolution2dQueueDescriptor::m_Bias, DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideY, DepthwiseConvolution2dQueueDescriptor::m_Weight, armnn::NCHW, armnn::NHWC, armnnUtils::Permute(), armnnUtils::Permuted(), TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
LayerTestResult<T, 4> DepthwiseConvolution2dTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const std::vector< T > & | originalInput, | ||
const std::vector< T > & | originalKernel, | ||
const std::vector< B > & | bias, | ||
const std::vector< T > & | originalOutputExpected, | ||
const armnn::TensorShape & | originalInputShape, | ||
const armnn::TensorShape & | originalKernelShape, | ||
const armnn::TensorShape & | originalOutputExpectedShape, | ||
float | qScale, | ||
int32_t | qOffset, | ||
const armnn::DataLayout | layout = armnn::DataLayout::NCHW , |
||
uint32_t | padLeft = 0 , |
||
uint32_t | padTop = 0 , |
||
uint32_t | padRight = 0 , |
||
uint32_t | padBottom = 0 , |
||
uint32_t | strideX = 1 , |
||
uint32_t | strideY = 1 , |
||
uint32_t | dilationX = 1 , |
||
uint32_t | dilationY = 1 |
||
) |
Definition at line 2291 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), ApplyBias(), ARMNN_ASSERT, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::DepthwiseConvolution2d, IWorkloadFactory::GetBackendId(), TensorInfo::GetNumElements(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), armnn::info, DepthwiseConvolution2dQueueDescriptor::m_Bias, DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_DilationX, DepthwiseConvolution2dDescriptor::m_DilationY, DepthwiseConvolution2dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideY, DepthwiseConvolution2dQueueDescriptor::m_Weight, armnn::NCHW, armnn::NHWC, armnn::numeric_cast(), armnnUtils::Permute(), armnnUtils::Permuted(), TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
LayerTestResult<uint8_t, 4> DepthwiseConvolution2dUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3844 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
std::vector<T> GetBias | ( | bool | biasEnabled, |
float | qScale, | ||
armnn::TensorInfo | outputInfo, | ||
armnn::DataLayout | layout | ||
) |
Definition at line 115 of file Conv2dTestImpl.cpp.
References DataLayoutIndexed::GetChannelsIndex(), and TensorInfo::GetShape().
std::vector<T> GetBias2 | ( | bool | biasEnabled, |
float | qScale | ||
) |
Definition at line 73 of file Conv2dTestImpl.cpp.
std::vector<T> GetBias4 | ( | bool | biasEnabled, |
float | qScale | ||
) |
Definition at line 87 of file Conv2dTestImpl.cpp.
std::vector<T> GetBias8 | ( | bool | biasEnabled, |
float | qScale | ||
) |
Definition at line 101 of file Conv2dTestImpl.cpp.
LayerTestResult<float, 4> SimpleConvolution2d3x3NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled | ||
) |
Definition at line 3524 of file Conv2dTestImpl.cpp.
References armnn::NHWC.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> SimpleConvolution2d3x3NhwcTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled, | ||
armnn::DataLayout | dataLayout | ||
) |
Definition at line 646 of file Conv2dTestImpl.cpp.
References armnn::IgnoreUnused().
LayerTestResult<int16_t, 4> SimpleConvolution2d3x3QSymm16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3579 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> SimpleConvolution2d3x3Stride2x2Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3540 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> SimpleConvolution2d3x3Stride2x2TestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled, | ||
const armnn::DataLayout & | dataLayout | ||
) |
Definition at line 701 of file Conv2dTestImpl.cpp.
References armnn::IgnoreUnused().
LayerTestResult<float, 4> SimpleConvolution2d3x3Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3513 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> SimpleConvolution2d3x3TestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 861 of file Conv2dTestImpl.cpp.
LayerTestResult<uint8_t, 4> SimpleConvolution2d3x3Uint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3557 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<int16_t, 4> SimpleConvolution2d3x5QSymm16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3568 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> SimpleConvolution2d3x5Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3491 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> SimpleConvolution2d3x5TestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 771 of file Conv2dTestImpl.cpp.
LayerTestResult<uint8_t, 4> SimpleConvolution2d3x5Uint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
const armnn::DataLayout | layout | ||
) |
Definition at line 3502 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> SimpleConvolution2dAsymmetricPaddingTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::DataLayout | layout, | ||
float | qScale, | ||
int32_t | qOffset | ||
) |
Definition at line 1016 of file Conv2dTestImpl.cpp.
LayerTestResult<O, 4> SimpleConvolution2dNhwcTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const std::vector< T > & | input, | ||
const std::vector< T > & | kernel, | ||
const std::vector< B > & | bias, | ||
const std::vector< O > & | outputExpected, | ||
const armnn::TensorShape & | inputShape, | ||
const armnn::TensorShape & | kernelShape, | ||
const armnn::TensorShape & | outputExpectedShape, | ||
const armnn::DataLayout | dataLayout, | ||
float | qScale, | ||
int32_t | qOffset, | ||
uint32_t | padLeft = 1 , |
||
uint32_t | padTop = 1 , |
||
uint32_t | padRight = 1 , |
||
uint32_t | padBottom = 1 , |
||
uint32_t | strideX = 1 , |
||
uint32_t | strideY = 1 |
||
) |
Definition at line 384 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), armnn::Convolution2d, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::IgnoreUnused(), armnn::info, Convolution2dQueueDescriptor::m_Bias, Convolution2dDescriptor::m_BiasEnabled, Convolution2dDescriptor::m_DataLayout, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, Convolution2dQueueDescriptor::m_Weight, and armnn::numeric_cast().
Referenced by Convolution2d3x3Stride2x2BFloat16SmallValueTest(), and Convolution2d3x3Stride2x2BFloat16Test().
LayerTestResult<T, 4> SimpleConvolution2dTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const std::vector< T > & | originalInput, | ||
const std::vector< T > & | originalKernel, | ||
const std::vector< B > & | bias, | ||
const std::vector< T > & | originalOutputExpected, | ||
const armnn::TensorShape & | originalInputShape, | ||
const armnn::TensorShape & | originalKernelShape, | ||
const armnn::TensorShape & | originalOutputExpectedShape, | ||
float | qScale, | ||
int32_t | qOffset, | ||
const armnn::DataLayout | layout = armnn::DataLayout::NCHW , |
||
uint32_t | padLeft = 0 , |
||
uint32_t | padTop = 0 , |
||
uint32_t | padRight = 0 , |
||
uint32_t | padBottom = 0 , |
||
uint32_t | strideX = 1 , |
||
uint32_t | strideY = 1 , |
||
uint32_t | dilationX = 1 , |
||
uint32_t | dilationY = 1 |
||
) |
Definition at line 194 of file Conv2dTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), ApplyBias(), ARMNN_ASSERT, armnn::Convolution2d, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), TensorInfo::GetNumElements(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), armnn::IgnoreUnused(), armnn::info, Convolution2dQueueDescriptor::m_Bias, Convolution2dDescriptor::m_BiasEnabled, Convolution2dDescriptor::m_DataLayout, Convolution2dDescriptor::m_DilationX, Convolution2dDescriptor::m_DilationY, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, Convolution2dQueueDescriptor::m_Weight, armnn::NHWC, armnn::numeric_cast(), armnnUtils::Permute(), TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
LayerTestResult<float, 4> SimpleDepthwiseConvolution2d3x3Dilation3x3NhwcTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 3866 of file Conv2dTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 4> SimpleDepthwiseConvolution2d3x3Dilation3x3NhwcTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | qScale, | ||
int32_t | qOffset, | ||
bool | biasEnabled | ||
) |
Definition at line 2651 of file Conv2dTestImpl.cpp.
References armnn::NHWC.