fix (#17179)
This commit is contained in:
@@ -326,7 +326,7 @@ bool AclEltwiseExecutor::init(const EltwiseAttrs &eltwiseAttrs, const std::vecto
|
||||
break;
|
||||
case Algorithm::EltwiseSwish:
|
||||
if (!NEActivationLayer::validate(&srcTensorsInfo[0], &dstTensorsInfo[0],
|
||||
{ActivationLayerInfo::ActivationFunction::SWISH, aclEltwiseAttrs.beta}))
|
||||
{ActivationLayerInfo::ActivationFunction::SWISH, aclEltwiseAttrs.alpha}))
|
||||
return false;
|
||||
exec_func = [this]{
|
||||
auto acl_op = std::make_unique<NEActivationLayer>();
|
||||
|
||||
@@ -60,7 +60,7 @@ public:
|
||||
// case Algorithm::EltwisePowerDynamic: // TODO: ACL version doesn't work https://github.com/ARM-software/ComputeLibrary/issues/1047
|
||||
case Algorithm::EltwiseSoftRelu:
|
||||
case Algorithm::EltwiseClamp:
|
||||
case Algorithm::EltwiseSwish:
|
||||
//case Algorithm::EltwiseSwish: // TODO: efficientdet-d0 accuracy drops if ACL Swish is used
|
||||
case Algorithm::EltwisePrelu:
|
||||
case Algorithm::EltwiseHswish:
|
||||
if (!(checkPrecision({Precision::FP16, Precision::FP16}, Precision::FP16) ||
|
||||
|
||||
Reference in New Issue
Block a user