diff --git a/ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp b/ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp index 37933d51b22b5b22cadac856e4411adba3711e95..49742ceebed6c139cf3f967947998b9ed60e6157 100644 --- a/ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp +++ b/ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp @@ -180,9 +180,10 @@ bool EthosU55Constraints::SupportedDtypes(OpType opType, DataType ifmType, DataT return false; } - if ( _arch->UseAvgPoolNop(opType) ) + if ( _arch->UseAvgPoolNop(opType) || opType == OpType::Rescale ) { - // The rules for UseAvgPoolNop are not the same as for a Pooling operation, so skip checks for now + // TODO MLBEDSW-10667: The rules for UseAvgPoolNop are not the same as for a Pooling operation, so skip checks + // for now return true; } diff --git a/ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp b/ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp index a7bc9315365867e80c5bf175d86e238d235d7273..f43a84077c898eb103215c7fedef139534a6d536 100644 --- a/ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp +++ b/ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp @@ -219,7 +219,8 @@ bool EthosU85Constraints::SupportedDtypes(OpType opType, DataType ifmType, DataT if ( _arch->UseAvgPoolNop(opType) ) { - // The rules for UseAvgPoolNop are not the same as for a Pooling operation, skip checks for now + // TODO MLBEDSW-10667: The rules for UseAvgPoolNop are not the same as for a Pooling operation, skip checks for + // now return true; }