aboutsummaryrefslogtreecommitdiff
path: root/src/backends/tosaCommon/TosaMappings.cpp
diff options
context:
space:
mode:
authorTianle Cheng <tianle.cheng@arm.com>2023-12-12 13:52:22 +0000
committerTianle Cheng <tianle.cheng@arm.com>2023-12-15 10:22:31 +0000
commit7790dc6531034778d92ba264fd61174bcff7051e (patch)
treefe24f0a8a1229296168db559501a7403d5cbf962 /src/backends/tosaCommon/TosaMappings.cpp
parentf0d8ec1f4be3ce5794a48c3d3449fe93f2e58d04 (diff)
downloadarmnn-7790dc6531034778d92ba264fd61174bcff7051e.tar.gz
IVGCVSW-8113 Update EndToEnd tests and TosaReference support for Maximum
* Added ElementwiseBinary EndToEnd tests with inputs of the same shape to avoid Reshape * Added Slice EndToEnd tests with 4D tensors * Added TosaReference support for Maximum and TosaRefEndToEnd tests Signed-off-by: Tianle Cheng <tianle.cheng@arm.com> Change-Id: I4fa24435a75559e00b110d0e542b4f2bf07b21b4
Diffstat (limited to 'src/backends/tosaCommon/TosaMappings.cpp')
-rw-r--r--src/backends/tosaCommon/TosaMappings.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/tosaCommon/TosaMappings.cpp b/src/backends/tosaCommon/TosaMappings.cpp
index 0bdb1fec3d..dff266d793 100644
--- a/src/backends/tosaCommon/TosaMappings.cpp
+++ b/src/backends/tosaCommon/TosaMappings.cpp
@@ -29,6 +29,11 @@ TosaSerializationBasicBlock* GetTosaMapping(const Layer* layer,
{
return ConvertElementwiseBinaryToTosaOperator(layer, type, inputs, outputs);
}
+ case LayerType::ElementwiseBinary:
+ {
+ auto binaryDesc = PolymorphicDowncast<const ElementwiseBinaryDescriptor*>(&descriptor);
+ return ConvertElementwiseBinaryToTosaOperator(layer, type, inputs, outputs, binaryDesc);
+ }
case LayerType::ElementwiseUnary:
{
auto unaryDesc = PolymorphicDowncast<const ElementwiseUnaryDescriptor*>(&descriptor);