diff options
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 |