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

Question

我按照以下说明在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。

Answer 1

您可以执行与mfcc kernel类似的操作。您在自定义tensorflow操作中定义的属性将显示为flexbuffer，并且可以轻松查询。