diff options
Diffstat (limited to 'include/attribute.def')
-rw-r--r-- | include/attribute.def | 18 |
1 files changed, 6 insertions, 12 deletions
diff --git a/include/attribute.def b/include/attribute.def index 30b432d..52d5179 100644 --- a/include/attribute.def +++ b/include/attribute.def @@ -43,32 +43,26 @@ DEF_ATTRIBUTE(Conv, 7, bool, S, local_bound, DType, S, acc_type) -DEF_ATTRIBUTE(TransposeConv, 7, +DEF_ATTRIBUTE(TransposeConv, 6, int32_t, V, out_pad, int32_t, V, stride, - int32_t, V, output_shape, int32_t, S, input_zp, int32_t, S, weight_zp, bool, S, local_bound, DType, S, acc_type) -DEF_ATTRIBUTE(Pad, 2, - uint8_t, V, pad_const, - DType, S, type) +DEF_ATTRIBUTE(Pad, 1, + uint8_t, V, pad_const) DEF_ATTRIBUTE(Axis, 1, int32_t, S, axis) -DEF_ATTRIBUTE(Resize, 4, - int16_t, V, scale, - int16_t, V, offset, - int16_t, V, border, +DEF_ATTRIBUTE(Resize, 1, ResizeMode, S, mode) -DEF_ATTRIBUTE(Clamp, 3, +DEF_ATTRIBUTE(Clamp, 2, uint8_t, V, min_val, - uint8_t, V, max_val, - DType, S, type) + uint8_t, V, max_val) DEF_ATTRIBUTE(Rescale, 7, int32_t, S, input_zp, |