aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonBackend.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon/NeonBackend.cpp')
-rw-r--r--src/backends/neon/NeonBackend.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp
index 240456f44b..d48b32bb45 100644
--- a/src/backends/neon/NeonBackend.cpp
+++ b/src/backends/neon/NeonBackend.cpp
@@ -163,7 +163,8 @@ OptimizationViews NeonBackend::OptimizeSubgraphView(const SubgraphView& subgraph
{
for (auto&& childInput : output->GetConnections())
{
- if (childInput->GetOwningLayer().GetType() == LayerType::Activation)
+ if ((childInput->GetOwningLayer().GetType() == LayerType::Activation) &&
+ (checkDataTypeInputandOutput(childInput->GetOwningLayer())))
{
Layer& child = childInput->GetOwningLayer();