TensorFlow: Is there a way to measure FLOPS for a model?

Question

The closest example I can get is found in this issue: https://github.com/tensorflow/tensorflow/issues/899

With this minimum reproducible code:

import         


        

Answer 1

The above approaches no longer work for TF2.0 as the profiler methods have been deprecated and moved under compat.v1. Seems like this feature still needs to be implemented.

Below is an issue on Github: https://github.com/tensorflow/tensorflow/issues/32809