ArmNN
 20.08
Functions
RankTestImpl.cpp File Reference
#include "RankTestImpl.hpp"
#include <backendsCommon/test/DataTypeUtils.hpp>
#include <backendsCommon/test/TensorCopyUtils.hpp>
#include <backendsCommon/test/WorkloadTestUtils.hpp>
#include <test/TensorHelpers.hpp>

Go to the source code of this file.

Functions

template<typename T , std::size_t n>
LayerTestResult< int32_t, 1 > RankTest (armnn::TensorInfo inputTensorInfo, boost::multi_array< T, n > input, armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template<armnn::DataType ArmnnType, typename T >
LayerTestResult< int32_t, 1 > RankDimSize1Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template<armnn::DataType ArmnnType, typename T >
LayerTestResult< int32_t, 1 > RankDimSize2Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template<armnn::DataType ArmnnType, typename T >
LayerTestResult< int32_t, 1 > RankDimSize3Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template<armnn::DataType ArmnnType, typename T >
LayerTestResult< int32_t, 1 > RankDimSize4Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize4Test< armnn::DataType::Float16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize4Test< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize4Test< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize4Test< armnn::DataType::Signed32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize4Test< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize4Test< armnn::DataType::QSymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize4Test< armnn::DataType::QAsymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize4Test< armnn::DataType::BFloat16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize3Test< armnn::DataType::Float16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize3Test< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize3Test< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize3Test< armnn::DataType::Signed32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize3Test< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize3Test< armnn::DataType::QSymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize3Test< armnn::DataType::QAsymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize3Test< armnn::DataType::BFloat16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize2Test< armnn::DataType::Float16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize2Test< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize2Test< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize2Test< armnn::DataType::Signed32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize2Test< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize2Test< armnn::DataType::QSymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize2Test< armnn::DataType::QAsymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize2Test< armnn::DataType::BFloat16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize1Test< armnn::DataType::Float16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize1Test< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize1Test< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize1Test< armnn::DataType::Signed32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize1Test< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize1Test< armnn::DataType::QSymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize1Test< armnn::DataType::QAsymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< int32_t, 1 > RankDimSize1Test< armnn::DataType::BFloat16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 

Function Documentation

◆ RankDimSize1Test()

LayerTestResult<int32_t, 1> RankDimSize1Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

Definition at line 54 of file RankTestImpl.cpp.

57 {
58  armnn::TensorInfo inputTensorInfo({6}, ArmnnType, 1.0f, 0);
59  auto input = MakeTensor<T, 1>(inputTensorInfo, ConvertToDataType<ArmnnType>(
60  { -37.5f, -15.2f, -8.76f, -2.0f, -1.3f, -0.5f },
61  inputTensorInfo));
62 
63  return RankTest<T, 1>(inputTensorInfo, input, workloadFactory, memoryManager);
64 }

◆ RankDimSize1Test< armnn::DataType::BFloat16 >()

template LayerTestResult<int32_t, 1> RankDimSize1Test< armnn::DataType::BFloat16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize1Test< armnn::DataType::Float16 >()

template LayerTestResult<int32_t, 1> RankDimSize1Test< armnn::DataType::Float16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize1Test< armnn::DataType::Float32 >()

template LayerTestResult<int32_t, 1> RankDimSize1Test< armnn::DataType::Float32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize1Test< armnn::DataType::QAsymmS8 >()

template LayerTestResult<int32_t, 1> RankDimSize1Test< armnn::DataType::QAsymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize1Test< armnn::DataType::QAsymmU8 >()

template LayerTestResult<int32_t, 1> RankDimSize1Test< armnn::DataType::QAsymmU8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize1Test< armnn::DataType::QSymmS16 >()

template LayerTestResult<int32_t, 1> RankDimSize1Test< armnn::DataType::QSymmS16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize1Test< armnn::DataType::QSymmS8 >()

template LayerTestResult<int32_t, 1> RankDimSize1Test< armnn::DataType::QSymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize1Test< armnn::DataType::Signed32 >()

template LayerTestResult<int32_t, 1> RankDimSize1Test< armnn::DataType::Signed32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize2Test()

LayerTestResult<int32_t, 1> RankDimSize2Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

Definition at line 67 of file RankTestImpl.cpp.

70 {
71  armnn::TensorInfo inputTensorInfo({1, 3}, ArmnnType, 1.0f, 0);
72  auto input = MakeTensor<T, 2>(inputTensorInfo, ConvertToDataType<ArmnnType>(
73  { -37.5f, -15.2f, -8.76f },
74  inputTensorInfo));
75 
76  return RankTest<T, 2>(inputTensorInfo, input, workloadFactory, memoryManager);
77 }

◆ RankDimSize2Test< armnn::DataType::BFloat16 >()

template LayerTestResult<int32_t, 1> RankDimSize2Test< armnn::DataType::BFloat16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize2Test< armnn::DataType::Float16 >()

template LayerTestResult<int32_t, 1> RankDimSize2Test< armnn::DataType::Float16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize2Test< armnn::DataType::Float32 >()

template LayerTestResult<int32_t, 1> RankDimSize2Test< armnn::DataType::Float32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize2Test< armnn::DataType::QAsymmS8 >()

template LayerTestResult<int32_t, 1> RankDimSize2Test< armnn::DataType::QAsymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize2Test< armnn::DataType::QAsymmU8 >()

template LayerTestResult<int32_t, 1> RankDimSize2Test< armnn::DataType::QAsymmU8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize2Test< armnn::DataType::QSymmS16 >()

template LayerTestResult<int32_t, 1> RankDimSize2Test< armnn::DataType::QSymmS16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize2Test< armnn::DataType::QSymmS8 >()

template LayerTestResult<int32_t, 1> RankDimSize2Test< armnn::DataType::QSymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize2Test< armnn::DataType::Signed32 >()

template LayerTestResult<int32_t, 1> RankDimSize2Test< armnn::DataType::Signed32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize3Test()

LayerTestResult<int32_t, 1> RankDimSize3Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

Definition at line 80 of file RankTestImpl.cpp.

83 {
84  armnn::TensorInfo inputTensorInfo({1, 3, 2}, ArmnnType, 1.0f, 0);
85  auto input = MakeTensor<T, 3>(inputTensorInfo, ConvertToDataType<ArmnnType>(
86  { -37.5f, -15.2f, -8.76f, -2.0f, -1.5f, -1.3f},
87  inputTensorInfo));
88 
89  return RankTest<T, 3>(inputTensorInfo, input, workloadFactory, memoryManager);
90 }

◆ RankDimSize3Test< armnn::DataType::BFloat16 >()

template LayerTestResult<int32_t, 1> RankDimSize3Test< armnn::DataType::BFloat16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize3Test< armnn::DataType::Float16 >()

template LayerTestResult<int32_t, 1> RankDimSize3Test< armnn::DataType::Float16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize3Test< armnn::DataType::Float32 >()

template LayerTestResult<int32_t, 1> RankDimSize3Test< armnn::DataType::Float32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize3Test< armnn::DataType::QAsymmS8 >()

template LayerTestResult<int32_t, 1> RankDimSize3Test< armnn::DataType::QAsymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize3Test< armnn::DataType::QAsymmU8 >()

template LayerTestResult<int32_t, 1> RankDimSize3Test< armnn::DataType::QAsymmU8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize3Test< armnn::DataType::QSymmS16 >()

template LayerTestResult<int32_t, 1> RankDimSize3Test< armnn::DataType::QSymmS16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize3Test< armnn::DataType::QSymmS8 >()

template LayerTestResult<int32_t, 1> RankDimSize3Test< armnn::DataType::QSymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize3Test< armnn::DataType::Signed32 >()

template LayerTestResult<int32_t, 1> RankDimSize3Test< armnn::DataType::Signed32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize4Test()

LayerTestResult<int32_t, 1> RankDimSize4Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

Definition at line 93 of file RankTestImpl.cpp.

96 {
97  armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, ArmnnType, 1.0f, 0);
98  auto input = MakeTensor<T, 4>(inputTensorInfo, ConvertToDataType<ArmnnType>(
99  { -37.5f, -15.2f, -8.76f, -2.0f, -1.5f, -1.3f, -0.5f, -0.4f, 0.0f,
100  1.0f, 0.4f, 0.5f, 1.3f, 1.5f, 2.0f, 8.76f, 15.2f, 37.5f },
101  inputTensorInfo));
102 
103  return RankTest<T, 4>(inputTensorInfo, input, workloadFactory, memoryManager);
104 }

