diff options
author | Tai Ly <tai.ly@arm.com> | 2024-02-22 23:26:28 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-02-26 15:36:59 +0000 |
commit | c5c2a7e4be64fef198c150063db9f241f21299d2 (patch) | |
tree | d46ee581b7992fb42c5484464e9c32a4a3adc646 /verif | |
parent | 97f1c0e602049ee43537ad6c8a08f476eb5c722b (diff) | |
download | reference_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.py | 11 | ||||
-rw-r--r-- | verif/generator/tosa_test_gen.py | 18 |
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, |