diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2020-07-03 10:12:03 +0100 |
|---|---|---|
| committer | Finn Williams <Finn.Williams@arm.com> | 2020-07-10 19:27:07 +0100 |
| commit | 87d0bda9b49d9df4455f1887027e5ead2527c27e (patch) | |
| tree | c58787cce03027d3e1969a169f162f59a8b06f37 /src/armnn/Layer.hpp | |
| parent | c9f74d775da0039fd899f9ee6ec02b98ad575250 (diff) | |
| download | armnn-87d0bda9b49d9df4455f1887027e5ead2527c27e.tar.gz | |
IVGCVSW-4929 Implement ShapeInferenceMethod in all Layers
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I2c2d99f97cf89814140b057a9f93f41b364197f5
Diffstat (limited to 'src/armnn/Layer.hpp')
| -rw-r--r-- | src/armnn/Layer.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp index 303de052fc..f1954b9d07 100644 --- a/src/armnn/Layer.hpp +++ b/src/armnn/Layer.hpp @@ -347,6 +347,14 @@ protected: CollectWorkloadOutputs(dataCollector); } + void ValidateAndCopyShape(const TensorShape& outputShape, + const TensorShape& inferredShape, + const ShapeInferenceMethod shapeInferenceMethod, + const std::string& layerName, + const unsigned int outputSlotIndex = 0); + + void VerifyShapeInferenceType(const TensorShape& outputShape, ShapeInferenceMethod shapeInferenceMethod); + /// Helper function to reduce duplication in *Layer::CreateWorkload. template <typename QueueDescriptor> WorkloadInfo PrepInfoAndDesc(QueueDescriptor& descriptor) const |