diff options
author | ramelg01 <ramy.elgammal@arm.com> | 2021-10-29 10:52:53 +0100 |
---|---|---|
committer | ramy.elgammal <ramy.elgammal@arm.com> | 2021-11-04 11:10:56 +0000 |
commit | 6049edadf0c89a026b3fcd1927ee7531d3c40278 (patch) | |
tree | c12fcea637e41cdb9e1f72dc734e4a87d0b31981 /utils/TypePrinter.h | |
parent | 71cbd28b7cf5115b0451d43e5c84cce4ae4d8ec7 (diff) | |
download | ComputeLibrary-6049edadf0c89a026b3fcd1927ee7531d3c40278.tar.gz |
Add PRelu to supported PostOps in:
- ClGemmMatrixMultiplyReshapedKernel
- ClGemmMatrixMultiplyNativeKernel
- ClGemmMatrixMultiplyReshapedOnlyRhsKernel
Resolves: COMPMID-4713
Change-Id: I3adcb1b3d4af37ebcbc3bee19cc1845885d08600
Signed-off-by: Ramy Elgammal <ramy.elgammal@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6553
Reviewed-by: SiCong Li <sicong.li@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'utils/TypePrinter.h')
-rw-r--r-- | utils/TypePrinter.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 950d32284a..785b41fc62 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -161,6 +161,11 @@ inline ::std::ostream &operator<<(::std::ostream &os, experimental::PostOpType p os << "Eltwise_Add"; break; } + case experimental::PostOpType::Eltwise_PRelu: + { + os << "Eltwise_PRelu"; + break; + } default: { ARM_COMPUTE_ERROR("Unsupported PostOpType"); @@ -208,6 +213,12 @@ inline ::std::ostream &operator<<(::std::ostream &os, const experimental::IPostO os << "convert_policy=" << _post_op->_policy; break; } + case experimental::PostOpType::Eltwise_PRelu: + { + const auto _post_op = utils::cast::polymorphic_downcast<const experimental::PostOpEltwisePRelu<T> *>(&post_op); + os << "convert_policy=" << _post_op->_policy; + break; + } default: { ARM_COMPUTE_ERROR("Unsupported PostOpType"); |