diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-22 16:10:44 +0000 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-22 16:10:44 +0000 |
commit | 649dd9515ddf4bd00a0bff64d51dfd835a6c7b39 (patch) | |
tree | c938bc8eb11dd24223c0cb00a57d4372a907b943 /src/backends/backendsCommon/test/SoftmaxTestImpl.hpp | |
parent | 382e21ce95c04479a6900afca81a57949b369f1e (diff) | |
download | armnn-649dd9515ddf4bd00a0bff64d51dfd835a6c7b39.tar.gz |
IVGCVSW-2467 Remove GetDataType<T> function
Change-Id: I7359617a307b9abb4c30b3d5f2364dc6d0f828f0
Diffstat (limited to 'src/backends/backendsCommon/test/SoftmaxTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/SoftmaxTestImpl.hpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/backends/backendsCommon/test/SoftmaxTestImpl.hpp b/src/backends/backendsCommon/test/SoftmaxTestImpl.hpp index 97199e3c53..25ceda1128 100644 --- a/src/backends/backendsCommon/test/SoftmaxTestImpl.hpp +++ b/src/backends/backendsCommon/test/SoftmaxTestImpl.hpp @@ -19,7 +19,7 @@ #include <algorithm> -template<typename T> +template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 2> SimpleSoftmaxTestImpl( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, @@ -32,13 +32,13 @@ LayerTestResult<T, 2> SimpleSoftmaxTestImpl( unsigned int inputShape[] = { 2, 4 }; - inputTensorInfo = armnn::TensorInfo(2, inputShape, armnn::GetDataType<T>()); + inputTensorInfo = armnn::TensorInfo(2, inputShape, ArmnnType); float qScale = 1.f / 256.f; int qOffset = 0; inputTensorInfo.SetQuantizationScale(qScale); inputTensorInfo.SetQuantizationOffset(qOffset); - outputTensorInfo = armnn::TensorInfo(2, inputShape, armnn::GetDataType<T>()); + outputTensorInfo = armnn::TensorInfo(2, inputShape, ArmnnType); outputTensorInfo.SetQuantizationScale(qScale); outputTensorInfo.SetQuantizationOffset(qOffset); @@ -87,7 +87,7 @@ LayerTestResult<T, 2> SimpleSoftmaxTestImpl( return ret; } -template<typename T> +template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 2> CompareSoftmaxTestImpl( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, @@ -103,8 +103,8 @@ LayerTestResult<T, 2> CompareSoftmaxTestImpl( unsigned int inputShape[] = { batchSize, channels }; - inputTensorInfo = armnn::TensorInfo(2, inputShape, armnn::GetDataType<T>()); - outputTensorInfo = armnn::TensorInfo(2, inputShape, armnn::GetDataType<T>()); + inputTensorInfo = armnn::TensorInfo(2, inputShape, ArmnnType); + outputTensorInfo = armnn::TensorInfo(2, inputShape, ArmnnType); float qScale = 1.f / 256.f; int qOffset = 0; inputTensorInfo.SetQuantizationScale(qScale); |