diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2024-03-05 17:59:27 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-03-12 16:57:28 +0000 |
| commit | 70dc5e9dc375c2feb306c7e709078c9f61ef8b1a (patch) | |
| tree | f85bea5ae8fdffe513d44edfc616d8a0eb690936 /src/armnn/layers/ScatterNdLayer.cpp | |
| parent | ce7f51f0a8f815ff0e0dd94e209a8cf1802ac914 (diff) | |
| download | armnn-70dc5e9dc375c2feb306c7e709078c9f61ef8b1a.tar.gz | |
IVGCVSW-8233 ScatterNd End to End tests added
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Id89233954dd8da600c2f82e718df849b098c8af4
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) |