diff options
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 |