diff options
| author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-14 14:56:48 +0200 |
|---|---|---|
| committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-15 13:21:16 +0200 |
| commit | f436ada9caea87ec2dd686a92e41a15c1dcdeb1d (patch) | |
| tree | 5d3c136de06e0ee54833d2a379eb48cfd12ccd75 /ethosu/vela/operation.py | |
| parent | 0957e3ef4b94f17efb67429c88bab8ba650f78e8 (diff) | |
| download | ethos-u-vela-f436ada9caea87ec2dd686a92e41a15c1dcdeb1d.tar.gz | |
TOSA: Support for TABLE operator (int8)
Added support to map TABLE operator to LUT.
Limitations:
-Only supported for int8
-TABLE input must be constant
This also adds the support for TFLite legalisation of
Tanh/Sigmoid (int8/uint8).
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I1a95f61fb02fdd42c4a690494418cc0765c8b275
Diffstat (limited to 'ethosu/vela/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index e9d364ea..1558b943 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -281,6 +281,7 @@ class Op(Enum): SubgraphInput = OperatorInfo() # Only used in CPU subgraphs Sum = OperatorInfo() Svdf = OperatorInfo() + Table = OperatorInfo(indices=NNG_IFM_INDICES) Tanh = OperatorInfo(indices=NNG_IFM_INDICES) Tile = OperatorInfo() TopKV2 = OperatorInfo() |