ArmNN
 20.05
Functions
RsqrtTestImpl.cpp File Reference
#include "ReshapeTestImpl.hpp"
#include "ElementwiseUnaryTestImpl.hpp"

Go to the source code of this file.

Functions

template<armnn::DataType ArmnnType, typename T >
LayerTestResult< T, 2 > Rsqrt2dTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template<armnn::DataType ArmnnType, typename T >
LayerTestResult< T, 3 > Rsqrt3dTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template<armnn::DataType ArmnnType, typename T >
LayerTestResult< T, 2 > RsqrtZeroTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template<armnn::DataType ArmnnType, typename T >
LayerTestResult< T, 2 > RsqrtNegativeTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::Float32 >, 2 > Rsqrt2dTest< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::Float16 >, 2 > Rsqrt2dTest< armnn::DataType::Float16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmS8 >, 2 > Rsqrt2dTest< armnn::DataType::QAsymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmU8 >, 2 > Rsqrt2dTest< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QSymmS16 >, 2 > Rsqrt2dTest< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::Float32 >, 3 > Rsqrt3dTest< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::Float16 >, 3 > Rsqrt3dTest< armnn::DataType::Float16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmS8 >, 3 > Rsqrt3dTest< armnn::DataType::QAsymmS8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmU8 >, 3 > Rsqrt3dTest< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::QSymmS16 >, 3 > Rsqrt3dTest< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::Float32 >, 2 > RsqrtZeroTest< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 
template LayerTestResult< armnn::ResolveType< armnn::DataType::Float32 >, 2 > RsqrtNegativeTest< armnn::DataType::Float32 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
 

Function Documentation

◆ Rsqrt2dTest()

LayerTestResult<T, 2> Rsqrt2dTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

Definition at line 11 of file RsqrtTestImpl.cpp.

References armnn::Rsqrt.

14 {
15  const unsigned int inputShape[] = { 2, 2 };
16 
17  std::vector<float> inputValues
18  {
19  1.f, 4.f,
20  16.f, 25.f
21  };
22 
23  std::vector<float> expectedOutputValues
24  {
25  1.f, 0.5f,
26  0.25f, 0.2f
27  };
28 
29  return ElementwiseUnaryTestHelper<2, ArmnnType>(
30  workloadFactory,
31  memoryManager,
32  armnn::UnaryOperation::Rsqrt,
33  inputShape,
34  inputValues,
35  inputShape,
36  expectedOutputValues);
37 }

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::Float16>, 2> Rsqrt2dTest< armnn::DataType::Float16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 2> Rsqrt2dTest< armnn::DataType::Float32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmS8>, 2> Rsqrt2dTest< armnn::DataType::QAsymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmU8>, 2> Rsqrt2dTest< armnn::DataType::QAsymmU8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 2> Rsqrt2dTest< armnn::DataType::QSymmS16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ Rsqrt3dTest()

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

Definition at line 40 of file RsqrtTestImpl.cpp.

References armnn::Rsqrt.

43 {
44  const unsigned int inputShape[] = { 3, 1, 2 };
45 
46  std::vector<float> inputValues
47  {
48  1.f, 4.f, 16.f,
49  25.f, 64.f, 100.f
50  };
51 
52  std::vector<float> expectedOutputValues
53  {
54  1.f, 0.5f, 0.25f,
55  0.2f, 0.125f, 0.1f
56  };
57 
58  return ElementwiseUnaryTestHelper<3, ArmnnType>(
59  workloadFactory,
60  memoryManager,
61  armnn::UnaryOperation::Rsqrt,
62  inputShape,
63  inputValues,
64  inputShape,
65  expectedOutputValues);
66 }

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::Float16>, 3> Rsqrt3dTest< armnn::DataType::Float16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 3> Rsqrt3dTest< armnn::DataType::Float32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmS8>, 3> Rsqrt3dTest< armnn::DataType::QAsymmS8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmU8>, 3> Rsqrt3dTest< armnn::DataType::QAsymmU8 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 3> Rsqrt3dTest< armnn::DataType::QSymmS16 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RsqrtNegativeTest()

LayerTestResult<T, 2> RsqrtNegativeTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

Definition at line 96 of file RsqrtTestImpl.cpp.

References armnn::Rsqrt.

99 {
100  const unsigned int inputShape[] = { 1, 2 };
101 
102  std::vector<float> inputValues
103  {
104  -25.f, -16.f
105  };
106 
107  std::vector<float> expectedOutputValues
108  {
109  -NAN, -NAN
110  };
111 
112  return ElementwiseUnaryTestHelper<2, ArmnnType>(
113  workloadFactory,
114  memoryManager,
115  armnn::UnaryOperation::Rsqrt,
116  inputShape,
117  inputValues,
118  inputShape,
119  expectedOutputValues);
120 }

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 2> RsqrtNegativeTest< armnn::DataType::Float32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

◆ RsqrtZeroTest()

LayerTestResult<T, 2> RsqrtZeroTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)

Definition at line 69 of file RsqrtTestImpl.cpp.

References armnn::Rsqrt.

72 {
73  const unsigned int inputShape[] = { 1, 2 };
74 
75  std::vector<float> inputValues
76  {
77  0.f, -0.f
78  };
79 
80  std::vector<float> expectedOutputValues
81  {
82  INFINITY, -INFINITY
83  };
84 
85  return ElementwiseUnaryTestHelper<2, ArmnnType>(
86  workloadFactory,
87  memoryManager,
88  armnn::UnaryOperation::Rsqrt,
89  inputShape,
90  inputValues,
91  inputShape,
92  expectedOutputValues);
93 }

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

template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 2> RsqrtZeroTest< armnn::DataType::Float32 > ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager 
)