问如何使用属性在Tensorflow Lite中添加自定义运算符
EN

Stack Overflow用户

提问于 2017-12-15 00:30:24

回答 1查看 1.6K关注 0票数 4

我按照下面的说明在Tensorflow中添加了自定义操作符：https://www.tensorflow.org/extend/adding_an_op

现在我想向Tensorflow Lite添加相同的运算符。我按照this的说明在TF Lite中添加了自定义运算符，但它没有说明如何添加带有属性的自定义运算符。如下所示：

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

我试图像内置操作符那样将TfLiteNode*节点转换为自定义参数结构，如TfLiteLocalResponseNormParams，但问题是TfLiteNode*节点等于NULL。

发布于 2018-07-28 04:34:21

您可以使用类似于mfcc kernel的方法来完成此操作。您在自定义tensorflow op中定义的属性将显示为flexbuffer，并且可以轻松地进行查询。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/47818079

复制

相似问题

问如何使用属性在Tensorflow Lite中添加自定义运算符EN