diff options
author | Colm Donelan <colm.donelan@arm.com> | 2024-04-04 11:20:29 +0100 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2024-04-18 09:12:22 +0000 |
commit | 02300aa6460441891e54342286358afe42f432c8 (patch) | |
tree | b36396e1b3052198af5c8c60fb31a8e5e2931dc5 /src/armnn/optimizations | |
parent | 4f1771ab4d321afba9f5a52411855b5dc33bf247 (diff) | |
download | armnn-02300aa6460441891e54342286358afe42f432c8.tar.gz |
IVGCVSW-8314 Broadcast handling for Comparison layer is inconsistent.
* Added Comparison and LogicalBinary to AddBroadcastReshapeLayer
optimization.
Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: I4f4bafb961daf63a733be9a1f17067fd246607ad
Diffstat (limited to 'src/armnn/optimizations')
-rw-r--r-- | src/armnn/optimizations/AddBroadcastReshapeLayer.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp index dbde72b917..b07ab54a3f 100644 --- a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp +++ b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2020-2021,2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2020-2021,2023-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -7,7 +7,6 @@ #include "Optimization.hpp" #include <armnn/backends/TensorHandle.hpp> -#include <armnn/utility/IgnoreUnused.hpp> #include <armnn/utility/PolymorphicDowncast.hpp> namespace armnn @@ -18,7 +17,8 @@ namespace optimizations static const std::set<armnn::LayerType> broadcastOps{ LayerType::Addition, LayerType::Division, LayerType::Maximum, LayerType::Minimum, LayerType::Multiplication, LayerType::Prelu, - LayerType::Subtraction, LayerType::ElementwiseBinary }; + LayerType::Subtraction, LayerType::ElementwiseBinary, + LayerType::Comparison, LayerType::LogicalBinary}; class AddBroadcastReshapeLayerImpl { |