如何在Tensorflow Lite中添加自定义运算符

How to add custom operator in Tensorflow Lite with attributes

本文关键字：添加 自定义 运算符 Lite Tensorflow      更新时间：2023-10-16

我在tensorflow中添加了自定义操作员，请按照以下说明：https://www.tensorflow.org/extend/adding_an_op

现在，我想在Tensorflow Lite中添加相同的操作员。我遵循此说明将自定义操作员添加到TF Lite中，但它没有说明如何添加具有属性的自定义操作员。这样：

REGISTER_OP("CustomOpName")
.Attr("T: {float}")
.Attr("scale_factor: int")
.Input("x: T")
.Output("out: T")

我试图这样做，就像内置运算符一样，将tfliteNode*降低到自定义参数结构，例如tflitelocalresponsenormparams，但问题是tflitenode*节点等于null。

您可以类似于MFCC内核。您在自定义TensorFlow OP中定义的属性将显示为Flexbuffer，并且可以轻松查询。

相关文章：