10 template<armnn::DataType ArmnnType,
typename T>
16 const unsigned int inputShape[] = { 2, 2 };
18 std::vector<float> inputValues
24 std::vector<float> expectedOutputValues
30 return ElementwiseUnaryTestHelper<2, ArmnnType>(
41 template<armnn::DataType ArmnnType,
typename T>
47 const unsigned int inputShape[] = { 3, 1, 2 };
49 std::vector<float> inputValues
55 std::vector<float> expectedOutputValues
61 return ElementwiseUnaryTestHelper<3, ArmnnType>(
72 template<armnn::DataType ArmnnType,
typename T>
78 const unsigned int inputShape[] = { 1, 2 };
80 std::vector<float> inputValues
85 std::vector<float> expectedOutputValues
90 return ElementwiseUnaryTestHelper<2, ArmnnType>(
101 template<armnn::DataType ArmnnType,
typename T>
107 const unsigned int inputShape[] = { 1, 2 };
109 std::vector<float> inputValues
114 std::vector<float> expectedOutputValues
119 return ElementwiseUnaryTestHelper<2, ArmnnType>(
126 expectedOutputValues,
127 tensorHandleFactory);
135 Rsqrt2dTest<armnn::DataType::Float32>(
141 Rsqrt2dTest<armnn::DataType::Float16>(
147 Rsqrt2dTest<armnn::DataType::QAsymmS8>(
153 Rsqrt2dTest<armnn::DataType::QAsymmU8>(
159 Rsqrt2dTest<armnn::DataType::QSymmS16>(
165 Rsqrt3dTest<armnn::DataType::Float32>(
171 Rsqrt3dTest<armnn::DataType::Float16>(
177 Rsqrt3dTest<armnn::DataType::QAsymmS8>(
183 Rsqrt3dTest<armnn::DataType::QAsymmU8>(
189 Rsqrt3dTest<armnn::DataType::QSymmS16>(
195 RsqrtZeroTest<armnn::DataType::Float32>(
201 RsqrtNegativeTest<armnn::DataType::Float32>(
LayerTestResult< T, 3 > Rsqrt3dTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< T, 2 > RsqrtNegativeTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
LayerTestResult< T, 2 > Rsqrt2dTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< T, 2 > RsqrtZeroTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)