diff options
Diffstat (limited to 'SUPPORTED_OPS.md')
| -rw-r--r-- | SUPPORTED_OPS.md | 28 |
1 files changed, 26 insertions, 2 deletions
diff --git a/SUPPORTED_OPS.md b/SUPPORTED_OPS.md index 20134cc..e48ebf5 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: `2.0.2.dev69+g83e3bb3.d20210212` +Vela version: `2.1.0` This file complies with [**Gitiles Markdown syntax**](https://github.com/google/gitiles/blob/master/Documentation/markdown.md) @@ -27,6 +27,7 @@ Please check the supported operator list for your chosen runtime for further inf | LOGISTIC | [Generic](#generic-constraints) | | MAXIMUM | [Generic](#generic-constraints), [Specific](#maximum-constraints) | | MAX_POOL_2D | [Generic](#generic-constraints), [Specific](#max_pool_2d-constraints) | +| MEAN | [Generic](#generic-constraints), [Specific](#mean-constraints) | | MINIMUM | [Generic](#generic-constraints), [Specific](#minimum-constraints) | | MUL | [Generic](#generic-constraints), [Specific](#mul-constraints) | | PACK | [Generic](#generic-constraints) | @@ -54,7 +55,7 @@ This is a list of constraints that all NPU operators must satisfy in order to be - Input(s) and Output tensors must not be dynamic - Input(s) and Output tensors must have a defined shape - Output tensors cannot be scalar -- Scalar Input tensors are only valid for op type: ADD, MAXIMUM, MINIMUM, MUL, SPLIT, SPLIT_V, SUB +- Scalar Input tensors are only valid for op type: ADD, MAXIMUM, MEAN, MINIMUM, MUL, SPLIT, SPLIT_V, SUB - Input(s) and Output tensors must not be greater than 4D - Tensors must be of type: int16, int32, int8, uint8 - Tensors which are int32 are only valid when op type is: ADD, MUL, SUB @@ -151,6 +152,7 @@ This is a list of constraints that the FULLY_CONNECTED operator must satisfy in - Optional Bias tensor must be of type: int32, int64 - Optional Bias tensor values must fit within 40-bits - The output tensor(s) must have 2D shape +- The IFM and OFM must have the same number of dimensions if keep_num_dims is set to true ## HARD_SWISH Constraints @@ -190,6 +192,28 @@ This is a list of constraints that the MAX_POOL_2D operator must satisfy in orde - Kernel filter height must be in the range [1, 256] - Product of kernel filter width and height must be in the range [1, 65536] +## MEAN Constraints + +This is a list of constraints that the MEAN operator must satisfy in order to be scheduled on the NPU. + +- IFM must be int8 or uint8 +- Every constraint in either one (or both) of the following sets of constraints must be fulfilled: + Set A: + IFM dimensions are 4, + Axis indices are 1 and 2, + keep_dims is set to True + Set B: + IFM zero point and OFM zero point are the same, + IFM scale and OFM scale are the same +- Input tensor must be at least 2D +- Axis indices must correspond to height and width axes +- Product of height and width can be at most 4096 +- Product of IFM height and width can be at most 256 when the following are true: + IFM dimensions are 4, + Axis indices are 1 and 2, + keep_dims is set to True and + IFM datatype is int8 + ## MINIMUM Constraints This is a list of constraints that the MINIMUM operator must satisfy in order to be scheduled on the NPU. |