tensorflow lite添加自定义操作

Posted 2021-05-01

我从源代码安装tensorflow。因为我需要在tensorflow lite模型中添加一些OP，所以我按照tensorflow官方指令的步骤进行操作。但最后，我从libtensorflow-lite.a grep我的操作名称，没有找到任何内容。在我的c ++演示中使用该模型时，错误是“未找到名为'MatmulCalc'的版本1的自定义操作”，但是编译后可以在libtensorflow_cc.so中使用该名称。模型转移过程还可以。

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

答案

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