diff options
author | Eric Kunze <eric.kunze@arm.com> | 2024-03-19 21:14:07 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-04-08 17:06:08 +0000 |
commit | 9c0a5075d9e184f6b92762b3bc903e021b700e65 (patch) | |
tree | 7c8f2f69ccb2383bb5e99748a6a1ce73bc91cd1e /reference_model/src/ops/type_conversion.h | |
parent | ad8e1e25e805f6face5fcf0b3906cd06db46e1d7 (diff) | |
download | reference_model-9c0a5075d9e184f6b92762b3bc903e021b700e65.tar.gz |
Modify Rescale signedness check to look at attributes
Also simplify the check to align the pesudo code structure.
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Change-Id: I6023046026d2784dedd963b2b4d34a1117d45c23
Diffstat (limited to 'reference_model/src/ops/type_conversion.h')
-rw-r--r-- | reference_model/src/ops/type_conversion.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/reference_model/src/ops/type_conversion.h b/reference_model/src/ops/type_conversion.h index da5537e..cf95f16 100644 --- a/reference_model/src/ops/type_conversion.h +++ b/reference_model/src/ops/type_conversion.h @@ -43,6 +43,16 @@ public: using TMultiplierI32 = Eigen::Tensor<I32EigenType, 1>; using TShift = Eigen::Tensor<I8EigenType, 1>; + bool isI8(TOSA_REF_TYPE Dtype) + { + return Dtype == TOSA_REF_TYPE_INT8 || Dtype == TOSA_REF_TYPE_UINT8; + } + + bool isI16(TOSA_REF_TYPE Dtype) + { + return Dtype == TOSA_REF_TYPE_INT16 || Dtype == TOSA_REF_TYPE_UINT16; + } + protected: TosaRescaleAttribute* attribute; TosaReference::TensorTemplate<TIn>* in; |