diff options
Diffstat (limited to 'src/armnn/layers/SwitchLayer.cpp')
| -rw-r--r-- | src/armnn/layers/SwitchLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/SwitchLayer.cpp b/src/armnn/layers/SwitchLayer.cpp index eae6e0dfe2..4f0eb9b703 100644 --- a/src/armnn/layers/SwitchLayer.cpp +++ b/src/armnn/layers/SwitchLayer.cpp @@ -39,7 +39,7 @@ void SwitchLayer::ValidateTensorShapesFromInputs() GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape(), GetInputSlot(1).GetConnection()->GetTensorInfo().GetShape() }); - BOOST_ASSERT(inferredShapes.size() == 1); + BOOST_ASSERT(inferredShapes.size() == 2); ConditionalThrowIfNotEqual<LayerValidationException>( "SwitchLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.", |