diff options
Diffstat (limited to 'src/armnn/layers/ScatterNdLayer.cpp')
-rw-r--r-- | src/armnn/layers/ScatterNdLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/ScatterNdLayer.cpp b/src/armnn/layers/ScatterNdLayer.cpp index a0b270fba5..853b87f9db 100644 --- a/src/armnn/layers/ScatterNdLayer.cpp +++ b/src/armnn/layers/ScatterNdLayer.cpp @@ -77,7 +77,7 @@ void ScatterNdLayer::ValidateTensorShapesFromInputs() // No input tensor, only shape provided via input slot // In this case, we cannot validate the output shape from the input shape, but we can // validate that the dimensions of shape and output tensor matched - unsigned int shapeDims = GetInputSlot(0).GetTensorInfo().GetNumDimensions(); + unsigned int shapeDims = GetInputSlot(0).GetTensorInfo().GetShape().GetNumElements(); unsigned int outputDims = GetOutputSlot(0).GetTensorInfo().GetNumDimensions(); if (shapeDims != outputDims) |