diff options
Diffstat (limited to 'include/attribute.def')
| -rw-r--r-- | include/attribute.def | 81 |
1 files changed, 40 insertions, 41 deletions
diff --git a/include/attribute.def b/include/attribute.def index ea91869..0e97629 100644 --- a/include/attribute.def +++ b/include/attribute.def @@ -1,5 +1,5 @@ -// Copyright (c) 2020-2021, ARM Limited. +// Copyright (c) 2020-2024, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -26,68 +26,55 @@ ...: variadic variables for more arguments, depending on NUM_ARGS_IN_ATTRIBUTES */ -DEF_ATTRIBUTE(Pool, 5, +DEF_ATTRIBUTE(Pool, 6, int32_t, V, pad, int32_t, V, kernel, int32_t, V, stride, int32_t, S, input_zp, - int32_t, S, output_zp) + int32_t, S, output_zp, + DType, S, acc_type) -DEF_ATTRIBUTE(Conv, 5, +DEF_ATTRIBUTE(Conv, 7, int32_t, V, pad, int32_t, V, stride, int32_t, V, dilation, int32_t, S, input_zp, - int32_t, S, weight_zp) + int32_t, S, weight_zp, + bool, S, local_bound, + DType, S, acc_type) -DEF_ATTRIBUTE(TransposeConv, 5, +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) + int32_t, S, weight_zp, + bool, S, local_bound, + DType, S, acc_type) -DEF_ATTRIBUTE(Pad, 3, - int32_t, V, padding, - int32_t, S, pad_const_int, - float, S, pad_const_fp) +DEF_ATTRIBUTE(Pad, 1, + uint8_t, V, pad_const) DEF_ATTRIBUTE(Axis, 1, int32_t, S, axis) -DEF_ATTRIBUTE(Reshape, 1, - int32_t, V, new_shape) - -DEF_ATTRIBUTE(Slice, 2, - int32_t, V, start, - int32_t, V, size) - -DEF_ATTRIBUTE(Tile, 1, - int32_t, V, multiples) - -DEF_ATTRIBUTE(Resize, 7, - int32_t, V, output_size, - int32_t, V, stride, - int32_t, V, offset, - int32_t, S, shift, - float, V, stride_fp, - float, V, offset_fp, +DEF_ATTRIBUTE(Resize, 4, + int16_t, V, scale, + int16_t, V, offset, + int16_t, V, border, ResizeMode, S, mode) -DEF_ATTRIBUTE(Clamp, 4, - int32_t, S, min_int, - int32_t, S, max_int, - float, S, min_fp, - float, S, max_fp) +DEF_ATTRIBUTE(Clamp, 2, + uint8_t, V, min_val, + uint8_t, V, max_val) DEF_ATTRIBUTE(Rescale, 7, int32_t, S, input_zp, int32_t, S, output_zp, - int32_t, V, multiplier, - int32_t, V, shift, bool, S, scale32, bool, S, double_round, - bool, S, per_channel) + bool, S, per_channel, + bool, S, input_unsigned, + bool, S, output_unsigned) DEF_ATTRIBUTE(Mul, 1, int32_t, S, shift) @@ -96,12 +83,12 @@ DEF_ATTRIBUTE(ArithmeticRightShift, 1, bool, S, round) DEF_ATTRIBUTE(CondIf, 2, - string, S, then_branch, - string, S, else_branch) + string, S, then_graph, + string, S, else_graph) DEF_ATTRIBUTE(WhileLoop, 2, - string, S, cond_branch, - string, S, body_branch) + string, S, cond_graph, + string, S, body_graph) DEF_ATTRIBUTE(Transpose, 1, int32_t, V, perms) @@ -120,3 +107,15 @@ DEF_ATTRIBUTE(FullyConnected, 2, DEF_ATTRIBUTE(Negate, 2, int32_t, S, input1_zp, int32_t, S, output_zp) + +DEF_ATTRIBUTE(Custom, 3, + string, S, operator_name, + string, S, domain_name, + uint8_t, V, implementation_attrs) + +DEF_ATTRIBUTE(FFT, 2, + bool, S, inverse, + bool, S, local_bound) + +DEF_ATTRIBUTE(RFFT, 1, + bool, S, local_bound) |