aosp12/external/armnn/delegate/TensorFlowLiteDelegateSuppo...

1.1 KiB

TensorFlow Lite operators that the Arm NN TensorFlow Lite Delegate supports

This reference guide provides a list of TensorFlow Lite operators the Arm NN SDK currently supports.

Fully supported

The Arm NN SDK TensorFlow Lite delegate currently supports the following operators:

  • ABS

  • ADD

  • AVERAGE_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE

  • CONCATENATION, Supported Fused Activation: RELU , RELU6 , TANH, NONE

  • CONV_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE

  • DEPTHWISE_CONV_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE

  • DEQUANTIZE

  • DIV

  • EQUAL

  • EXP

  • FULLY_CONNECTED

  • GREATER

  • GREATER_OR_EQUAL

  • LESS

  • LESS_OR_EQUAL

  • LOGISTIC

  • LOG_SOFTMAX

  • L2_POOL_2D

  • MAXIMUM

  • MAX_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE

  • MEAN

  • MINIMUM

  • MUL

  • NEG

  • NOT_EQUAL

  • QUANTIZE

  • RESHAPE

  • RESIZE_BILINEAR

  • RESIZE_NEAREST_NEIGHBOR

  • RELU

  • RELU6

  • RSQRT

  • SOFTMAX

  • SQRT

  • SUB

  • TANH

  • TRANSPOSE

  • TRANSPOSE_CONV

More machine learning operators will be supported in future releases.