22.08
|
#include <armnnTestUtils/LayerTestResult.hpp>
#include <ResolveType.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <armnn/backends/WorkloadFactory.hpp>
Go to the source code of this file.
Functions | |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 4 > | SimpleChannelShuffleTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 2 > | ChannelShuffle2DTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 4 > | ChannelShuffle4DTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
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().
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().
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().