There Is a quite long thread for this in TF 2.x:

  

      github.com/tensorflow/tensorflow
  

  
    
  
	  
  

  
    
      TF 2.0 Feature: Flops calculation 29
    

    
      
        opened 
        
          
        
        Sep 25, 2019
      
      


      
        
          
          pzobel
        
      
    

    
        stat:awaiting tensorflower
        type:feature
        comp:tfdbg
        TF 2.0
    
  


  
    <em>Please make sure that this is a feature request. As per our [GitHub Policy](…https://github.com/tensorflow/tensorflow/blob/master/ISSUES.md), we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:feature_template</em>


**System information**
- TensorFlow version (you are using): TF 2.0 RC2
- Are you willing to contribute it (Yes/No):



**Describe the feature and the current behavior/state.**

I am missing the opportunity to compute the number of floating point operations of a tf.keras Model in TF 2.0. 
In TF 1.x tf.profiler was available [see here](https://stackoverflow.com/questions/45085938) but I can find anything equivalent for TF 2.0 yet.

**Will this change the current api? How?**

**Who will benefit with this feature?**

Everbody interested in the computational complexity of a TensorFlow model.

**Any Other info.**