22.08
|
#include "ChannelShuffleTestImpl.hpp"
#include <DataTypeUtils.hpp>
#include <armnnTestUtils/TensorCopyUtils.hpp>
#include <armnnTestUtils/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 101 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 149 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 58 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 | ||
) |