aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_arg_gen.py
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/generator/tosa_arg_gen.py
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/generator/tosa_arg_gen.py')
-rw-r--r--verif/generator/tosa_arg_gen.py11
1 files changed, 11 insertions, 0 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
)