aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/ConstTensorLayerVisitor.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/test/ConstTensorLayerVisitor.hpp')
-rw-r--r--src/armnn/test/ConstTensorLayerVisitor.hpp280
1 files changed, 280 insertions, 0 deletions
diff --git a/src/armnn/test/ConstTensorLayerVisitor.hpp b/src/armnn/test/ConstTensorLayerVisitor.hpp
new file mode 100644
index 0000000000..3b0f723542
--- /dev/null
+++ b/src/armnn/test/ConstTensorLayerVisitor.hpp
@@ -0,0 +1,280 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#pragma once
+
+#include "TestLayerVisitor.hpp"
+#include <armnn/Descriptors.hpp>
+#include <armnn/LstmParams.hpp>
+
+namespace armnn
+{
+
+class TestConvolution2dLayerVisitor : public TestLayerVisitor
+{
+public:
+ explicit TestConvolution2dLayerVisitor(const Convolution2dDescriptor& convolution2dDescriptor,
+ const ConstTensor& weights,
+ const char* name = nullptr) : TestLayerVisitor(name),
+ m_Descriptor(convolution2dDescriptor),
+ m_Weights(weights) {};
+
+ virtual ~TestConvolution2dLayerVisitor() {};
+
+ void VisitConvolution2dLayer(const IConnectableLayer* layer,
+ const Convolution2dDescriptor& convolution2dDescriptor,
+ const ConstTensor& weights,
+ const char* name = nullptr) override
+ {
+ CheckLayerPointer(layer);
+ CheckLayerName(name);
+ CheckDescriptor(convolution2dDescriptor);
+ CheckConstTensors(m_Weights, weights);
+ }
+
+protected:
+ void CheckDescriptor(const Convolution2dDescriptor& convolution2dDescriptor);
+
+private:
+ armnn::Convolution2dDescriptor m_Descriptor;
+ armnn::ConstTensor m_Weights;
+};
+
+class TestConvolution2dWithBiasLayerVisitor : public TestConvolution2dLayerVisitor
+{
+public:
+ explicit TestConvolution2dWithBiasLayerVisitor(const Convolution2dDescriptor& convolution2dDescriptor,
+ const ConstTensor& weights,
+ const ConstTensor& biases,
+ const char* name = nullptr) :
+ TestConvolution2dLayerVisitor(
+ convolution2dDescriptor, weights, name),
+ m_Biases(biases) {};
+
+ // needed to suppress crappy error message about base class function i.e. version
+ // without the biases argument being hidden
+ using TestConvolution2dLayerVisitor::VisitConvolution2dLayer;
+
+ void VisitConvolution2dLayer(const IConnectableLayer* layer,
+ const Convolution2dDescriptor& convolution2dDescriptor,
+ const ConstTensor& weights,
+ const ConstTensor& biases,
+ const char* name = nullptr) override
+ {
+ TestConvolution2dLayerVisitor::VisitConvolution2dLayer(layer, convolution2dDescriptor, weights, name);
+ CheckConstTensors(m_Biases, biases);
+ }
+
+private:
+ armnn::ConstTensor m_Biases;
+};
+
+class TestDepthwiseConvolution2dLayerVisitor : public TestLayerVisitor
+{
+public:
+ explicit TestDepthwiseConvolution2dLayerVisitor(const DepthwiseConvolution2dDescriptor& descriptor,
+ const ConstTensor& weights,
+ const char* name = nullptr) : TestLayerVisitor(name),
+ m_Descriptor(descriptor),
+ m_Weights(weights) {};
+
+ virtual ~TestDepthwiseConvolution2dLayerVisitor() {};
+
+ void VisitDepthwiseConvolution2dLayer(const IConnectableLayer* layer,
+ const DepthwiseConvolution2dDescriptor& convolution2dDescriptor,
+ const ConstTensor& weights,
+ const char* name = nullptr) override
+ {
+ CheckLayerPointer(layer);
+ CheckLayerName(name);
+ CheckDescriptor(convolution2dDescriptor);
+ CheckConstTensors(m_Weights, weights);
+ }
+
+protected:
+ void CheckDescriptor(const DepthwiseConvolution2dDescriptor& convolution2dDescriptor);
+
+private:
+ armnn::DepthwiseConvolution2dDescriptor m_Descriptor;
+ armnn::ConstTensor m_Weights;
+};
+
+class TestDepthwiseConvolution2dWithBiasLayerVisitor : public TestDepthwiseConvolution2dLayerVisitor
+{
+public:
+ explicit TestDepthwiseConvolution2dWithBiasLayerVisitor(const DepthwiseConvolution2dDescriptor& descriptor,
+ const ConstTensor& weights,
+ const ConstTensor& biases,
+ const char* name = nullptr) :
+ TestDepthwiseConvolution2dLayerVisitor(descriptor, weights, name),
+ m_Biases(biases) {};
+
+ ~TestDepthwiseConvolution2dWithBiasLayerVisitor() {};
+
+ // needed to suppress crappy error message about base class function i.e. version
+ // without the biases argument being hidden
+ using TestDepthwiseConvolution2dLayerVisitor::VisitDepthwiseConvolution2dLayer;
+
+ void VisitDepthwiseConvolution2dLayer(const IConnectableLayer* layer,
+ const DepthwiseConvolution2dDescriptor& convolution2dDescriptor,
+ const ConstTensor& weights,
+ const ConstTensor& biases,
+ const char* name = nullptr) override
+ {
+ TestDepthwiseConvolution2dLayerVisitor::VisitDepthwiseConvolution2dLayer(
+ layer, convolution2dDescriptor, weights, name);
+ CheckConstTensors(m_Biases, biases);
+ }
+
+private:
+ armnn::ConstTensor m_Biases;
+};
+
+class TestFullyConnectedLayerVistor : public TestLayerVisitor
+{
+public:
+ explicit TestFullyConnectedLayerVistor(const FullyConnectedDescriptor& descriptor,
+ const ConstTensor& weights,
+ const char* name = nullptr) : TestLayerVisitor(name),
+ m_Descriptor(descriptor),
+ m_Weights(weights) {};
+
+ virtual ~TestFullyConnectedLayerVistor() {};
+
+ void VisitFullyConnectedLayer(const IConnectableLayer* layer,
+ const FullyConnectedDescriptor& fullyConnectedDescriptor,
+ const ConstTensor& weights,
+ const char* name = nullptr) override
+ {
+ CheckLayerPointer(layer);
+ CheckLayerName(name);
+ CheckDescriptor(fullyConnectedDescriptor);
+ CheckConstTensors(m_Weights, weights);
+ }
+
+protected:
+ void CheckDescriptor(const FullyConnectedDescriptor& descriptor);
+private:
+ FullyConnectedDescriptor m_Descriptor;
+ ConstTensor m_Weights;
+};
+
+class TestFullyConnectedLayerWithBiasesVisitor : public TestFullyConnectedLayerVistor
+{
+public:
+ explicit TestFullyConnectedLayerWithBiasesVisitor(const FullyConnectedDescriptor& descriptor,
+ const ConstTensor& weights,
+ const ConstTensor& biases,
+ const char* name = nullptr) :
+ TestFullyConnectedLayerVistor(descriptor, weights, name),
+ m_Biases(biases) {};
+
+ // needed to suppress crappy error message about base class function i.e. version
+ // without the biases argument being hidden
+ using TestFullyConnectedLayerVistor::VisitFullyConnectedLayer;
+
+ void VisitFullyConnectedLayer(const IConnectableLayer* layer,
+ const FullyConnectedDescriptor& fullyConnectedDescriptor,
+ const ConstTensor& weights,
+ const ConstTensor& biases,
+ const char* name = nullptr) override
+ {
+ TestFullyConnectedLayerVistor::VisitFullyConnectedLayer(layer, fullyConnectedDescriptor, weights, name);
+ CheckConstTensors(m_Biases, biases);
+ }
+
+private:
+ ConstTensor m_Biases;
+};
+
+class TestBatchNormalizationLayerVisitor : public TestLayerVisitor
+{
+public:
+ TestBatchNormalizationLayerVisitor(const BatchNormalizationDescriptor& descriptor,
+ const ConstTensor& mean,
+ const ConstTensor& variance,
+ const ConstTensor& beta,
+ const ConstTensor& gamma,
+ const char* name = nullptr) : TestLayerVisitor(name),
+ m_Descriptor(descriptor),
+ m_Mean(mean),
+ m_Variance(variance),
+ m_Beta(beta),
+ m_Gamma(gamma) {};
+ void VisitBatchNormalizationLayer(const IConnectableLayer* layer,
+ const BatchNormalizationDescriptor& descriptor,
+ const ConstTensor& mean,
+ const ConstTensor& variance,
+ const ConstTensor& beta,
+ const ConstTensor& gamma,
+ const char* name = nullptr) override
+ {
+ CheckLayerPointer(layer);
+ CheckLayerName(name);
+ CheckDescriptor(descriptor);
+ CheckConstTensors(m_Mean, mean);
+ CheckConstTensors(m_Variance, variance);
+ CheckConstTensors(m_Beta, beta);
+ CheckConstTensors(m_Gamma, gamma);
+ }
+
+protected:
+ void CheckDescriptor(const BatchNormalizationDescriptor& descriptor);
+private:
+ BatchNormalizationDescriptor m_Descriptor;
+ ConstTensor m_Mean;
+ ConstTensor m_Variance;
+ ConstTensor m_Beta;
+ ConstTensor m_Gamma;
+};
+
+class TestConstantLayerVisitor : public TestLayerVisitor
+{
+public:
+ explicit TestConstantLayerVisitor(const ConstTensor& input, const char* name = nullptr) :
+ TestLayerVisitor(name), m_Input(input) {};
+
+ void VisitConstantLayer(const IConnectableLayer* layer,
+ const ConstTensor& input,
+ const char* name = nullptr)
+ {
+ CheckLayerPointer(layer);
+ CheckLayerName(name);
+ CheckConstTensors(m_Input, input);
+ }
+
+private:
+ ConstTensor m_Input;
+};
+
+class TestLstmLayerVisitor : public TestLayerVisitor
+{
+public:
+ explicit TestLstmLayerVisitor(const LstmDescriptor& descriptor,
+ const LstmInputParams& params,
+ const char* name = nullptr) :
+ TestLayerVisitor(name), m_Descriptor(descriptor), m_InputParams(params) {};
+
+ void VisitLstmLayer(const IConnectableLayer* layer,
+ const LstmDescriptor& descriptor,
+ const LstmInputParams& params,
+ const char* name = nullptr)
+ {
+ CheckLayerPointer(layer);
+ CheckLayerName(name);
+ CheckDescriptor(descriptor);
+ CheckInputParameters(params);
+ }
+
+protected:
+ void CheckDescriptor(const LstmDescriptor& descriptor);
+ void CheckInputParameters(const LstmInputParams& inputParams);
+ void CheckConstTensorPtrs(const std::string& name, const ConstTensor* expected, const ConstTensor* actual);
+
+private:
+ LstmDescriptor m_Descriptor;
+ LstmInputParams m_InputParams;
+};
+
+} // namespace armnn \ No newline at end of file
generated by cgit v1.2.1 (git 2.23.0) at 2024-06-29 14:09:29 +0000