diff options
| author | Matthew Haddon <matthew.haddon@arm.com> | 2021-09-21 16:13:37 +0100 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2021-09-23 17:43:09 +0000 |
| commit | 6132cfe3f76a66ff44d373ee79ad6ea289c005ed (patch) | |
| tree | 4851ca7307a2a7be9b42861da1f01a9dc47e76d6 | |
| parent | 7e7fccd32c6e17ee5aef19746eff85d791a25f59 (diff) | |
| download | reference_model-6132cfe3f76a66ff44d373ee79ad6ea289c005ed.tar.gz | |
Add stride larger than dimension error_if check for resize
Change-Id: Ib70f6bdbfacfe125283821f1e3858542b05c60ad
Signed-off-by: Matthew Haddon <matthew.haddon@arm.com>
| -rw-r--r-- | reference_model/src/ops/image.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/reference_model/src/ops/image.cc b/reference_model/src/ops/image.cc index 72647cc..6dec1bc 100644 --- a/reference_model/src/ops/image.cc +++ b/reference_model/src/ops/image.cc @@ -179,6 +179,7 @@ int OpResize<DType_FLOAT, DType_FLOAT>::eval() "OpResize: exceeds maximum dimension"); ERROR_IF(shift != 0, "OpResize: float mode must have 0 shift"); ERROR_IF(stride_fp[0] <= 0.0f || stride_fp[1] <= 0.0f, "OpResize: invalid attribute stride"); + ERROR_IF(stride_fp[0] > in_height || stride_fp[1] > in_width, "OpResize: stride larger than dimension"); ERROR_IF(in_batch != out_batch, "OpResize: output tensor batch mismatch"); ERROR_IF(in_channels != out_channels, "OpResize: output tensor channel mismatch"); |