diff options
Diffstat (limited to 'src/armnn/layers/ReduceLayer.cpp')
-rw-r--r-- | src/armnn/layers/ReduceLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/ReduceLayer.cpp b/src/armnn/layers/ReduceLayer.cpp index ebee213659..21095dda74 100644 --- a/src/armnn/layers/ReduceLayer.cpp +++ b/src/armnn/layers/ReduceLayer.cpp @@ -49,7 +49,7 @@ void ReduceLayer::ValidateTensorShapesFromInputs() VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); - const TensorInfo& input = GetInputSlot(0).GetConnection()->GetTensorInfo(); + const TensorInfo& input = GetInputSlot(0).GetTensorInfo(); ARMNN_ASSERT_MSG(input.GetNumDimensions() > 0 && input.GetNumDimensions() <= 4, "ReduceLayer: Reduce supports up to 4D input."); |