OAID / Tengine-Convert-Tools

Tengine Convert Tool supports converting multi framworks' models into tmfile that suitable for Tengine-Lite AI framework.
Apache License 2.0
94 stars 34 forks source link

oneflow->tengine init #28

Closed daquexian closed 3 years ago

daquexian commented 3 years ago

Note: The oneflow model file should be generated in oneflow's dev_serving branch

Tested on alexnet and mobilenetv2

daquexian commented 3 years ago

The code is mainly in tools/oneflow/oneflow_serializer.cpp and tools/include/oneflow_serializer.hpp (~800 lines)