For CPU and GPU in tf2xla we had something like this (but unmaintained)

  
      

      github.com
  

  
    

tensorflow/tensorflow 8

  master/tensorflow/compiler/tf2xla/g3doc

  An Open Source Machine Learning Framework for Everyone