◆ RankDimSize4Test< armnn::DataType::BFloat16 >()

template LayerTestResult<int32_t, 1> RankDimSize4Test< armnn::DataType::BFloat16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize4Test< armnn::DataType::Float16 >()

template LayerTestResult<int32_t, 1> RankDimSize4Test< armnn::DataType::Float16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize4Test< armnn::DataType::Float32 >()

template LayerTestResult<int32_t, 1> RankDimSize4Test< armnn::DataType::Float32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize4Test< armnn::DataType::QAsymmS8 >()

template LayerTestResult<int32_t, 1> RankDimSize4Test< armnn::DataType::QAsymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize4Test< armnn::DataType::QAsymmU8 >()

template LayerTestResult<int32_t, 1> RankDimSize4Test< armnn::DataType::QAsymmU8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize4Test< armnn::DataType::QSymmS16 >()

template LayerTestResult<int32_t, 1> RankDimSize4Test< armnn::DataType::QSymmS16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize4Test< armnn::DataType::QSymmS8 >()

template LayerTestResult<int32_t, 1> RankDimSize4Test< armnn::DataType::QSymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankDimSize4Test< armnn::DataType::Signed32 >()

template LayerTestResult<int32_t, 1> RankDimSize4Test< armnn::DataType::Signed32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RankTest()

