surmeh01 | 49b9e10 | 2018-05-17 14:11:25 +0100 | [diff] [blame] | 1 | ------ ArmNN for Android NNAPI supported operations ------ |
telsoa01 | 5307bc1 | 2018-03-09 13:51:08 +0000 | [diff] [blame] | 2 | |
Matthew Bentham | a3e23ca | 2019-05-13 12:49:59 +0100 | [diff] [blame] | 3 | This release of ArmNN for Android supports use as a driver for the Android Neural Networks API. It implements the |
Ryan OShea | 7cc2bc5 | 2020-08-20 19:58:07 +0100 | [diff] [blame] | 4 | android.hardware.neuralnetworks@1.0, android.hardware.neuralnetworks@1.1, android.hardware.neuralnetworks@1.2 and |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 5 | android.hardware.neuralnetworks@1.3 |
Aron Virginas-Tar | b23732b | 2019-06-17 15:48:45 +0100 | [diff] [blame] | 6 | HAL interfaces. |
telsoa01 | 5307bc1 | 2018-03-09 13:51:08 +0000 | [diff] [blame] | 7 | |
| 8 | For more information on the Android Neural Networks API, see https://developer.android.com/ndk/guides/neuralnetworks/index.html |
| 9 | |
| 10 | For integration and usage documentation, please see README.md. |
| 11 | |
| 12 | --- Support for Android Neural Networks HAL operations --- |
| 13 | |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 14 | The following AndroidNN HAL 1.0, 1.1, 1.2 and 1.3 operations are currently supported: |
telsoa01 | 5307bc1 | 2018-03-09 13:51:08 +0000 | [diff] [blame] | 15 | |
| 16 | AndroidNN operator Tensor type supported |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 17 | ABS (FLOAT32, FLOAT16, INT32) |
| 18 | ADD (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 19 | ARGMAX (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 20 | ARGMIN (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 21 | AVERAGE_POOL_2D (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 22 | BATCH_TO_SPACE_ND (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Sadik Armagan | 92b5fd1 | 2021-04-26 09:52:06 +0100 | [diff] [blame] | 23 | CAST (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM) |
Teresa Charlin | 7f5b51e | 2021-09-01 14:19:38 +0100 | [diff] [blame] | 24 | CHANNEL_SHUFFLE (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Kevin May | 8bb7c0e | 2024-02-08 10:11:36 +0000 | [diff] [blame] | 25 | CONCATENATION (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 26 | CONV_2D (FLOAT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 27 | DEPTH_TO_SPACE (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 28 | DEPTHWISE_CONV_2D (FLOAT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 29 | DEQUANTIZE (FLOAT32 (output only), QUANT8_ASYMM and QUANT8_ASYMM_SIGNED (input only)) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 30 | DIV (FLOAT32, FLOAT16, INT32) |
| 31 | ELU (FLOAT32, FLOAT16, QUANT8_ASYMM) |
| 32 | EQUAL (BOOL8, FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Teresa Charlin | cb31b3b | 2020-09-30 18:28:18 +0100 | [diff] [blame] | 33 | EXP (FLOAT32, FLOAT16) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 34 | EXPAND_DIMS (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Sadik Armagan | 2e32961 | 2020-06-24 10:57:23 +0100 | [diff] [blame] | 35 | FILL (FLOAT32, FLOAT16, INT32) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 36 | FLOOR (FLOAT32, FLOAT16) |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 37 | FULLY_CONNECTED (FLOAT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Kevin May | 8bb7c0e | 2024-02-08 10:11:36 +0000 | [diff] [blame] | 38 | GATHER (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 39 | GREATER (BOOL8, FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 40 | GREATER_EQUAL (BOOL8, FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 41 | GROUPED_CONV_2D (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 42 | HARD_SWISH (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 43 | INSTANCE_NORMALIZATION (FLOAT32, FLOAT16) |
telsoa01 | 5307bc1 | 2018-03-09 13:51:08 +0000 | [diff] [blame] | 44 | L2_NORMALIZATION (FLOAT32) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 45 | L2_POOL_2D (FLOAT32, FLOAT16) |
| 46 | LESS (BOOL8, FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 47 | LESS_EQUAL (BOOL8, FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
telsoa01 | 5307bc1 | 2018-03-09 13:51:08 +0000 | [diff] [blame] | 48 | LOCAL_RESPONSE_NORMALIZATION (FLOAT32) |
Teresa Charlin | 2d71002 | 2021-05-31 19:35:58 +0100 | [diff] [blame] | 49 | LOG (FLOAT32, FLOAT16) |
Narumol Prangnawarat | 0629eb8 | 2020-11-12 18:27:37 +0000 | [diff] [blame] | 50 | LOGICAL_AND (BOOL8) |
| 51 | LOGICAL_NOT (BOOL8) |
| 52 | LOGICAL_OR (BOOL8) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 53 | LOGISTIC (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 54 | LOG_SOFTMAX (FLOAT32, FLOAT16) |
Ferran Balaguer | d04c043 | 2018-11-15 14:48:05 +0000 | [diff] [blame] | 55 | LSTM (FLOAT32) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 56 | MAXIMUM (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 57 | MAX_POOL_2D (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 58 | MEAN (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 59 | MINIMUM (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 60 | MUL (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 61 | NEG (FLOAT32, FLOAT16) |
| 62 | NOT_EQUAL (BOOL8, FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | 0c65ddb | 2022-02-10 10:01:33 +0000 | [diff] [blame] | 63 | PAD (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Matthew Sloyan | 63dda91 | 2021-08-11 15:15:41 +0100 | [diff] [blame] | 64 | PAD_V2 (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Kevin May | 8bb7c0e | 2024-02-08 10:11:36 +0000 | [diff] [blame] | 65 | POW (FLOAT32, FLOAT16) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 66 | PRELU (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 67 | QUANTIZE (FLOAT32 (input only), QUANT8_ASYMM and QUANT8_ASYMM_SIGNED (output only)) |
Sadik Armagan | 1ba9989 | 2019-08-21 15:03:44 +0100 | [diff] [blame] | 68 | QUANTIZED_16BIT_LSTM (QUANT8_ASYMM) |
Sadik Armagan | 813f230 | 2020-05-19 14:10:30 +0100 | [diff] [blame] | 69 | QUANTIZED_LSTM (QUANT8_ASYMM) |
Kevin May | 8bb7c0e | 2024-02-08 10:11:36 +0000 | [diff] [blame] | 70 | RANK (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 71 | REDUCE_MAX (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 72 | REDUCE_MIN (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Teresa Charlin | 1099960 | 2021-11-02 16:05:12 +0000 | [diff] [blame] | 73 | REDUCE_PROD (FLOAT32, FLOAT16) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 74 | REDUCE_SUM (FLOAT32, FLOAT16) |
| 75 | RELU (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 76 | RELU1 (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 77 | RELU6 (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 78 | RESHAPE (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | 0c65ddb | 2022-02-10 10:01:33 +0000 | [diff] [blame] | 79 | RESIZE_BILINEAR (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 80 | RESIZE_NEAREST_NEIGHBOR (FLOAT32, FLOAT16, QUANT8_ASYMM) |
| 81 | RSQRT (FLOAT32, FLOAT16) |
Teresa Charlin | 2d71002 | 2021-05-31 19:35:58 +0100 | [diff] [blame] | 82 | SIN (FLOAT32, FLOAT16) |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 83 | SOFTMAX (FLOAT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 84 | SPACE_TO_BATCH_ND (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 85 | SPACE_TO_DEPTH (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Kevin May | 8bb7c0e | 2024-02-08 10:11:36 +0000 | [diff] [blame] | 86 | SPLIT (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 87 | SQRT (FLOAT32, FLOAT16) |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 88 | SQUEEZE (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 89 | STRIDED_SLICE (FLOAT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 90 | SUB (FLOAT32, FLOAT16, INT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 91 | TANH (FLOAT32, FLOAT16, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Nikhil Raj | d735550 | 2023-07-31 16:51:00 +0100 | [diff] [blame] | 92 | TILE (All data types) |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 93 | TRANSPOSE (FLOAT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
| 94 | TRANSPOSE_CONV_2D (FLOAT32, QUANT8_ASYMM, QUANT8_ASYMM_SIGNED) |
Cathal Corbett | 0fa5e6d | 2022-01-21 16:55:13 +0000 | [diff] [blame] | 95 | UNIDIRECTIONAL_SEQUENCE_LSTM (FLOAT32, FLOAT16) |
telsoa01 | 5307bc1 | 2018-03-09 13:51:08 +0000 | [diff] [blame] | 96 | |
Mike Kelly | 56df76c | 2019-06-14 15:51:39 +0100 | [diff] [blame] | 97 | Where operations are not supported by the ArmNN Android NN Driver, the driver indicates this to the framework |
| 98 | appropriately and the framework implements those operations using a CPU implementation. |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 99 | |
Aron Virginas-Tar | 9acf579 | 2019-11-18 16:21:23 +0000 | [diff] [blame] | 100 | NOTE: By convention, only those tensor types have been listed above, which are fully supported across all |
Teresa Charlin | e75d09e | 2020-05-22 13:41:04 +0100 | [diff] [blame] | 101 | ArmNN backends. |
Nikhil Raj | a593af9 | 2021-05-10 16:27:26 +0100 | [diff] [blame] | 102 | - FLOAT16 input tensors are partially supported on most HAL 1.2 and 1.3 operators on the GpuAcc and |
| 103 | CpuRef backends, however not on CpuAcc. |