How are TFLite models generated from the MLIR tfl dialect IR?

[Simon King]

Hi tflite team and friends,

I am working on a project of building TFLite models based on TFLite MLIR IR. According to one of your videos (https://youtu.be/R5LLIj8EMxw?t=1226 1), TFLite converter can translate tfl dialect MLIR IR to TFLite models. Therefore, I assume that the converter has integrated the tool that I need.

My questions are: (1) if I have the TFLite dialect IR, could I generate the tflite flatbuf file? (2) if the answer for (1) is yes, is there any resource that I can leverage? I know the mlir part of tflite is hosted in https://github.com/tensorflow/tensorflow/tree/5dcfc51118817f27fad5246812d83e5dccdc5f72/tensorflow/compiler/mlir/lite. But I don't know where to start.

Thanks!

[Jaesung Chung]

Hi Simon King,

If you have the TFLite dialect IR-based module op in the MLIR, you can convert it through the following method:

MlirToFlatBufferTranslateFunction

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/compiler/mlir/lite/flatbuffer_translate.h#L37

Also I would recommend looking at the flatbuffer_* files under the tensorflow/compiler/mlir/lite directory.

Best regards,

Jaesung

--
You received this message because you are subscribed to the Google Groups "TensorFlow Lite" group.
To unsubscribe from this group and stop receiving emails from it, send an email to tflite+un...@tensorflow.org.
To view this discussion on the web visit https://groups.google.com/a/tensorflow.org/d/msgid/tflite/c5e69eb5-314f-46b8-8bb6-5dac14538c4bn%40tensorflow.org.

[Simon King]

Thanks for your information!