10 template<armnn::DataType ArmnnType,
typename T>
15 const unsigned int inputShape[] = { 2, 2 };
17 std::vector<float> inputValues
23 std::vector<float> expectedOutputValues
29 return ElementwiseUnaryTestHelper<2, ArmnnType>(
36 expectedOutputValues);
39 template<armnn::DataType ArmnnType,
typename T>
44 const unsigned int inputShape[] = { 3, 1, 2 };
46 std::vector<float> inputValues
52 std::vector<float> expectedOutputValues
58 return ElementwiseUnaryTestHelper<3, ArmnnType>(
65 expectedOutputValues);
68 template<armnn::DataType ArmnnType,
typename T>
73 const unsigned int inputShape[] = { 1, 2 };
75 std::vector<float> inputValues
80 std::vector<float> expectedOutputValues
85 return ElementwiseUnaryTestHelper<2, ArmnnType>(
92 expectedOutputValues);
95 template<armnn::DataType ArmnnType,
typename T>
100 const unsigned int inputShape[] = { 1, 2 };
102 std::vector<float> inputValues
107 std::vector<float> expectedOutputValues
112 return ElementwiseUnaryTestHelper<2, ArmnnType>(
119 expectedOutputValues);
127 Rsqrt2dTest<armnn::DataType::Float32>(
132 Rsqrt2dTest<armnn::DataType::Float16>(
137 Rsqrt2dTest<armnn::DataType::QAsymmU8>(
142 Rsqrt2dTest<armnn::DataType::QSymmS16>(
147 Rsqrt3dTest<armnn::DataType::Float32>(
152 Rsqrt3dTest<armnn::DataType::Float16>(
157 Rsqrt3dTest<armnn::DataType::QAsymmU8>(
162 Rsqrt3dTest<armnn::DataType::QSymmS16>(
167 RsqrtZeroTest<armnn::DataType::Float32>(
172 RsqrtNegativeTest<armnn::DataType::Float32>(
LayerTestResult< T, 2 > RsqrtNegativeTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
LayerTestResult< T, 2 > RsqrtZeroTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
LayerTestResult< T, 3 > Rsqrt3dTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
LayerTestResult< T, 2 > Rsqrt2dTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)