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 /include/attribute.def | |
| 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 'include/attribute.def')
| -rw-r--r-- | include/attribute.def | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/include/attribute.def b/include/attribute.def index 38f6d29..8ac5d8d 100644 --- a/include/attribute.def +++ b/include/attribute.def @@ -36,10 +36,9 @@ DEF_ATTRIBUTE(Conv, 3, int32_t, V, stride, int32_t, V, dilation) -DEF_ATTRIBUTE(TransposeConv, 4, - int32_t, V, outpad, +DEF_ATTRIBUTE(TransposeConv, 3, + int32_t, V, out_pad, int32_t, V, stride, - int32_t, V, dilation, int32_t, V, output_shape) DEF_ATTRIBUTE(Pad, 3, |