diff options
author | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-23 14:29:06 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-23 14:22:19 +0000 |
commit | 88d44b893684be7a14776331c47e3408d29da109 (patch) | |
tree | 45338b2bdc59dd5e8cdeb2904b7d63f08af08d42 /src/backends/backendsCommon/WorkloadData.cpp | |
parent | e3a4245bb88175d5493b6660feb4117b8ec9230c (diff) | |
download | armnn-88d44b893684be7a14776331c47e3408d29da109.tar.gz |
IVGCVSW-3074 Extend the DepthwiseConvolution2d workload to support QSymm16
Change-Id: I47bb0f782acfa5b2d2fee9132875f9a655ea635e
Signed-off-by: Ruomei Yan <ruomei.yan@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.cpp')
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp index a3470ad601..a1c74df17b 100644 --- a/src/backends/backendsCommon/WorkloadData.cpp +++ b/src/backends/backendsCommon/WorkloadData.cpp @@ -32,6 +32,8 @@ DataType GetBiasDataType(DataType inputDataType) return DataType::Float32; case DataType::QuantisedAsymm8: return DataType::Signed32; + case DataType::QuantisedSymm16: + return DataType::Signed32; default: BOOST_ASSERT_MSG(false, "Invalid input data type"); return DataType::Float32; @@ -693,6 +695,22 @@ void DepthwiseConvolution2dQueueDescriptor::Validate(const WorkloadInfo& workloa ValidateTensorQuantizationMultiplier(workloadInfo.m_InputTensorInfos[0], m_Weight->GetTensorInfo(), workloadInfo.m_OutputTensorInfos[0], "DepthwiseConvolution2dQueueDescriptor", "input", "weights", "output"); + + // Check the supported data types + std::vector<DataType> supportedTypes = { + DataType::Float32, + DataType::QuantisedAsymm8, + DataType::QuantisedSymm16, + DataType::Float16 + }; + + ValidateDataTypes(workloadInfo.m_InputTensorInfos[0], + supportedTypes, + "DepthwiseConvolution2dQueueDescriptor"); + + ValidateDataTypes(workloadInfo.m_OutputTensorInfos[0], + {workloadInfo.m_InputTensorInfos[0].GetDataType()}, + "DepthwiseConvolution2dQueueDescriptor"); } void PermuteQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const |