diff options
| author | David Monahan <david.monahan@arm.com> | 2020-06-19 16:43:48 +0100 |
|---|---|---|
| committer | David Monahan <david.monahan@arm.com> | 2020-06-20 09:46:26 +0000 |
| commit | ab219756319667ae7abb0d767cf15e766ab343b1 (patch) | |
| tree | b68fce250273230d59a2d446f156283e576d6b0d /src/armnn/layers | |
| parent | 7e989834019e49f904df5c5ab0a299bde37c90c9 (diff) | |
| download | armnn-ab219756319667ae7abb0d767cf15e766ab343b1.tar.gz | |
IVGCVSW-4707 - Add AlignCorners and HalfPixelCenters to Resize
* Added AlignCorners and HalfPixelCenters Parameters to Resize
* Added Unit Tests
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I83420a9bcb7beec9073d201448f64eb53090e1f1
Diffstat (limited to 'src/armnn/layers')
| -rw-r--r-- | src/armnn/layers/ResizeLayer.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/layers/ResizeLayer.cpp b/src/armnn/layers/ResizeLayer.cpp index 9654e58b43..b16adeb860 100644 --- a/src/armnn/layers/ResizeLayer.cpp +++ b/src/armnn/layers/ResizeLayer.cpp @@ -50,6 +50,11 @@ std::vector<TensorShape> ResizeLayer::InferOutputShapes(const std::vector<Tensor TensorShape( { outBatch, outHeight, outWidth, outChannels } ) : TensorShape( { outBatch, outChannels, outHeight, outWidth }); + if (m_Param.m_HalfPixelCenters && m_Param.m_AlignCorners) + { + throw LayerValidationException("ResizeLayer: AlignCorners cannot be true when HalfPixelCenters is true"); + } + return std::vector<TensorShape>({ tensorShape }); } |