From 24594f55ee3bf0e95c764e51b94c3ec7f9cfa54a Mon Sep 17 00:00:00 2001 From: TatWai Chong Date: Wed, 8 Jun 2022 00:48:04 -0700 Subject: Update transpose_conv2d to align with TOSA spec Rename outpad to out_pad, and also fix the dilation in the generator. Change-Id: I4c1599871f0d0b41856e819d8c644a85ca6d8267 Signed-off-by: TatWai Chong --- verif/generator/tosa_test_gen.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'verif/generator/tosa_test_gen.py') diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index fc2e476..262a652 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -718,14 +718,13 @@ class TosaTestGen: filter, bias, stride, - outpad, - dilation, + out_pad, output_shape, validator_fcns=None, error_name=None, qinfo=None, ): - assert len(outpad) == 4 + assert len(out_pad) == 4 result_tens = OutputShaper.transposeConv2DOp( self.ser, self.rng, ifm, output_shape, error_name ) @@ -761,7 +760,7 @@ class TosaTestGen: input_list=input_list, num_operands=num_operands, output_list=output_list, - pad=outpad, + pad=out_pad, stride=stride, input_shape=ifm.shape, weight_shape=filter.shape, @@ -770,7 +769,7 @@ class TosaTestGen: return None attr = ts.TosaSerializerAttribute() - attr.TransposeConvAttribute(outpad, stride, dilation, output_shape) + attr.TransposeConvAttribute(out_pad, stride, output_shape) self.ser.addOperator(op["op"], input_list, output_list, attr, qinfo) return result_tens -- cgit v1.2.1