diff options
| author | Eric Kunze <eric.kunze@arm.com> | 2022-06-01 17:26:48 -0700 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2022-06-13 15:54:03 -0700 |
| commit | 7ffa1ff137b573e775892836821976e190f28687 (patch) | |
| tree | 7d39117a9ae9caf13f00b199de87ffe05576552b /python/serializer/tosa_serializer.py | |
| parent | 9a293aa4be98947c5d6fdb85798c6181e09b8b50 (diff) | |
| download | serialization_lib-7ffa1ff137b573e775892836821976e190f28687.tar.gz | |
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 <eric.kunze@arm.com>
Change-Id: Ifb0fb7121a6bf33601420366ce86f2f0b56353de
Diffstat (limited to 'python/serializer/tosa_serializer.py')
| -rw-r--r-- | python/serializer/tosa_serializer.py | 3 |
1 files changed, 1 insertions, 2 deletions
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): |