diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-01-12 17:23:53 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-01-13 15:12:55 +0000 |
commit | 9c4c4498e1e72a55af3a28aeee17c0f14c5f8c8c (patch) | |
tree | 1acc486220990e1f1d8410f819ac8ae57cac776a /src | |
parent | d7fca093be53b314e100e15a8fa080cb506b60a3 (diff) | |
download | armnn-9c4c4498e1e72a55af3a28aeee17c0f14c5f8c8c.tar.gz |
Do the same for both ShapeInferenceMethod in Elementwise inferOutputShape
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Iefe255ed27e1f69d99dc0cae9cfa1d6d1a51720e
Diffstat (limited to 'src')
-rw-r--r-- | src/armnn/layers/ElementwiseBaseLayer.cpp | 19 |
1 files changed, 2 insertions, 17 deletions
diff --git a/src/armnn/layers/ElementwiseBaseLayer.cpp b/src/armnn/layers/ElementwiseBaseLayer.cpp index 87093f684e..87fddfea79 100644 --- a/src/armnn/layers/ElementwiseBaseLayer.cpp +++ b/src/armnn/layers/ElementwiseBaseLayer.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017-2018,2020-2021,2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -26,20 +26,7 @@ std::vector<TensorShape> ElementwiseBaseLayer::InferOutputShapes(const std::vect TensorShape input0 = inputShapes[0]; TensorShape input1 = inputShapes[1]; - if (m_ShapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) - { - if (input0.GetNumDimensions() != input1.GetNumDimensions()) - { - std::stringstream errorMessage; - errorMessage << GetLayerTypeAsCString(GetType()) << " layer \"" << GetName() << "\": "; - errorMessage << "The tensor inputs to an element-wise operator are expected to have equal number of " - "dimensions. First = " - << input0.GetNumDimensions() << " second = " << input1.GetNumDimensions(); - throw InvalidArgumentException(errorMessage.str(), CHECK_LOCATION()); - } - } - else if (m_ShapeInferenceMethod == ShapeInferenceMethod::InferAndValidate && - inputShapes[0].GetNumDimensions() < inputShapes[1].GetNumDimensions()) + if (inputShapes[0].GetNumDimensions() < inputShapes[1].GetNumDimensions()) { input1 = inputShapes[0]; input0 = inputShapes[1]; @@ -55,11 +42,9 @@ std::vector<TensorShape> ElementwiseBaseLayer::InferOutputShapes(const std::vect unsigned int dim0 = input0[i]; unsigned int dim1 = input1[i - shiftedDims]; -#if !NDEBUG // Validate inputs are broadcast compatible. ARMNN_ASSERT_MSG(dim0 == dim1 || dim0 == 1 || dim1 == 1, "Dimensions should either match or one should be of size 1."); -#endif dims[i] = std::max(dim0, dim1); } |