我从源代码安装 tensorflow。由于我需要在 tensorflow lite 模型中添加一些 OP,我按照 tensorflow 官方说明的步骤进行操作。但最后,我从 libtensorflow-lite.a 中提取了我的操作名称,但什么也没找到。在我的 c++ 演示中使用模型时,错误是“没有找到名称为 'MatmulCalc' 版本 1 的自定义操作”,但是编译后可以在 libtensorflow_cc.so 中找到该名称。模型转移过程没问题。
所以我的问题是如何在lite中添加新的OP,可能官方步骤没有完成。https://www.tensorflow.org/guide/extend/op
您分享的链接用于在 TensorFlow 中创建自定义操作。如果您尝试在 TensorFlow Lite 中添加自定义操作,可在此处获得指南:https : //github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/custom_operators.md
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句