From 80de4c16421207874c29d91b9617eb05f68e43c7 Mon Sep 17 00:00:00 2001
From: Alexander Bengtsson <Alexander.Bengtsson@arm.com>
Date: Mon, 31 Mar 2025 17:14:02 +0200
Subject: [PATCH] MLBEDSW-10630: Allow 32-bit reshape-like operations

Change-Id: I264f98bb481c4a72903f6aa8c4c141b5c9d098b2
Signed-off-by: Alexander Bengtsson <Alexander.Bengtsson@arm.com>
---
 ethosu/regor/tflite/tflite_supported_operators_u55.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/ethosu/regor/tflite/tflite_supported_operators_u55.cpp b/ethosu/regor/tflite/tflite_supported_operators_u55.cpp
index 4b757201..33eeaf47 100644
--- a/ethosu/regor/tflite/tflite_supported_operators_u55.cpp
+++ b/ethosu/regor/tflite/tflite_supported_operators_u55.cpp
@@ -166,7 +166,6 @@ bool TfLiteSupportedOperatorsU55::Constraint32bitOps(const Operation *op)
 {
     static const std::unordered_set<OpType> supported = {
         OpType::ReduceSum,
-        OpType::Shape,
         OpType::ArgMax,
         OpType::Transpose,
         OpType::MirrorPad,
@@ -175,6 +174,11 @@ bool TfLiteSupportedOperatorsU55::Constraint32bitOps(const Operation *op)
         OpType::Sub,
         OpType::BatchMatMul,
         OpType::FullyConnected,
+        OpType::Reshape,
+        OpType::QuantizedReshape,
+        OpType::Squeeze,
+        OpType::ExpandDims,
+        OpType::Identity,
     };
 
     OpType opType = op->Type();
-- 
GitLab