diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-04-04 13:33:10 +0100 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-04-04 13:33:10 +0100 |
commit | d5fd9767d426ca465eb660be062eecce51ad1097 (patch) | |
tree | b2a2b684c9072dda7b52b40e579a5b0c4aae3c21 /ConversionUtils.cpp | |
parent | 16196e267833178dae62926ff090c50ec6813ad4 (diff) | |
download | android-nn-driver-d5fd9767d426ca465eb660be062eecce51ad1097.tar.gz |
IVGCVSW-2886 Support multiple backends in Android driver
Change-Id: I4abe1f750801911570b6dc65c187b828c5929b5f
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'ConversionUtils.cpp')
-rw-r--r-- | ConversionUtils.cpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/ConversionUtils.cpp b/ConversionUtils.cpp index 60d1a1f4..fb71c759 100644 --- a/ConversionUtils.cpp +++ b/ConversionUtils.cpp @@ -150,12 +150,12 @@ armnn::IConnectableLayer* ProcessActivation(const armnn::TensorInfo& tensorInfo, } } - if (!IsLayerSupported(__func__, - armnn::IsActivationSupported, - data.m_Compute, - prevLayer->GetOutputSlot(0).GetTensorInfo(), - tensorInfo, - activationDesc)) + if (!IsLayerSupportedForAnyBackend(__func__, + armnn::IsActivationSupported, + data.m_Backends, + prevLayer->GetOutputSlot(0).GetTensorInfo(), + tensorInfo, + activationDesc)) { return nullptr; } @@ -169,4 +169,4 @@ armnn::IConnectableLayer* ProcessActivation(const armnn::TensorInfo& tensorInfo, return activationLayer; } -} // namespace armnn_driver
\ No newline at end of file +} // namespace armnn_driver |