aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_arg_gen.py
diff options
context:
space:
mode:
authorTai Ly <tai.ly@arm.com>2024-01-25 22:00:18 +0000
committerEric Kunze <eric.kunze@arm.com>2024-01-30 17:35:41 +0000
commite095da79cc3575c8cbf818acd8a856ba136b0927 (patch)
tree2f400f2d5b13d465256d65030e35675f9abb3c38 /verif/generator/tosa_arg_gen.py
parent455e8700fbd7c249dd6b5ea63869026b5e995669 (diff)
downloadreference_model-e095da79cc3575c8cbf818acd8a856ba136b0927.tar.gz
[ref model] Change PadOp's padding to Shape
Changed to use pad input of PadOp for testing. The pad input is now a tensor of tosa.shape type. moved padding error checking from checkTensorAttributes to eval modified pad's PadOutputShapeMismatch test generation to avoid generating output shapes with dimensions <= 0 Signed-off-by: Tai Ly <tai.ly@arm.com> Change-Id: I437c86d9a012903458a648667f6693db67b97d76
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
-rw-r--r--verif/generator/tosa_arg_gen.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py
index 386e243..0db9717 100644
--- a/verif/generator/tosa_arg_gen.py
+++ b/verif/generator/tosa_arg_gen.py
@@ -1046,6 +1046,19 @@ class TosaTensorValuesGen:
)
@staticmethod
+ def tvgPad(testGen, op, dtypeList, shapeList, argsDict, error_name=None):
+ # argsDict["pad"] is 2D array, need to flatten it to get list of values
+ pad_values = argsDict["pad"].flatten()
+ dtypeList[1] = DType.SHAPE
+ shapeList[1] = [len(pad_values)]
+ # Create a new list for the pre-generated data in argsDict["fixed_data"]
+ argsDict["fixed_data"] = [None, pad_values]
+
+ return TosaTensorValuesGen.tvgLazyGenDefault(
+ testGen, op, dtypeList, shapeList, argsDict, error_name
+ )
+
+ @staticmethod
def tvgTile(testGen, op, dtypeList, shapeList, argsDict, error_name=None):
dtypeList[1] = DType.SHAPE
shapeList[1] = [len(argsDict["multiples"])]