For Directml the RFC is still in WIP see the last messages in

  

      github.com/tensorflow/community
  

  
    
  
    
  

  
    
      RFC: TensorFlow on DirectML 18
    

    
      tensorflow:master ← wchao1115:master
    

    
      
        opened 
        
          
        
        May 12, 2020
      
      

      
        
          
          wchao1115
        
      

      
        
          +105
          -0
        
      
    
  


  
    This RFC will be open for comment until Monday, May 25th, 2020.

| Status     …   | Proposed                                             |
:-------------- |:---------------------------------------------------- |
| **RFC #**     | [243](https://github.com/tensorflow/community/pull/243) |
| **Author(s)** | Chai Chaoweeraprasit (wchao@microsoft.com), Justin Stoecker (justoeck@microsoft.com), Adrian Tsai (adtsai@microsoft.com), Patrice Vignola (pavignol@microsoft.com) |
| **Sponsor**   | Penporn Koanantakool (penporn@google.com)  |
| **Updated**   | 2020-06-08                                           |

## Objective
Implement a new TensorFlow device type and a new set of kernels based on [DirectML](https://docs.microsoft.com/en-us/windows/win32/direct3d12/dml-intro), a hardware-accelerated machine learning library on the DirectX 12 Compute platform. This change broadens the reach of TensorFlow beyond its existing GPU footprint and enables high-performance training and inferencing on Windows devices with any DirectX12-capable GPU.
  

  

  
    
    
  

  


/Cc @penporn
Currently I still suggest to use the mentioned ROCM wheel on AMD GPUS