tensorflow lite 添加自定义操作

大卫.ju321

我从源代码安装 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] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

在Tensorflow Lite C API中注册自定义运算符

带有自定义模型的 Tensorflow lite 示例 - “input_product_scale < output_scale was not true”

添加自定义TensorFlow OP

如何在tensorflow的自定义操作/图层后添加slim.batch_norm?

Tensorflow js VS Tensorflow Lite

Tensorflow:自定义操作需要定义哪些梯度?

在 tensorflow 中添加自定义操作时,“tensorflow/core/framework/common_shape_fns.h: No such file or directory”

关于 Tensorflow 和 PyTorch 中的自定义操作

自定义 Tensorflow 操作采用可变输入张量列表

tensorflow lite转换失败。出现“未定义的符号:_ZTIN10tensorflow6DeviceE”

是否可以在 C++ 中为 TensorFlow Lite 模型实现自定义运算符,即使它已经有一个 bultin op 可用?

带有 NativeScript 的 Tensorflow lite

将Tensorflow转换为Tensorflow-lite

将 BATCH_TO_SPACE_ND 和 SPACE_TO_BATCH_ND 添加到 tensorflow lite

Vega-lite:根据选择自定义轴样式

Vega-lite:如何自定义图例标签?

自定义按钮以显示侧边栏Material Design Lite

自定义vega-lite线图轴标签

ServiceStack ORM Lite自定义SQL LIKE语句通配符

建立自定义的Material Design Lite Palette

HERE-lite API - 自定义线条样式

将ONNX模型转换为TensorFlow Lite

TensorFlow Lite C ++ API示例进行推理

为Edge TPU构建Tensorflow Lite失败

如何使用TensorFlow Lite进行批处理?

如何设置Tensorflow Lite C ++的输入

Tensorflow Lite模型输出错误

Tensorflow Lite现在支持设备培训吗

TensorFlow Lite是否支持keras函数Flatten()?