diff options
author | giuros01 <giuseppe.rossini@arm.com> | 2019-08-23 14:27:30 +0100 |
---|---|---|
committer | Giuseppe Rossini <giuseppe.rossini@arm.com> | 2019-08-30 13:37:28 +0000 |
commit | 351bd137e48c5276963274ac741b172483e98d21 (patch) | |
tree | 3ede92537c406d24f948acc51c1e6c0fac011036 /utils/GraphUtils.h | |
parent | ebe2e8ccc6f9504fdad95884a794be1e9f58803e (diff) | |
download | ComputeLibrary-351bd137e48c5276963274ac741b172483e98d21.tar.gz |
compmid-2573: Investigate FP16 Winograd reference implementations
Change-Id: I5a3e692c046a5ad28a676c03e3e51950c64cf503
Signed-off-by: giuros01 <giuseppe.rossini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1845
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'utils/GraphUtils.h')
-rw-r--r-- | utils/GraphUtils.h | 6 |
1 files changed, 6 insertions, 0 deletions
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 <typename T> + void preprocess_typed(ITensor &tensor); + std::array<float, 3> _mean; bool _bgr; float _scale; @@ -90,6 +93,9 @@ public: void preprocess(ITensor &tensor) override; private: + template <typename T> + void preprocess_typed(ITensor &tensor); + float _min_range; float _max_range; }; |