ArmNN
 20.11
Functions
MeanTestImpl.hpp File Reference
#include "LayerTestResult.hpp"
#include <ResolveType.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <backendsCommon/WorkloadFactory.hpp>

Go to the source code of this file.

Functions

template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 1 > MeanSimpleTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 3 > MeanSimpleAxisTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > MeanKeepDimsTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > MeanMultipleDimsTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 1 > MeanVts1Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 3 > MeanVts2Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 3 > MeanVts3Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 

Function Documentation

◆ MeanKeepDimsTest()

LayerTestResult<T, 4> MeanKeepDimsTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 109 of file MeanTestImpl.hpp.

113 {
114  const unsigned int inputShape[] = { 1, 1, 3, 2 };
115  const unsigned int outputShape[] = { 1, 1, 1, 2 };
116 
117  std::vector<float> input({ 1.5f, 1.5f, 2.5f, 2.5f, 3.5f, 3.5f });
118  std::vector<float> output({ 2.5f, 2.5f });
119 
120  return MeanTestHelper<ArmnnType, T, 4, 4>(
121  workloadFactory, memoryManager, tensorHandleFactory, inputShape, input, { 2 }, true, outputShape, output);
122 }

◆ MeanMultipleDimsTest()

LayerTestResult<T, 4> MeanMultipleDimsTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 125 of file MeanTestImpl.hpp.

129 {
130  const unsigned int inputShape[] = { 2, 3, 1, 2 };
131  const unsigned int outputShape[] = { 1, 3, 1, 1 };
132 
133  std::vector<float> input({ 1.5f, 2.5f, 3.5f, 4.5f, 5.5f, 6.5f, 1.5f, 2.5f, 3.5f, 4.5f, 5.5f, 6.5 });
134  std::vector<float> output({ 2.0f, 4.0f, 6.0f });
135 
136  return MeanTestHelper<ArmnnType, T, 4, 4>(
137  workloadFactory, memoryManager, tensorHandleFactory,
138  inputShape, input, { 0, 3 }, true, outputShape, output);
139 }

◆ MeanSimpleAxisTest()

LayerTestResult<T, 3> MeanSimpleAxisTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 93 of file MeanTestImpl.hpp.

97 {
98  const unsigned int inputShape[] = { 2, 3, 1, 2 };
99  const unsigned int outputShape[] = { 3, 1, 2 };
100 
101  std::vector<float> input({ 1.5f, 2.5f, 3.5f, 4.5f, 5.5f, 6.5f, 1.5f, 2.5f, 3.5f, 4.5f, 5.5f, 6.5f });
102  std::vector<float> output({ 1.5f, 2.5f, 3.5f, 4.5f, 5.5f, 6.5f });
103 
104  return MeanTestHelper<ArmnnType, T, 4, 3>(
105  workloadFactory, memoryManager, tensorHandleFactory, inputShape, input, { 0 }, false, outputShape, output);
106 }

◆ MeanSimpleTest()

LayerTestResult<T, 1> MeanSimpleTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 77 of file MeanTestImpl.hpp.

81 {
82  const unsigned int inputShape[] = { 3, 2 };
83  const unsigned int outputShape[] = { 1 };
84 
85  std::vector<float> input({ 1.5f, 1.5f, 2.5f, 2.5f, 3.5f, 3.5f });
86  std::vector<float> output({ 2.5f });
87 
88  return MeanTestHelper<ArmnnType, T, 2, 1>(
89  workloadFactory, memoryManager, tensorHandleFactory, inputShape, input, {}, false, outputShape, output);
90 }

◆ MeanVts1Test()

LayerTestResult<T, 1> MeanVts1Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 142 of file MeanTestImpl.hpp.

146 {
147  const unsigned int inputShape[] = { 4, 3, 2 };
148  const unsigned int outputShape[] = { 2 };
149 
150  std::vector<float> input({ 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f,
151  15.0f, 16.0f, 17.0f, 18.0f, 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f });
152  std::vector<float> output({ 12.0f, 13.0f });
153 
154  return MeanTestHelper<ArmnnType, T, 3, 1>(
155  workloadFactory, memoryManager, tensorHandleFactory,
156  inputShape, input, { 0, 1 }, false, outputShape, output);
157 }

◆ MeanVts2Test()

LayerTestResult<T, 3> MeanVts2Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 160 of file MeanTestImpl.hpp.

164 {
165  const unsigned int inputShape[] = { 4, 3, 2 };
166  const unsigned int outputShape[] = { 1, 3, 1 };
167 
168  std::vector<float> input({ 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f,
169  15.0f, 16.0f, 17.0f, 18.0f, 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f });
170  std::vector<float> output({ 10.5f, 12.5f, 14.5f });
171 
172  return MeanTestHelper<ArmnnType, T, 3, 3>(
173  workloadFactory, memoryManager, tensorHandleFactory,
174  inputShape, input, { 0, 2 }, true, outputShape, output);
175 }

◆ MeanVts3Test()

LayerTestResult<T, 3> MeanVts3Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 178 of file MeanTestImpl.hpp.

182 {
183  const unsigned int inputShape[] = { 1, 2, 2, 1 };
184  const unsigned int outputShape[] = { 1, 2, 1 };
185 
186  std::vector<float> input({ 1.0f, 2.0f, 3.0f, 4.0f });
187  std::vector<float> output({ 1.5f, 3.5f });
188 
189  return MeanTestHelper<ArmnnType, T, 4, 3>(
190  workloadFactory, memoryManager, tensorHandleFactory, inputShape, input, { 2 }, false, outputShape, output);
191 }