aboutsummaryrefslogtreecommitdiff
path: root/src/cpu/kernels/CpuQuantizeKernel.h
diff options
context:
space:
mode:
authorRamy Elgammal <ramy.elgammal@arm.com>2024-05-14 15:21:07 +0100
committerPablo Marquez Tello <pablo.tello@arm.com>2024-05-14 18:35:14 +0000
commitb4b61a649deef5374fa070b79562ae83d7f64f23 (patch)
tree946ef0967cb3a5575f3e3a76eeaee4cd42b25bce /src/cpu/kernels/CpuQuantizeKernel.h
parent2217f1e60964fe586cae7ef996af7ef1c0bef2ab (diff)
downloadComputeLibrary-b4b61a649deef5374fa070b79562ae83d7f64f23.tar.gz
Rework CpuQuantizeKernel to enable FP16 in multi_isa builds
Resolves: COMPMID-7054 Signed-off-by: Ramy Elgammal <ramy.elgammal@arm.com> Change-Id: I68d125b81ad7f74b2594ccda8d6ec08beef1ebd7 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11555 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com> Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/CpuQuantizeKernel.h')
-rw-r--r--src/cpu/kernels/CpuQuantizeKernel.h26
1 files changed, 1 insertions, 25 deletions
diff --git a/src/cpu/kernels/CpuQuantizeKernel.h b/src/cpu/kernels/CpuQuantizeKernel.h
index c2f7ac6d9d..750310c811 100644
--- a/src/cpu/kernels/CpuQuantizeKernel.h
+++ b/src/cpu/kernels/CpuQuantizeKernel.h
@@ -76,31 +76,7 @@ private:
*
* @param[in] window Region on which to execute the kernel.
*/
- using QuantizeFunctionExecutorPtr = void (CpuQuantizeKernel::*)(const ITensor *src,
- ITensor *dst,
- const Window &window);
- /** Function to apply QASYMM8 or QASYMM8_SIGNED quantization on a tensor.
- *
- * @param[in] window Region on which to execute the kernel.
- */
- template <typename TIn, typename TOut>
- void run_quantize_qasymm8(const ITensor *src, ITensor *dst, const Window &window);
- /** Function to apply QASYMM16 quantization on a tensor.
- *
- * @param[in] window Region on which to execute the kernel.
- */
- template <typename T>
- void run_quantize_qasymm16(const ITensor *src, ITensor *dst, const Window &window);
-
- template <typename TIn, typename TOut>
- void run_quantize_qsymm8(const ITensor *src, ITensor *dst, const Window &window);
-
- template <typename TIn, typename TOut>
- void run_requantize_offset_only(const ITensor *src, ITensor *dst, const Window &window);
-
- template <typename TIn, typename TOut>
- void run_requantize_offset_only_convert(const ITensor *src, ITensor *dst, const Window &window);
-
+ using QuantizeFunctionExecutorPtr = void (*)(const ITensor *src, ITensor *dst, const Window &window);
QuantizeFunctionExecutorPtr _func{nullptr};
size_t _split_dimension{Window::DimY};
};