LayerTestResult<int32_t, 1> RankTest ( armnn::TensorInfo  inputTensorInfo,
boost::multi_array< T, n >  input,
armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

Definition at line 15 of file RankTestImpl.cpp.

References ARMNN_NO_DEPRECATE_WARN_BEGIN, ARMNN_NO_DEPRECATE_WARN_END, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), IWorkloadFactory::CreateRank(), IWorkloadFactory::CreateTensorHandle(), armnn::IgnoreUnused(), LayerTestResult< T, n >::output, LayerTestResult< T, n >::outputExpected, armnn::Scalar, and armnn::Signed32.

20 {
21  IgnoreUnused(memoryManager);
22 
23  const armnn::TensorShape outputShape{armnn::Dimensionality::Scalar};
24  armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32);
25 
26  LayerTestResult<int32_t , 1> ret(outputTensorInfo);
27  ret.outputExpected = MakeTensor<uint32_t, 1>(outputTensorInfo, { n });
28 
29  ARMNN_NO_DEPRECATE_WARN_BEGIN
30  std::unique_ptr<armnn::ITensorHandle> inputHandle = workloadFactory.CreateTensorHandle(inputTensorInfo);
31  std::unique_ptr<armnn::ITensorHandle> outputHandle = workloadFactory.CreateTensorHandle(outputTensorInfo);
32  ARMNN_NO_DEPRECATE_WARN_END
33 
34  armnn::RankQueueDescriptor data;
35  armnn::WorkloadInfo info;
36  AddInputToWorkload(data, info, inputTensorInfo, inputHandle.get());
37  AddOutputToWorkload(data, info, outputTensorInfo, outputHandle.get());
38 
39  std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateRank(data, info);
40 
41  inputHandle->Allocate();
42  outputHandle->Allocate();
43 
44  CopyDataToITensorHandle(inputHandle.get(), input.origin());
45 
46  workload->Execute();
47 
48  CopyDataFromITensorHandle(&ret.output[0], outputHandle.get());
49 
50  return ret;
51 }
ARMNN_NO_DEPRECATE_WARN_BEGIN
#define ARMNN_NO_DEPRECATE_WARN_BEGIN
Definition: Deprecated.hpp:33
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
armnn::IWorkloadFactory::CreateRank
virtual std::unique_ptr< IWorkload > CreateRank(const RankQueueDescriptor &descriptor, const WorkloadInfo &info) const
Definition: WorkloadFactory.cpp:1524
ARMNN_NO_DEPRECATE_WARN_END
#define ARMNN_NO_DEPRECATE_WARN_END
Definition: Deprecated.hpp:34
CopyDataFromITensorHandle
void CopyDataFromITensorHandle(void *memory, const armnn::ITensorHandle *tensorHandle)
Definition: TensorCopyUtils.cpp:14
armnn::IWorkloadFactory::CreateTensorHandle
virtual std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const =0
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition: WorkloadInfo.hpp:16
CopyDataToITensorHandle
void CopyDataToITensorHandle(armnn::ITensorHandle *tensorHandle, const void *memory)
Definition: TensorCopyUtils.cpp:9