Tengine, developed by OPEN AI LAB, is a lite, high-performance, and modular inference engine for embedded device.
Tengine is composed of six modules: core/operator/serializer/executor/driver/wrapper.
- core provides the basic components and functionalities of the system.
- operator defines the schema of operators, such as convolution, relu, pooling, etc. al. Here is the current support operator list.
- serializer is to load the saved model. The serializer framework is extensible to support different format, including the customized one. Current version can support caffe and MXNet models. Tensorflow support will be the next.
- executor implements the code to run graph and operators. Current version provides a highly optimized implementation for multi A72 cores.
- driver is the adapter of real H/W and provides service to device executor by HAL API. It is possible for single driver to create multiple devices.
- wrapper provides the wrapper of APIs for different frameworks. Current version only supports caffe API wrapper. Tensorflow API support will be the next.
This version can load and run caffe/MXNet model of mobilenet and squeezenet directly. For more details, please goto install.
NOTE
: Old caffe model has to be upgraded using upgrade_net_proto_binary/upgrade_net_proto_binary from caffe's package.
The data is collected on 1.8G A72 and on chip RK3399, by repeating calling the forward interface to get the average time cost (ms) per run.
- Single A72 core (1xA72)
NN | Caffe(Openblas) | Tengine |
---|---|---|
squeezenet | 147 | 91 |
mobilenet | 306 | 122 |
- Two A72 cores (2xA72)
NN | Caffe(Openblas) | Tengine |
---|---|---|
squeezenet | 102 | 55 |
mobilenet | 232 | 77 |
For details to run benchmark, please visit benchmark page.
please refer to the install page.
It is easy to add new operator to Tengine. Here is the guide on new operator.
Tengine can be extended to support new serialization format, by building new serializer module.
How to build new serializer module
Introduce the driver/device model to support MT(Multi-Thread)
Support new NN: Inception-v4
Caffe Wrapper examples: squeezenet/mobilenet/mtcnn
MXNet model load examples: squeezenet/mobilenet
Support new operator: Eltwise, PReLU, Slice
Support new NN: mtcnn, resnet and lighten_cnn
Experimental caffe API wrapper: caffe based application just needs to recompile to use Tengine
Update documents, as well a few fixes.
Initial release of single A72 support