From e095da79cc3575c8cbf818acd8a856ba136b0927 Mon Sep 17 00:00:00 2001 From: Tai Ly Date: Thu, 25 Jan 2024 22:00:18 +0000 Subject: [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 Change-Id: I437c86d9a012903458a648667f6693db67b97d76 --- verif/generator/tosa_arg_gen.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'verif/generator/tosa_arg_gen.py') 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 @@ -1045,6 +1045,19 @@ class TosaTensorValuesGen: testGen, op, dtypeList, shapeList, argsDict, error_name ) + @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 -- cgit v1.2.1