diff options
Diffstat (limited to 'include/attribute.def')
-rw-r--r-- | include/attribute.def | 29 |
1 files changed, 23 insertions, 6 deletions
diff --git a/include/attribute.def b/include/attribute.def index 8ac5d8d..ea91869 100644 --- a/include/attribute.def +++ b/include/attribute.def @@ -26,20 +26,26 @@ ...: variadic variables for more arguments, depending on NUM_ARGS_IN_ATTRIBUTES */ -DEF_ATTRIBUTE(Pool, 3, +DEF_ATTRIBUTE(Pool, 5, int32_t, V, pad, int32_t, V, kernel, - int32_t, V, stride) + int32_t, V, stride, + int32_t, S, input_zp, + int32_t, S, output_zp) -DEF_ATTRIBUTE(Conv, 3, +DEF_ATTRIBUTE(Conv, 5, int32_t, V, pad, int32_t, V, stride, - int32_t, V, dilation) + int32_t, V, dilation, + int32_t, S, input_zp, + int32_t, S, weight_zp) -DEF_ATTRIBUTE(TransposeConv, 3, +DEF_ATTRIBUTE(TransposeConv, 5, int32_t, V, out_pad, int32_t, V, stride, - int32_t, V, output_shape) + int32_t, V, output_shape, + int32_t, S, input_zp, + int32_t, S, weight_zp) DEF_ATTRIBUTE(Pad, 3, int32_t, V, padding, @@ -103,3 +109,14 @@ DEF_ATTRIBUTE(Transpose, 1, DEF_ATTRIBUTE(Table, 1, int16_t, V, table) +DEF_ATTRIBUTE(MatMul, 2, + int32_t, S, a_zp, + int32_t, S, b_zp) + +DEF_ATTRIBUTE(FullyConnected, 2, + int32_t, S, input_zp, + int32_t, S, weight_zp) + +DEF_ATTRIBUTE(Negate, 2, + int32_t, S, input1_zp, + int32_t, S, output_zp) |