onnxruntime/orttraining/tools
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
..
amdgpu Enable OneHot kernel for ROCm EP and add Dockerfile for ROCm 4.3.1 (#9656) 2021-12-07 12:47:00 -08:00
ci_test Switch AMD CI pipeline to use environment image from onnxruntimecibuildenvironment (#9206) 2021-09-28 13:06:16 -07:00
mnist_model_builder
scripts Fix Training Packaging pipeline (#9885) 2021-11-30 15:26:10 -08:00