[FIXED] Measuring Flops for TFLite Model in Tensorflow 2.X

Issue

I am trying to measure FLOPS for a TFLite model in TF2. I know that Tensorflow 1.x had the tf.profiler, which was awesome for measuring flops. It doesn't seem to work well with tf.keras.

Could anybody please describe how to measure FLOPs for a TFLite model in TF2? I can't seem to find an answer online. Thank you all so much for your time.

Edit: The link commented below does not help with tflite.

Solution

I encountered the same problem and wrote a simple python package to roughly calculate FLOPS.

https://github.com/lisosia/tflite-flops

Only Conv and DepthwiseConv layers are considered, but it was sufficient for my use case.

Answered By - asao