diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2017-06-26 14:18:47 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:16:42 +0100 |
commit | 172e57028ef14f2f8d6c56edc53c5c85f97e07cd (patch) | |
tree | b3fe8c05902f07fb2381cf6dfd893654c8ccb63f /src/core/CL/cl_kernels/fixed_point.h | |
parent | 579c0498e161215be1a36080b0b454e5198a992a (diff) | |
download | ComputeLibrary-172e57028ef14f2f8d6c56edc53c5c85f97e07cd.tar.gz |
COMPMID-425 Port CLBatchnormalization to support QS8/QS16
Change-Id: I46c93305f377666ea0915ff789b7dfdfff596087
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/78862
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'src/core/CL/cl_kernels/fixed_point.h')
-rw-r--r-- | src/core/CL/cl_kernels/fixed_point.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/core/CL/cl_kernels/fixed_point.h b/src/core/CL/cl_kernels/fixed_point.h index bb534f5a51..4de7fc576b 100644 --- a/src/core/CL/cl_kernels/fixed_point.h +++ b/src/core/CL/cl_kernels/fixed_point.h @@ -471,4 +471,16 @@ CONVERTQ_DOWN_SAT_IMPL(float16, qs16x16) CONVERTQ_UP_IMPL(qs8x16, float16) CONVERTQ_UP_IMPL(qs16x16, float16) +#define SQCVT_SAT_IMPL(type) \ + inline type sqcvt_##type##_sat(float a, int fixed_point_position) \ + { \ + return CONVERT_SAT((a * (1 << fixed_point_position) + ((a < 0) ? -0.5f : 0.5f)), type); \ + } + +SQCVT_SAT_IMPL(qs8) +SQCVT_SAT_IMPL(qs16) + +#define SQCVT_SAT_OP_EXPAND_STR(a, type, position) sqcvt_##type##_sat((a), (position)) +#define SQCVT_SAT_OP_EXPAND(a, type, position) SQCVT_SAT_OP_EXPAND_STR((a), type, position) + #endif // ARM_COMPUTE_FIXED_POINT_H |