21.11
|
#include "ChannelShuffleTestImpl.hpp"
#include <backendsCommon/test/DataTypeUtils.hpp>
#include <backendsCommon/test/TensorCopyUtils.hpp>
#include <backendsCommon/test/WorkloadTestUtils.hpp>
Go to the source code of this file.
LayerTestResult<T, 2> ChannelShuffle2DTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 100 of file ChannelShuffleTestImpl.cpp.
References ChannelShuffleDescriptor::m_Axis, ChannelShuffleDescriptor::m_NumGroups, and TensorInfo::SetQuantizationScale().
template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 2> ChannelShuffle2DTest< armnn::DataType::Float32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmS8>, 2> ChannelShuffle2DTest< armnn::DataType::QAsymmS8 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmU8>, 2> ChannelShuffle2DTest< armnn::DataType::QAsymmU8 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
LayerTestResult<T, 4> ChannelShuffle4DTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 148 of file ChannelShuffleTestImpl.cpp.
References ChannelShuffleDescriptor::m_Axis, ChannelShuffleDescriptor::m_NumGroups, and TensorInfo::SetQuantizationScale().
template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4> ChannelShuffle4DTest< armnn::DataType::Float32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmS8>, 4> ChannelShuffle4DTest< armnn::DataType::QAsymmS8 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmU8>, 4> ChannelShuffle4DTest< armnn::DataType::QAsymmU8 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
LayerTestResult<T, 4> SimpleChannelShuffleTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 57 of file ChannelShuffleTestImpl.cpp.
References ChannelShuffleDescriptor::m_Axis, ChannelShuffleDescriptor::m_NumGroups, and TensorInfo::SetQuantizationScale().
template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4> SimpleChannelShuffleTest< armnn::DataType::Float32 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmS8>, 4> SimpleChannelShuffleTest< armnn::DataType::QAsymmS8 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmU8>, 4> SimpleChannelShuffleTest< armnn::DataType::QAsymmU8 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |