search

Tencent / TNN

TNN: developed by Tencent Youtu Lab and Guangying Lab, a uniform deep learning inference framework for mobile、desktop and server. TNN is distinguished by several outstanding features, including its cross-platform capability, high performance, model compression and code pruning. Based on ncnn and Rapidnet, TNN further strengthens the support and performance optimization for mobile devices, and also draws on the advantages of good extensibility and high performance from existed open source efforts. TNN has been deployed in multiple Apps from Tencent, such as Mobile QQ, Weishi, Pitu, etc. Contributions are welcome to work in collaborative with us and make TNN a better framework.
Other
4.38k stars 767 forks source link

LSTM、GRU算子支持 #874

Open huachao1001 opened 3 years ago

huachao1001 commented 3 years ago

TNN目前还不支持LSTM、GRU，预计什么时候能支持呢？

darrenyao87 commented 3 years ago

LSTM内部已经支持，GRU能否提供onnx对应的onnx模型链接？ @huachao1001

liziru commented 3 years ago

LSTM内部已经支持，GRU能否提供onnx对应的onnx模型链接？ @huachao1001

有更新到github里吗 我成功的将lstm model 转为tnn model， 但是推理时候报错。 image

https://github.com/Tencent/TNN/issues/1227