onnxruntime/orttraining/tools/amdgpu
Weixing Zhang 840212e115
Enable OneHot kernel for ROCm EP and add Dockerfile for ROCm 4.3.1 (#9656)
* enable OneHot for ROCm EP

* add dockerfile for ROCm 4.3.1

Co-authored-by: Weixing Zhang <wezhan@microsoft.com>
2021-12-07 12:47:00 -08:00
..
model
script
Dockerfile.rocm3.7
Dockerfile.rocm3.7.pytorch
Dockerfile.rocm3.8.pytorch
Dockerfile.rocm3.9
Dockerfile.rocm3.9.pytorch
Dockerfile.rocm3.10.pytorch
Dockerfile.rocm4.0.pytorch
Dockerfile.rocm4.1.pytorch [ROCm] dockerfile updates (#7955) 2021-06-10 23:50:19 -07:00
Dockerfile.rocm4.2.pytorch [ROCm] dockerfile updates (#7955) 2021-06-10 23:50:19 -07:00
Dockerfile.rocm4.3.1.pytorch Enable OneHot kernel for ROCm EP and add Dockerfile for ROCm 4.3.1 (#9656) 2021-12-07 12:47:00 -08:00