From 351bd137e48c5276963274ac741b172483e98d21 Mon Sep 17 00:00:00 2001 From: giuros01 Date: Fri, 23 Aug 2019 14:27:30 +0100 Subject: compmid-2573: Investigate FP16 Winograd reference implementations Change-Id: I5a3e692c046a5ad28a676c03e3e51950c64cf503 Signed-off-by: giuros01 Reviewed-on: https://review.mlplatform.org/c/1845 Reviewed-by: Pablo Marquez Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- utils/GraphUtils.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'utils/GraphUtils.h') diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h index 3417135f17..4c25dd2460 100644 --- a/utils/GraphUtils.h +++ b/utils/GraphUtils.h @@ -70,6 +70,9 @@ public: void preprocess(ITensor &tensor) override; private: + template + void preprocess_typed(ITensor &tensor); + std::array _mean; bool _bgr; float _scale; @@ -90,6 +93,9 @@ public: void preprocess(ITensor &tensor) override; private: + template + void preprocess_typed(ITensor &tensor); + float _min_range; float _max_range; }; -- cgit v1.2.1