Tensorflow 中的“Flex Op”是什么意思？

Question

当我研究 Tensorflow Lite 时，我发现自定义操作将作为“flex op”导出，而不是作为本机导出。我不明白什么是“flex op”，什么是“native”。谢谢你！

与我的问题相关的代码位于https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/toco/tflite/export.cc#L368

Answer 1

请参阅此处的文档：将 TensorFlow Lite 与选定的 TensorFlow 操作结合使用

这是一项实验性功能，通过链接和运行 TensorFlow 内核来支持 TensorFlow Lite 不支持的 TensorFlow 操作。

“Flex”是“使用 TensorFlow Lite 和选择 TensorFlow 操作”功能的内部代码名称。