diff options
author | TatWai Chong <tatwai.chong@arm.com> | 2022-06-08 00:48:04 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2022-06-13 21:56:27 -0700 |
commit | 24594f55ee3bf0e95c764e51b94c3ec7f9cfa54a (patch) | |
tree | 2c4b4a1062b3ab2d204b306b2bd4017d9803e122 /verif/generator/tosa_test_gen.py | |
parent | 61f6622945d8ef339c99c4b437f985c62aa81bcf (diff) | |
download | reference_model-24594f55ee3bf0e95c764e51b94c3ec7f9cfa54a.tar.gz |
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 <tatwai.chong@arm.com>
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 9 |
1 files changed, 4 insertions, 5 deletions
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 |