Skip to content
forked from liuguoyou/tengine

Tengine is a lite, high performance, modular inference engine for embedded device

License

Notifications You must be signed in to change notification settings

rogerou/tengine

 
 

Repository files navigation

Tengine Overview

GitHub license

Tengine, developed by Open AI lab, is a lite, high-performance, and modular inference engine for embedded device.

Tengine is composed of four modules: core/operator/serializer/executor.

  • core provides the basic components and functionalities of the system.

  • operator defines the schema of operators, such as convolution, relu, pooling, etc. al. Here are 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 only support caffe model. Tensorflow and MXNet support will be the next.

  • executor implements the code to run graph and operators. Current version only provides a highly optimized implementation for single A72.

This version can load and run caffe 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

Performance

The data is collected on single 1.8G A72 on chip RK3399, by repeating calling the forward interface to get the average time cost (ms) per run.

NN Caffe(Openblas) Tengine
squeezenet 147 91
mobilenet 306 122

For details to run benchmark, please visit benchmark page.

Build and Install

please refer to the install page

Develop New Operator

It is easy to add new operator to Tengine. Here is the guide on new operator

Support New Model Format

Tengine can be extended to support new serialization format, by building new serializer module.

How to build new serializer module

Release History

version 0.1.0 - 2018/01

Initial release of single A72 support

About

Tengine is a lite, high performance, modular inference engine for embedded device

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 83.2%
  • Assembly 11.7%
  • C 3.3%
  • Makefile 1.8%