diff --git a/ethosu/regor/tflite/tflite_supported_operators_u55.cpp b/ethosu/regor/tflite/tflite_supported_operators_u55.cpp index 4b75720110ad09e79db920c795d10466199420d0..33eeaf478b6019715625187f4f847daa69fe085b 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 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();