diff --git a/ethosu/regor/test/test_tflite_supported_operators.cpp b/ethosu/regor/test/test_tflite_supported_operators.cpp index 987185d054186d0afd861a36be7c0e6eb5e3b57a..154c1be104d0f2aa4fb74749968af28f24e067a5 100644 --- a/ethosu/regor/test/test_tflite_supported_operators.cpp +++ b/ethosu/regor/test/test_tflite_supported_operators.cpp @@ -278,7 +278,6 @@ TEST_CASE("Supported operators EthosU85") DataType::Float16, DataType::Float32, DataType::Float64, - DataType::Bool8, DataType::Complex, DataType::Complex64, DataType::Complex128, @@ -293,6 +292,7 @@ TEST_CASE("Supported operators EthosU85") DataType::Int16, DataType::Int32, DataType::Bool, + DataType::Bool8, DataType::Int64, }; for ( auto dtype : unsupported ) diff --git a/ethosu/regor/tflite/tflite_supported_operators.cpp b/ethosu/regor/tflite/tflite_supported_operators.cpp index 25d2296e931252a442cf155d44931d8b7fa2a8b9..7fdd133748589c636a5c931ef752e47837161b44 100644 --- a/ethosu/regor/tflite/tflite_supported_operators.cpp +++ b/ethosu/regor/tflite/tflite_supported_operators.cpp @@ -151,6 +151,14 @@ bool TfLiteSupportedOperators::ConstraintTensQuantized(const Operation *op) case OpType::Transpose: case OpType::GatherNd: case OpType::GatherV2: + case OpType::Select: + case OpType::SelectV2: + case OpType::ScatterNd: + case OpType::Pad: + case OpType::PadV2: + case OpType::ReduceAll: + case OpType::ReduceAny: + case OpType::ExpandDims: return true; default: break; diff --git a/ethosu/regor/tflite/tflite_supported_operators_u55.cpp b/ethosu/regor/tflite/tflite_supported_operators_u55.cpp index b65df18acda70363b726dc1beaed5a43893382fe..c3dc15821bcb6a6a709c52fba79775547409e847 100644 --- a/ethosu/regor/tflite/tflite_supported_operators_u55.cpp +++ b/ethosu/regor/tflite/tflite_supported_operators_u55.cpp @@ -53,6 +53,7 @@ TfLiteSupportedOperatorsU55::TfLiteSupportedOperatorsU55(IArchitectureConstraint OpType::Transpose, OpType::Mean, OpType::Sub, + OpType::Squeeze, OpType::StridedSlice, OpType::Exp, OpType::Split, @@ -64,11 +65,13 @@ TfLiteSupportedOperatorsU55::TfLiteSupportedOperatorsU55(IArchitectureConstraint OpType::Slice, OpType::TransposeConv2D, OpType::Tile, + OpType::ExpandDims, OpType::ReduceSum, OpType::Rsqrt, OpType::Pack, OpType::Unpack, OpType::LeakyRelu, + OpType::SquaredDifference, OpType::MirrorPad, OpType::Abs, OpType::SplitV, diff --git a/ethosu/regor/tflite/tflite_supported_operators_u85.cpp b/ethosu/regor/tflite/tflite_supported_operators_u85.cpp index 918f81d1972a9c864e94f3877cccb6f38789b2e8..a5db7132f0ca51039ceb64ff6854bb0ccb6ce37d 100644 --- a/ethosu/regor/tflite/tflite_supported_operators_u85.cpp +++ b/ethosu/regor/tflite/tflite_supported_operators_u85.cpp @@ -55,6 +55,7 @@ TfLiteSupportedOperatorsU85::TfLiteSupportedOperatorsU85(IArchitectureConstraint OpType::Mean, OpType::Sub, OpType::Div, + OpType::Squeeze, OpType::StridedSlice, OpType::Exp, OpType::Split, @@ -71,6 +72,7 @@ TfLiteSupportedOperatorsU85::TfLiteSupportedOperatorsU85(IArchitectureConstraint OpType::Slice, OpType::TransposeConv2D, OpType::Tile, + OpType::ExpandDims, OpType::Equal, OpType::NotEqual, OpType::ReduceSum, @@ -85,6 +87,7 @@ TfLiteSupportedOperatorsU85::TfLiteSupportedOperatorsU85(IArchitectureConstraint OpType::LogicalNot, OpType::ResizeNearestNeighbor, OpType::LeakyRelu, + OpType::SquaredDifference, OpType::MirrorPad, OpType::Abs, OpType::SplitV, @@ -108,7 +111,8 @@ TfLiteSupportedOperatorsU85::TfLiteSupportedOperatorsU85(IArchitectureConstraint DataType::Int16, DataType::Int32, DataType::Int64, - DataType::Bool + DataType::Bool, + DataType::Bool8 // clang-format on }; _checks = {