我按照以下说明在 Tensorflow 中添加了自定义运算符: https://www.tensorflow.org/extend/adding_an_op
现在我想将相同的运算符添加到 Tensorflow Lite。 我关注了this instructions to add custom operator in TF Lite,但它没有说明如何添加带有属性的自定义运算符。 像这样:
REGISTER_OP("CustomOpName")
.Attr("T: {float}")
.Attr("scale_factor: int")
.Input("x: T")
.Output("out: T")
我尝试像内置运算符一样将 TfLiteNode* 节点转换为自定义参数结构,如 TfLiteLocalResponseNormParams,但问题是 TfLiteNode* 节点等于 NULL。
最佳答案
您可以像 mfcc kernel 那样做.您在自定义 tensorflow op 中定义的属性将显示为 flexbuffer方便查询。
关于c++ - 如何使用属性在 Tensorflow Lite 中添加自定义运算符,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47818079/