Как добавить пользовательский оператор в Tensorflow Lite с атрибутами

Я добавил пользовательский оператор в 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.

2

Решение

Вы можете сделать это аналогично ядро mfcc. Атрибуты, которые вы определяете в своей пользовательской операции tenorflow, будут отображаться как flexbuffer и может быть легко опрошен.

0

Другие решения

Других решений пока нет …

По вопросам рекламы [email protected]