diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2017-06-22 12:57:51 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | bbd9fb95daa08d6da67c567b40ca2cd032f7a2d3 (patch) | |
tree | c1401585f64396d6f22bb790442d8183f3a17a9e /tests/validation/TensorVisitors.h | |
parent | 2eac5bd444d16e4e81c427d5a99e1534b387e211 (diff) | |
download | ComputeLibrary-bbd9fb95daa08d6da67c567b40ca2cd032f7a2d3.tar.gz |
COMPMID-412: Port PoolingLayer to use fixed point 16.
Change-Id: I2005de4c7c14526996309826d33a0ec8e732d2d5
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/78720
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Steven Niu <steven.niu@arm.com>
Diffstat (limited to 'tests/validation/TensorVisitors.h')
-rw-r--r-- | tests/validation/TensorVisitors.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/validation/TensorVisitors.h b/tests/validation/TensorVisitors.h index 5ee7ae3a9f..193697acf0 100644 --- a/tests/validation/TensorVisitors.h +++ b/tests/validation/TensorVisitors.h @@ -27,6 +27,7 @@ #include "Tensor.h" #include "TensorOperations.h" #include "arm_compute/core/Error.h" +#include "arm_compute/core/Helpers.h" #include "arm_compute/runtime/Lut.h" #include "boost_wrapper.h" @@ -258,8 +259,8 @@ private: struct pooling_layer_visitor : public boost::static_visitor<> { public: - explicit pooling_layer_visitor(const TensorVariant &in, PoolingLayerInfo pool_info, int fixed_point_position = 0) - : _in(in), _pool_info(pool_info), _fixed_point_position(fixed_point_position) + explicit pooling_layer_visitor(const TensorVariant &in, PoolingLayerInfo pool_info) + : _in(in), _pool_info(pool_info) { } @@ -267,13 +268,12 @@ public: void operator()(Tensor<T> &out) const { const Tensor<T> &in = boost::get<Tensor<T>>(_in); - tensor_operations::pooling_layer(in, out, _pool_info, _fixed_point_position); + tensor_operations::pooling_layer(in, out, _pool_info); } private: const TensorVariant &_in; PoolingLayerInfo _pool_info; - int _fixed_point_position; }; // ROI Pooling layer |