diff options
author | Gunes Bayir <gunes.bayir@arm.com> | 2023-07-18 15:57:23 +0100 |
---|---|---|
committer | Gunes Bayir <gunes.bayir@arm.com> | 2023-08-14 09:57:15 +0000 |
commit | 338ef4699735db16d346e52f05b822f8fd5e3263 (patch) | |
tree | 91ffe06a4a3960ee577673a47be0a99629012679 /src/core/Utils.cpp | |
parent | 633ebd18d982496a6a626df3b8336f610ab09eb4 (diff) | |
download | ComputeLibrary-338ef4699735db16d346e52f05b822f8fd5e3263.tar.gz |
Optimize CLReduce for Min/Max Axis=0
Resolves: COMPMID-6400
Change-Id: Id9935f9727f77a824afc75c35f044e3f5c173e0d
Signed-off-by: Gunes Bayir <gunes.bayir@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10120
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/Utils.cpp')
-rw-r--r-- | src/core/Utils.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/Utils.cpp b/src/core/Utils.cpp index 0701ee7c90..1ca7adb3a8 100644 --- a/src/core/Utils.cpp +++ b/src/core/Utils.cpp @@ -390,7 +390,7 @@ bool needs_serialized_reduction(ReductionOperation op, DataType dt, unsigned int const bool is_quantized_type = is_data_type_quantized(dt); const bool is_first_dim = (axis == 0); - return !is_first_dim || is_min_max || is_quantized_type; + return !is_first_dim || (is_quantized_type && !is_min_max); } QuantizationInfo get_softmax_output_quantization_info(DataType input_type, bool is_log) |