From 6132cfe3f76a66ff44d373ee79ad6ea289c005ed Mon Sep 17 00:00:00 2001 From: Matthew Haddon Date: Tue, 21 Sep 2021 16:13:37 +0100 Subject: Add stride larger than dimension error_if check for resize Change-Id: Ib70f6bdbfacfe125283821f1e3858542b05c60ad Signed-off-by: Matthew Haddon --- reference_model/src/ops/image.cc | 1 + 1 file changed, 1 insertion(+) 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::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"); -- cgit v1.2.1