Check also:

  
      

      TensorFlow
  

  
    

Better performance with the tf.data API  |  TensorFlow Core 3



  

  
    
    
  

  


And If available on Metal

  
      

      TensorFlow
  

  
    

TensorFlow Profiler: Profile model performance  |  TensorBoard 2