aboutsummaryrefslogtreecommitdiff
path: root/SUPPORTED_OPS.md
diff options
context:
space:
mode:
authorJohan Alfven <johan.alfven@arm.com>2023-03-31 10:19:23 +0200
committerJohan Alfven <johan.alfven@arm.com>2023-04-12 12:23:22 +0200
commitc1ad80b3a581dd39b39a112d6c2026f6560207a4 (patch)
treeb26ce4eb5399ed5e96aa96cc2ec5521037452dd3 /SUPPORTED_OPS.md
parent347c57bb88c1286bcd1c2775e7c67296410e2e6d (diff)
downloadethos-u-vela-c1ad80b3a581dd39b39a112d6c2026f6560207a4.tar.gz
MLBEDSW-7437: Add 64-bit output support for ArgMax
- Added 64-bit support for ArgMax - Updated constraints for ArgMax and regenerated SUPPORTED_OPS.md Change-Id: I4ef7d2e6fccab0088b87757f6afe40a006c77bbd Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'SUPPORTED_OPS.md')
-rw-r--r--SUPPORTED_OPS.md5
1 files changed, 3 insertions, 2 deletions
diff --git a/SUPPORTED_OPS.md b/SUPPORTED_OPS.md
index ba5b7919..08c63e7c 100644
--- a/SUPPORTED_OPS.md
+++ b/SUPPORTED_OPS.md
@@ -1,7 +1,7 @@
# Supported Ops
This file was automatically generated by Vela using the `--supported-ops-report` parameter.
-Vela version: `3.7.1.dev10+g521c494`
+Vela version: `3.7.1.dev15+g2b5f66e`
This file complies with
[**Gitiles Markdown syntax**](https://github.com/google/gitiles/blob/master/Documentation/markdown.md)
@@ -71,7 +71,7 @@ This is a list of constraints most NPU operators must satisfy in order to be sch
- Input(s), Output and Weight tensors with quantization scales must be finite
- Input and Output tensors must have quantization scales that fit within float32 precision
- Constant tensors should not have NoneType-values
-- Tensors must be of type: int16, int32, int8, uint8
+- Tensors must be of type: int16, int32, int8, uint8 - [ARG_MAX]
- Tensors which are int32 are only valid when op type is: ADD, ARG_MAX, MUL, SHAPE, SUB
- Tensor dimensions must be in the range [1, 65535]
- Per-axis quantization is only supported for the following op types: CONV_2D, DEPTHWISE_CONV_2D, TRANSPOSE_CONV
@@ -101,6 +101,7 @@ This is a list of constraints that the ADD operator must satisfy in order to be
This is a list of constraints that the ARG_MAX operator must satisfy in order to be scheduled on the NPU.
- IFM must be int8 or uint8
+- OFM must be int32 or int64
- Operation must be performed along the depth axis
- IFM depth must be no greater than 127