From 7ffa1ff137b573e775892836821976e190f28687 Mon Sep 17 00:00:00 2001 From: Eric Kunze Date: Wed, 1 Jun 2022 17:26:48 -0700 Subject: Remove dilation from transpose_conv2d The specification does not have dilation for transpose_conv2d. Removed dilation from the attributes for transpose. Rename outpad to out_pad for transpose conv2d to align with specification Signed-off-by: Eric Kunze Change-Id: Ifb0fb7121a6bf33601420366ce86f2f0b56353de --- python/serializer/tosa_serializer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'python/serializer/tosa_serializer.py') diff --git a/python/serializer/tosa_serializer.py b/python/serializer/tosa_serializer.py index 929aebd..3582924 100644 --- a/python/serializer/tosa_serializer.py +++ b/python/serializer/tosa_serializer.py @@ -161,7 +161,7 @@ class TosaSerializerAttribute(TosaSerializerUnion): self.intvecs.append((a.AddStride, stride)) self.intvecs.append((a.AddDilation, dilation)) - def TransposeConvAttribute(self, outpad, stride, dilation, output_shape): + def TransposeConvAttribute(self, outpad, stride, output_shape): from tosa import TransposeConvAttribute as a, Attribute self.utype = Attribute.Attribute().TransposeConvAttribute @@ -169,7 +169,6 @@ class TosaSerializerAttribute(TosaSerializerUnion): self.intvecs.append((a.AddOutpad, outpad)) self.intvecs.append((a.AddStride, stride)) - self.intvecs.append((a.AddDilation, dilation)) self.intvecs.append((a.AddOutputShape, output_shape)) def PadAttribute(self, padding, pad_const_int, pad_const_fp): -- cgit v1.2.1