aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/LayerVisitorBase.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/LayerVisitorBase.hpp')
-rw-r--r--include/armnn/LayerVisitorBase.hpp203
1 files changed, 102 insertions, 101 deletions
diff --git a/include/armnn/LayerVisitorBase.hpp b/include/armnn/LayerVisitorBase.hpp
index 62673ace07..657051f2fa 100644
--- a/include/armnn/LayerVisitorBase.hpp
+++ b/include/armnn/LayerVisitorBase.hpp
@@ -12,12 +12,12 @@ namespace armnn
struct VisitorThrowingPolicy
{
- static void Apply() { throw UnimplementedException(); }
+ static void Apply(const std::string& errorMessage = "") { throw UnimplementedException(errorMessage); }
};
struct VisitorNoThrowPolicy
{
- static void Apply() {}
+ static void Apply(const std::string& unusedMessage = "") {}
};
// Visitor base class with empty implementations.
@@ -29,162 +29,163 @@ protected:
virtual ~LayerVisitorBase() {}
public:
- void VisitInputLayer(const IConnectableLayer*,
- LayerBindingId,
- const char*) override { DefaultPolicy::Apply(); }
+
+ void VisitActivationLayer(const IConnectableLayer*,
+ const ActivationDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
+
+ void VisitAdditionLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
+
+ void VisitBatchNormalizationLayer(const IConnectableLayer*,
+ const BatchNormalizationDescriptor&,
+ const ConstTensor&,
+ const ConstTensor&,
+ const ConstTensor&,
+ const ConstTensor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
+
+ void VisitBatchToSpaceNdLayer(const IConnectableLayer*,
+ const BatchToSpaceNdDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitConcatLayer(const IConnectableLayer*,
const OriginsDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
+
+ void VisitConstantLayer(const IConnectableLayer*,
+ const ConstTensor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitConvolution2dLayer(const IConnectableLayer*,
const Convolution2dDescriptor&,
const ConstTensor&,
const Optional<ConstTensor>&,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitDepthwiseConvolution2dLayer(const IConnectableLayer*,
const DepthwiseConvolution2dDescriptor&,
const ConstTensor&,
const Optional<ConstTensor>&,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitDequantizeLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitDetectionPostProcessLayer(const IConnectableLayer*,
const DetectionPostProcessDescriptor&,
const ConstTensor&,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
+
+ void VisitDivisionLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
+
+ void VisitEqualLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
+
+ void VisitFloorLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitFullyConnectedLayer(const IConnectableLayer*,
const FullyConnectedDescriptor&,
const ConstTensor&,
const Optional<ConstTensor>&,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitPermuteLayer(const IConnectableLayer*,
- const PermuteDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitGatherLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitBatchToSpaceNdLayer(const IConnectableLayer*,
- const BatchToSpaceNdDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitGreaterLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitPooling2dLayer(const IConnectableLayer*,
- const Pooling2dDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitInputLayer(const IConnectableLayer*,
+ LayerBindingId,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitActivationLayer(const IConnectableLayer*,
- const ActivationDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitL2NormalizationLayer(const IConnectableLayer*,
+ const L2NormalizationDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitNormalizationLayer(const IConnectableLayer*,
- const NormalizationDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitLstmLayer(const IConnectableLayer*,
+ const LstmDescriptor&,
+ const LstmInputParams&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitSoftmaxLayer(const IConnectableLayer*,
- const SoftmaxDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitMaximumLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitSplitterLayer(const IConnectableLayer*,
- const ViewsDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitMeanLayer(const IConnectableLayer*,
+ const MeanDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitMergeLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitMergerLayer(const IConnectableLayer*,
const OriginsDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitAdditionLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitMinimumLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitMultiplicationLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitBatchNormalizationLayer(const IConnectableLayer*,
- const BatchNormalizationDescriptor&,
- const ConstTensor&,
- const ConstTensor&,
- const ConstTensor&,
- const ConstTensor&,
- const char*) override { DefaultPolicy::Apply(); }
-
- void VisitResizeBilinearLayer(const IConnectableLayer*,
- const ResizeBilinearDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
-
- void VisitL2NormalizationLayer(const IConnectableLayer*,
- const L2NormalizationDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
-
- void VisitConstantLayer(const IConnectableLayer*,
- const ConstTensor&,
- const char*) override { DefaultPolicy::Apply(); }
-
- void VisitReshapeLayer(const IConnectableLayer*,
- const ReshapeDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
-
- void VisitSpaceToBatchNdLayer(const IConnectableLayer*,
- const SpaceToBatchNdDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
-
- void VisitFloorLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitNormalizationLayer(const IConnectableLayer*,
+ const NormalizationDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitOutputLayer(const IConnectableLayer*,
LayerBindingId id,
- const char*) override { DefaultPolicy::Apply(); }
-
- void VisitLstmLayer(const IConnectableLayer*,
- const LstmDescriptor&,
- const LstmInputParams&,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitDivisionLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitPadLayer(const IConnectableLayer*,
+ const PadDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitSubtractionLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitPermuteLayer(const IConnectableLayer*,
+ const PermuteDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitMaximumLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitPooling2dLayer(const IConnectableLayer*,
+ const Pooling2dDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitMeanLayer(const IConnectableLayer*,
- const MeanDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitQuantizeLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitPadLayer(const IConnectableLayer*,
- const PadDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitReshapeLayer(const IConnectableLayer*,
+ const ReshapeDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitQuantizeLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitResizeBilinearLayer(const IConnectableLayer*,
+ const ResizeBilinearDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitStridedSliceLayer(const IConnectableLayer*,
- const StridedSliceDescriptor&,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitRsqrtLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitMinimumLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitSoftmaxLayer(const IConnectableLayer*,
+ const SoftmaxDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitGreaterLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitSpaceToBatchNdLayer(const IConnectableLayer*,
+ const SpaceToBatchNdDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitEqualLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitSplitterLayer(const IConnectableLayer*,
+ const ViewsDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitRsqrtLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitStridedSliceLayer(const IConnectableLayer*,
+ const StridedSliceDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
- void VisitGatherLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ void VisitSubtractionLayer(const IConnectableLayer*,
+ const char*) override { DefaultPolicy::Apply(__func__); }
void VisitSwitchLayer(const IConnectableLayer*,
- const char*) override { DefaultPolicy::Apply(); }
+ const char*) override { DefaultPolicy::Apply(__func__); }
};
} //namespace armnn