diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-05-15 13:03:54 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2019-05-16 13:53:26 +0100 |
commit | cbb66aa4b7ec93e9a64a1dec5ebc4158056ec061 (patch) | |
tree | 3cfa5bb1cb4fe687d8b8530c1b5a081aea9877df /src/backends/reference/RefLayerSupport.cpp | |
parent | 60281579b022439ad650e9cec9a2cbfb3c2d500d (diff) | |
download | armnn-cbb66aa4b7ec93e9a64a1dec5ebc4158056ec061.tar.gz |
IVGCVSW-2972 Support QSymm16 for Merger
Change-Id: If2289b2d1fc20f4524fcf4620f01ea3a36d727d2
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
-rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 26 |
1 files changed, 21 insertions, 5 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index f79c152139..858bd878ae 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -729,11 +729,27 @@ bool RefLayerSupport::IsMergerSupported(const std::vector<const TensorInfo*> inp Optional<std::string&> reasonIfUnsupported) const { ignore_unused(descriptor); - ignore_unused(output); - return IsSupportedForDataTypeRef(reasonIfUnsupported, - inputs[0]->GetDataType(), - &TrueFunc<>, - &TrueFunc<>); + + bool supported = true; + std::array<DataType,3> supportedTypes = + { + DataType::Float32, + DataType::QuantisedAsymm8, + DataType::QuantisedSymm16 + }; + + supported &= CheckSupportRule(TypeAnyOf(output, supportedTypes), reasonIfUnsupported, + "Reference concatenation: output type not supported"); + for (const TensorInfo* input : inputs) + { + supported &= CheckSupportRule(TypeAnyOf(*input, supportedTypes), reasonIfUnsupported, + "Reference concatenation: input type not supported"); + + supported &= CheckSupportRule(TypesAreEqual(*input, output), reasonIfUnsupported, + "Reference concatenation: input and output types mismatched."); + } + + return supported; } bool RefLayerSupport::IsMemCopySupported(const TensorInfo &input, |