在TFlite微环境中实现SELU激活

Question

我希望能够使用SELU激活在CNN模型上部署在单片机上使用TFlite微控制器。

SELU只是ELU的一个扩展版本，ELU是作为TFLite微内核之一实现的。

我实现了基于/lite/micro/kernels/elu.cc和/lite/micro/kernels/elu.h副本的SELU作为selu.cc和selu.h。对于int8操作，SeluEval()中的计算只是从一个查找表返回，该查找表是在SeluPrepare()期间计算和填充的。

从阅读自定义到现在我还不清楚，允许TFLiteMicro端口获取我的SELU激活，以及是否所有的 TFLite 框架，似乎坐在TFLite微的顶部，也需要能够捡起它来成功地转换和量化我的模型？

从我在TFLite微端口中可以看到的情况来看，所有内置的操作系统都在all_ops_resolver.cc中注册，而micro_mutable_op_resolver.h在这里添加了我的立方体op，这可能是一个继续前进的黑客，如果是nessecery的话。

有没有人成功地实现了一个自定义op，在TFLite微，并能够指出一个方向，这应该如何做？

或者，是否有关于该过程的其他信息可供选择？

Answer 1

您可以使用TFLu将自定义op添加到MicroMutableOpResolver运行时。测试代码包含了如何做到这一点的最小示例。如果使用自定义名称注册selu，则可以在运行时使用它。

如果空间允许，您可以使用‘'AllOpsResolver`’并添加自定义op。