onnxruntime/onnxruntime/contrib_ops
Dmitri Smirnov 58521fb822
Make training CUDA kernels to adhere established code structure patterns (#10735)
Current training optimizer kernels include CPU headers
  that affects changes that we can make in the CPU code with C++14 compiler and
  other refactoring efforts. Rearrange the kernel according to the established patterns
  and do not include headers that are not needed.
2022-03-09 09:06:45 -08:00
..
cpu update with onnx 1.11 release (#10441) 2022-03-07 21:10:55 -08:00
cuda Make training CUDA kernels to adhere established code structure patterns (#10735) 2022-03-09 09:06:45 -08:00
rocm [ROCm] Code sync from CUDA (#10631) 2022-02-22 22:40:06 -08:00