diff options
Diffstat (limited to 'src/armnn/StaticRangeVisitor.hpp')
-rw-r--r-- | src/armnn/StaticRangeVisitor.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/armnn/StaticRangeVisitor.hpp b/src/armnn/StaticRangeVisitor.hpp index af59dace9e..4276a178f3 100644 --- a/src/armnn/StaticRangeVisitor.hpp +++ b/src/armnn/StaticRangeVisitor.hpp @@ -27,6 +27,13 @@ public: /// Functions to set the Range on a per-layer-type basis void VisitAdditionLayer(const IConnectableLayer *layer, const char *name = nullptr) override; + void VisitBatchNormalizationLayer(const IConnectableLayer* layer, + const BatchNormalizationDescriptor& desc, + const ConstTensor& mean, + const ConstTensor& variance, + const ConstTensor& beta, + const ConstTensor& gamma, + const char* name = nullptr) override; /// Retreive the default range MinMaxRange DefaultRange() const { return std::make_pair(-15.0f, 15.0f); } |