diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2019-05-28 08:15:28 +0100 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-05-28 09:13:56 +0000 |
commit | 46c09d0094b708c70bb4770693c9e704b1fbfeb1 (patch) | |
tree | 9d4c62b2fa6efe70b2d08e7fbd344f629aec86fa /src/backends/backendsCommon/WorkloadFactory.cpp | |
parent | af8b72d3a97baa0c86583b03321e8a8d3f4bc2aa (diff) | |
download | armnn-46c09d0094b708c70bb4770693c9e704b1fbfeb1.tar.gz |
IVGCVSW-2970 Support QSymm16 for FullyConnected workloads
* Add support for QSymm16 for FullyConnected
* Add templating to Uint8 RefLayerTest to test QSymm16
Change-Id: Ie6e989daf2ca966d6c6805b8017126eb77ebfec4
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/WorkloadFactory.cpp')
-rw-r--r-- | src/backends/backendsCommon/WorkloadFactory.cpp | 21 |
1 files changed, 1 insertions, 20 deletions
diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp index 0490a94864..763107123f 100644 --- a/src/backends/backendsCommon/WorkloadFactory.cpp +++ b/src/backends/backendsCommon/WorkloadFactory.cpp @@ -17,6 +17,7 @@ #include <backendsCommon/BackendRegistry.hpp> #include <backendsCommon/WorkloadFactory.hpp> #include <backendsCommon/IBackendInternal.hpp> +#include <backendsCommon/test/WorkloadTestUtils.hpp> #include <boost/cast.hpp> #include <boost/iterator/transform_iterator.hpp> @@ -40,26 +41,6 @@ const TensorInfo OverrideDataType(const TensorInfo& info, Optional<DataType> typ return TensorInfo(info.GetShape(), type.value(), info.GetQuantizationScale(), info.GetQuantizationOffset()); } -Optional<DataType> GetBiasTypeFromWeightsType(Optional<DataType> weightsType) -{ - if (!weightsType) - { - return weightsType; - } - - switch(weightsType.value()) - { - case DataType::Float16: - case DataType::Float32: - return weightsType; - case DataType::QuantisedAsymm8: - return DataType::Signed32; - default: - BOOST_ASSERT_MSG(false, "GetBiasTypeFromWeightsType(): Unsupported data type."); - } - return EmptyOptional(); -} - } // anonymous namespace bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId, |