tensorflow lite添加自定义操作

时间:2018-11-05 07:30:19

标签: tensorflow compilation

我从源代码安装tensorflow。由于我需要在tensorflow lite模型中添加一些OP,因此我遵循了tensorflow官方指令的步骤。但是最后,我从libtensorflow-lite.a复制了我的操作名,但没有找到任何东西。在我的c ++演示中使用模型时,错误是“找不到版本1的名称'MatmulCalc'的自定义操作”,但是可以在编译后在libtensorflow_cc.so中使用该名称。并且模型传输过程可以。

所以我的问题是 如何在lite中添加新的OP,也许官方步骤尚未完成。https://www.tensorflow.org/guide/extend/op

1 个答案:

答案 0 :(得分:0)

您共享的链接用于在TensorFlow中创建自定义Op。如果您尝试在TensorFlow Lite中添加自定义Op,则可以在此处找到该指南:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/custom_operators.md