diff options
Diffstat (limited to 'include/attribute.def')
| -rw-r--r-- | include/attribute.def | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/include/attribute.def b/include/attribute.def index 723543e..30b432d 100644 --- a/include/attribute.def +++ b/include/attribute.def @@ -52,8 +52,9 @@ DEF_ATTRIBUTE(TransposeConv, 7, bool, S, local_bound, DType, S, acc_type) -DEF_ATTRIBUTE(Pad, 1, - uint8_t, V, pad_const) +DEF_ATTRIBUTE(Pad, 2, + uint8_t, V, pad_const, + DType, S, type) DEF_ATTRIBUTE(Axis, 1, int32_t, S, axis) @@ -64,9 +65,10 @@ DEF_ATTRIBUTE(Resize, 4, int16_t, V, border, ResizeMode, S, mode) -DEF_ATTRIBUTE(Clamp, 2, +DEF_ATTRIBUTE(Clamp, 3, uint8_t, V, min_val, - uint8_t, V, max_val) + uint8_t, V, max_val, + DType, S, type) DEF_ATTRIBUTE(Rescale, 7, int32_t, S, input_zp, |