Have you tried with:

  
      

      Apple Developer
  

  
    

Tensorflow Plugin - Metal - Apple Developer 5

  Find presentations, documentation, sample code, and resources for building macOS, iOS, and tvOS apps with the Metal framework.