diff options
author | Sanghoon Lee <sanghoon.lee@arm.com> | 2017-09-15 14:10:48 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | 9688378ce14f0c2663a27b2c879ed1928247a08e (patch) | |
tree | cf7241f58b054eb3acf6a8e5331cb8c7b74b8e62 /tests/validation_old/TensorVisitors.h | |
parent | 6a3e976d71ecca2e6fdb604618fd94969eff9861 (diff) | |
download | ComputeLibrary-9688378ce14f0c2663a27b2c879ed1928247a08e.tar.gz |
COMPMID-494: Port BatchNormalizationLayer to new validation
Change-Id: Ief5334dd1cf571d977acf4ce9e5f580c5c9ab433
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/88158
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'tests/validation_old/TensorVisitors.h')
-rw-r--r-- | tests/validation_old/TensorVisitors.h | 27 |
1 files changed, 0 insertions, 27 deletions
diff --git a/tests/validation_old/TensorVisitors.h b/tests/validation_old/TensorVisitors.h index dafbfe0235..8af035be08 100644 --- a/tests/validation_old/TensorVisitors.h +++ b/tests/validation_old/TensorVisitors.h @@ -128,33 +128,6 @@ private: RoundingPolicy _rounding_policy; }; -// Batch Normalization Layer visitor -struct batch_normalization_layer_visitor : public boost::static_visitor<> -{ -public: - explicit batch_normalization_layer_visitor(const TensorVariant &in, const TensorVariant &mean, const TensorVariant &var, const TensorVariant &beta, const TensorVariant &gamma, float epsilon, - int fixed_point_position = 0) - : _in(in), _mean(mean), _var(var), _beta(beta), _gamma(gamma), _epsilon(epsilon), _fixed_point_position(fixed_point_position) - { - } - - template <typename T> - void operator()(Tensor<T> &out) const - { - const Tensor<T> &in = boost::get<Tensor<T>>(_in); - const Tensor<T> &mean = boost::get<Tensor<T>>(_mean); - const Tensor<T> &var = boost::get<Tensor<T>>(_var); - const Tensor<T> &beta = boost::get<Tensor<T>>(_beta); - const Tensor<T> &gamma = boost::get<Tensor<T>>(_gamma); - tensor_operations::batch_normalization_layer(in, out, mean, var, beta, gamma, _epsilon, _fixed_point_position); - } - -private: - const TensorVariant &_in, &_mean, &_var, &_beta, &_gamma; - float _epsilon; - int _fixed_point_position; -}; - // ROI Pooling layer struct roi_pooling_layer_visitor : public boost::static_visitor<> { |