Activation count analysis for TF?

Hi,
In pytorch there is a library for doing operation counting and more importantly for me activation count analysis tools in fvcore library: fvcore/activation_count.py at main · facebookresearch/fvcore · GitHub
Is there a similar utility/library for TF? Especially for TFLITE models.

Thanks