自定义操作被另一个操作取代

Custom op is replaced by another ops

我需要在语音识别模型中使用 Conv1D 层以在微控制器上 运行。由于 TFLM 不支持 Conv1D,我想使用 keras 层 class:之后我尝试定义和注册 TF Lite 支持的操作。

不过我的op似乎被另一个op取代了。怎么可能?

TFLite 通过将现有的 Conv2D 操作与 Reshape 操作包装在一起来支持 Conv1D。你试过转换吗?我认为 TensorFlow Lite 内置操作集已经支持您的案例。如果没有,请在 TensorFlow github.

提交功能请求