diff options
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, |