diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-06-28 07:41:58 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-07-08 10:57:25 +0200 |
commit | 8f1f9aaa58175b17cd2e505bfcdb0e40c955ea72 (patch) | |
tree | 0174f8ef15007f5e220cfc4d283046451282102e /ethosu/vela/tosa/Op.py | |
parent | 6f4955aa7097b123bbf31aae4654547bb3e3c68c (diff) | |
download | ethos-u-vela-8f1f9aaa58175b17cd2e505bfcdb0e40c955ea72.tar.gz |
MLBEDSW-4838 Added basic TOSA support.
Added basic TOSA support, enabling Vela to
read and compile a .tosa file corresponding to
CONV2D + Rescale + Clamp, and writing it to an
optimized .tflite file.
The optimized .tflite file, will in this case, hold
a commandstream where the Rescale and Clamp has been
fused into the CONV2D.
The optimized tflite file is not output from Vela.
-Added support to read .tosa file into Vela
internal structure.
- Added tosa_reader.py, tosa_mapper.py and
helper files stored under tosa/
- Support for this limited to ~10 ops
-Added reader_util.py for functions common
for TOSA and TFLite
-Added tosa_graph_optimiser.py
-Added support to fuse Rescale into convolution
-Modified handling for padding
-Added support to fuse Clamp to previous op
-Added graph_optimiser_util.py
-Moved functions common for TOSA/TFLite graph
optimization to this file.
-Renamed graph_optimiser.py to tflite_graph_optmiser.py
-Added separate tosa_supported_operators.py
-Added supported_operator_util.py
-For functions in common for TOSA/TFLite
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Ic3c540504ec8c5eb4771397fdc6882050ecf33ab
Diffstat (limited to 'ethosu/vela/tosa/Op.py')
-rw-r--r-- | ethosu/vela/tosa/Op.py | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/ethosu/vela/tosa/Op.py b/ethosu/vela/tosa/Op.py new file mode 100644 index 00000000..c71ac44a --- /dev/null +++ b/ethosu/vela/tosa/Op.py @@ -0,0 +1,75 @@ +# automatically generated by the FlatBuffers compiler, do not modify + +# namespace: tosa + +class Op(object): + UNKNOWN = 0 + ARGMAX = 1 + AVG_POOL2D = 2 + CONV2D = 3 + CONV3D = 4 + DEPTHWISE_CONV2D = 5 + FULLY_CONNECTED = 6 + MATMUL = 7 + MAX_POOL2D = 8 + TRANSPOSE_CONV2D = 9 + CLAMP = 10 + RELUN = 11 + SIGMOID = 12 + TANH = 13 + ADD = 14 + ARITHMETIC_RIGHT_SHIFT = 15 + BITWISE_AND = 16 + BITWISE_OR = 17 + BITWISE_XOR = 18 + DIV = 19 + LOGICAL_AND = 20 + LOGICAL_LEFT_SHIFT = 21 + LOGICAL_RIGHT_SHIFT = 22 + LOGICAL_OR = 23 + LOGICAL_XOR = 24 + MAXIMUM = 25 + MINIMUM = 26 + MUL = 27 + POW = 28 + SUB = 29 + TABLE = 30 + ABS = 31 + BITWISE_NOT = 32 + CEIL = 33 + CLZ = 34 + EXP = 35 + FLOOR = 36 + LOG = 37 + LOGICAL_NOT = 38 + NEGATE = 39 + RECIPROCAL = 40 + RSQRT = 41 + SELECT = 42 + EQUAL = 43 + GREATER = 44 + GREATER_EQUAL = 45 + REDUCE_ANY = 46 + REDUCE_ALL = 47 + REDUCE_MAX = 48 + REDUCE_MIN = 49 + REDUCE_PRODUCT = 50 + REDUCE_SUM = 51 + CONCAT = 52 + PAD = 53 + RESHAPE = 54 + REVERSE = 55 + SLICE = 56 + TILE = 57 + TRANSPOSE = 58 + GATHER = 59 + SCATTER = 60 + RESIZE = 61 + CAST = 62 + RESCALE = 63 + CONST = 64 + IDENTITY = 65 + CUSTOM = 66 + COND_IF = 67 + WHILE_LOOP = 68 + |