Activation count analysis for TF?

TensorFlow General Discussion
,
1

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

Thanks

3

Hi @mkal, You can use TensorFlow Profiler to understand the hardware resource consumption (time and memory) of the various TensorFlow operations (ops) in your model. For more details please refer to this document. Thank You.