diff options
Diffstat (limited to 'src/armnn/test/OptimizerTests.cpp')
| -rw-r--r-- | src/armnn/test/OptimizerTests.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/test/OptimizerTests.cpp b/src/armnn/test/OptimizerTests.cpp index 8416a8dd0d..3cea1b540e 100644 --- a/src/armnn/test/OptimizerTests.cpp +++ b/src/armnn/test/OptimizerTests.cpp @@ -714,7 +714,8 @@ TEST_CASE("BackendHintTest") case armnn::LayerType::Input: { auto inputLayer = PolymorphicDowncast<const InputLayer*>(layer); - CHECK((inputLayer->GetBackendId() == "MockBackend")); + const auto connectedLayerBackendId = inputLayer->GetOutputSlot(0).GetOwningLayer().GetBackendId(); + CHECK((inputLayer->GetBackendId() == connectedLayerBackendId)); break; } case armnn::LayerType::Output: |