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