ArmNN
 21.02
ReshapeTestImpl.cpp File Reference

Go to the source code of this file.

Functions

template<armnn::DataType ArmnnType, typename T >
LayerTestResult< T, 4 > SimpleReshapeTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T >
LayerTestResult< T, 5 > Reshape5dTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< uint8_t, 2 > ReshapeBooleanTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::Float32 >, 4 > SimpleReshapeTest< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmS8 >, 4 > SimpleReshapeTest< armnn::DataType::QAsymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmU8 >, 4 > SimpleReshapeTest< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QSymmS16 >, 4 > SimpleReshapeTest< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::Float32 >, 5 > Reshape5dTest< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmS8 >, 5 > Reshape5dTest< armnn::DataType::QAsymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmU8 >, 5 > Reshape5dTest< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QSymmS16 >, 5 > Reshape5dTest< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 

Function Documentation

◆ Reshape5dTest()

LayerTestResult<T, 5> Reshape5dTest ( armnn::IWorkloadFactory workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr memoryManager,
const armnn::ITensorHandleFactory tensorHandleFactory 
)

Definition at line 111 of file ReshapeTestImpl.cpp.

References TensorInfo::SetQuantizationScale().

115 {
116  armnn::TensorInfo inputTensorInfo;
117  armnn::TensorInfo outputTensorInfo;
118 
119  unsigned int inputShape[] = { 2, 2, 8, 1, 1 };
120  unsigned int outputShape[] = { 2, 2, 2, 2, 2 };
121 
122  inputTensorInfo = armnn::TensorInfo(5, inputShape, ArmnnType);
123  inputTensorInfo.SetQuantizationScale(1.0f);
124  outputTensorInfo = armnn::TensorInfo(5, outputShape, ArmnnType);
125  outputTensorInfo.SetQuantizationScale(1.0f);
126 
127  auto input = ConvertToDataType<ArmnnType>(
128  {
129  0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f,
130  8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f,
131 
132  16.0f, 17.0f, 18.0f, 19.0f, 20.0f, 21.0f, 22.0f, 23.0f,
133  24.0f, 25.0f, 26.0f, 27.0f, 28.0f, 29.0f, 30.0f, 31.0f,
134  },
135  inputTensorInfo);
136 
137  auto outputExpected = ConvertToDataType<ArmnnType>(
138  {
139  0.0f, 1.0f,
140  2.0f, 3.0f,
141 
142  4.0f, 5.0f,
143  6.0f, 7.0f,
144 
145 
146  8.0f, 9.0f,
147  10.0f, 11.0f,
148 
149  12.0f, 13.0f,
150  14.0f, 15.0f,
151 
152 
153 
154  16.0f, 17.0f,
155  18.0f, 19.0f,
156 
157  20.0f, 21.0f,
158  22.0f, 23.0f,
159 
160 
161  24.0f, 25.0f,
162  26.0f, 27.0f,
163 
164  28.0f, 29.0f,
165  30.0f, 31.0f,
166  },
167  outputTensorInfo);
168 
169  return SimpleReshapeTestImpl<T, 5>(
170  workloadFactory, memoryManager, tensorHandleFactory, inputTensorInfo, outputTensorInfo, input, outputExpected);
171 }
void SetQuantizationScale(float scale)
Definition: Tensor.cpp:464

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

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

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

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

◆ ReshapeBooleanTest()

LayerTestResult<uint8_t, 2> ReshapeBooleanTest ( armnn::IWorkloadFactory workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr memoryManager,
const armnn::ITensorHandleFactory tensorHandleFactory 
)

Definition at line 173 of file ReshapeTestImpl.cpp.

References armnn::Boolean, and TensorInfo::SetQuantizationScale().

177 {
178  armnn::TensorInfo inputTensorInfo;
179  armnn::TensorInfo outputTensorInfo;
180 
181  unsigned int inputShape[] = { 1, 4 };
182  unsigned int outputShape[] = { 2, 2 };
183 
184  inputTensorInfo = armnn::TensorInfo(2, inputShape, armnn::DataType::Boolean);
185  inputTensorInfo.SetQuantizationScale(1.0f);
186  outputTensorInfo = armnn::TensorInfo(2, outputShape, armnn::DataType::Boolean);
187  outputTensorInfo.SetQuantizationScale(1.0f);
188 
189  const std::vector<uint8_t> input = { true, false, false, true };
190 
191  const std::vector<uint8_t> outputExpected = { true, false, false, true };
192 
193  return SimpleReshapeTestImpl<uint8_t, 2>(
194  workloadFactory, memoryManager, tensorHandleFactory, inputTensorInfo, outputTensorInfo, input, outputExpected);
195 }
void SetQuantizationScale(float scale)
Definition: Tensor.cpp:464

◆ SimpleReshapeTest()

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

Definition at line 58 of file ReshapeTestImpl.cpp.

References TensorInfo::SetQuantizationScale().

62 {
63  armnn::TensorInfo inputTensorInfo;
64  armnn::TensorInfo outputTensorInfo;
65 
66  unsigned int inputShape[] = { 2, 2, 3, 3 };
67  unsigned int outputShape[] = { 2, 2, 9, 1 };
68 
69  inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType);
70  inputTensorInfo.SetQuantizationScale(1.0f);
71  outputTensorInfo = armnn::TensorInfo(4, outputShape, ArmnnType);
72  outputTensorInfo.SetQuantizationScale(1.0f);
73 
74  auto input = ConvertToDataType<ArmnnType>(
75  {
76  0.0f, 1.0f, 2.0f,
77  3.0f, 4.0f, 5.0f,
78  6.0f, 7.0f, 8.0f,
79 
80  9.0f, 10.0f, 11.0f,
81  12.0f, 13.0f, 14.0f,
82  15.0f, 16.0f, 17.0f,
83 
84  18.0f, 19.0f, 20.0f,
85  21.0f, 22.0f, 23.0f,
86  24.0f, 25.0f, 26.0f,
87 
88  27.0f, 28.0f, 29.0f,
89  30.0f, 31.0f, 32.0f,
90  33.0f, 34.0f, 35.0f,
91  },
92  inputTensorInfo);
93 
94  auto outputExpected = ConvertToDataType<ArmnnType>(
95  {
96  0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
97 
98  9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f, 17.0f,
99 
100  18.0f, 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f, 25.0f, 26.0f,
101 
102  27.0f, 28.0f, 29.0f, 30.0f, 31.0f, 32.0f, 33.0f, 34.0f, 35.0f,
103  },
104  outputTensorInfo);
105 
106  return SimpleReshapeTestImpl<T, 4>(
107  workloadFactory, memoryManager, tensorHandleFactory, inputTensorInfo, outputTensorInfo, input, outputExpected);
108 }
void SetQuantizationScale(float scale)
Definition: Tensor.cpp:464

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

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

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

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