tensorflow lite添加自定义操作

问题描述 投票:0回答:1

我从源代码安装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 compilation
1个回答
0
投票

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

© www.soinside.com 2019 - 2024. All rights reserved.