aboutsummaryrefslogtreecommitdiff
path: root/verif
diff options
context:
space:
mode:
authorTai Ly <tai.ly@arm.com>2024-02-22 23:26:28 +0000
committerEric Kunze <eric.kunze@arm.com>2024-02-26 15:36:59 +0000
commitc5c2a7e4be64fef198c150063db9f241f21299d2 (patch)
treed46ee581b7992fb42c5484464e9c32a4a3adc646 /verif
parent97f1c0e602049ee43537ad6c8a08f476eb5c722b (diff)
downloadreference_model-c5c2a7e4be64fef198c150063db9f241f21299d2.tar.gz
[ref_model] Change resize attrs to inputs
This patch implements changes needed for resize op's scale/offset/border changing from attributes to inputs Signed-off-by: Tai Ly <tai.ly@arm.com> Change-Id: I20db0687fad40711f3ded644af51096292dd05b3
Diffstat (limited to 'verif')
-rw-r--r--verif/generator/tosa_arg_gen.py11
-rw-r--r--verif/generator/tosa_test_gen.py18
2 files changed, 25 insertions, 4 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py
index d0b9eb9..0851aca 100644
--- a/verif/generator/tosa_arg_gen.py
+++ b/verif/generator/tosa_arg_gen.py
@@ -1402,6 +1402,17 @@ class TosaTensorValuesGen:
# Needed for compliance
argsDict["max_abs_value"] = data_range[1]
+ scale_values = argsDict["scale"]
+ offset_values = argsDict["offset"]
+ border_values = argsDict["border"]
+ dtypeList[1] = DType.SHAPE
+ dtypeList[2] = DType.SHAPE
+ dtypeList[3] = DType.SHAPE
+ shapeList[1] = [len(scale_values)]
+ shapeList[2] = [len(offset_values)]
+ shapeList[3] = [len(border_values)]
+ argsDict["fixed_data"] = [None, scale_values, offset_values, border_values]
+
return TosaTensorValuesGen.tvgLazyGenDefault(
testGen, opName, dtypeList, shapeList, argsDict, error_name
)
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py
index 8440853..a03a763 100644
--- a/verif/generator/tosa_test_gen.py
+++ b/verif/generator/tosa_test_gen.py
@@ -1986,8 +1986,12 @@ class TosaTestGen:
error_name=None,
qinfo=None,
):
- assert len(inputs) == 1
+ assert len(inputs) == 4
input = inputs[0]
+ scale_input = inputs[1]
+ offset_input = inputs[2]
+ border_input = inputs[3]
+
mode = args_dict["mode"]
scale = args_dict["scale"]
offset = args_dict["offset"]
@@ -2008,7 +2012,12 @@ class TosaTestGen:
)
# Invalidate Input/Output list for error if checks.
- input_list = [input.name]
+ input_list = [
+ input.name,
+ scale_input.name,
+ offset_input.name,
+ border_input.name,
+ ]
output_list = [result_tensor.name]
pCount, cCount = op["operands"]
num_operands = pCount + cCount
@@ -2037,7 +2046,8 @@ class TosaTestGen:
return None
attr = ts.TosaSerializerAttribute()
- attr.ResizeAttribute(scale, offset, border, mode)
+ # write empty scale/offset/border into ResizeAttribute
+ attr.ResizeAttribute([], [], [], mode)
self.ser.addOperator(op["op"], input_list, output_list, attr)
compliance = self.tensorComplianceMetaData(
@@ -4688,7 +4698,7 @@ class TosaTestGen:
# Image operations
"resize": {
"op": Op.RESIZE,
- "operands": (1, 0),
+ "operands": (4, 0),
"rank": (4, 4),
"build_fcn": (
build_resize,