onnxruntime/include/onnxruntime/core
Weixing Zhang 74ee24cf7f
rename cuda_mem_limit and hip_mem_limit to gpu_mem_limit for both CUDA EP and ROCm EP (#7226)
With this change, differentiating CUDA EP and ROCm EP is not needed in training script when mem_limit option needs to be set.

Co-authored-by: Weixing Zhang <wezhan@microsoft.com>
2021-04-05 09:04:04 -07:00
..
common Cuda Profiler (#7110) 2021-03-29 12:04:36 -07:00
framework Check kernel def hashes (#7120) 2021-04-01 17:42:58 -07:00
graph Add support to save onnx graph with external initializers file. (#6911) 2021-03-12 09:15:25 +00:00
optimizer Enable NHWC transformer when generating ORT format model (#7126) 2021-03-29 18:39:48 +10:00
platform Exclude profiler from minimal build (#7115) 2021-03-25 09:06:14 -07:00
providers rename cuda_mem_limit and hip_mem_limit to gpu_mem_limit for both CUDA EP and ROCm EP (#7226) 2021-04-05 09:04:04 -07:00
session rename cuda_mem_limit and hip_mem_limit to gpu_mem_limit for both CUDA EP and ROCm EP (#7226) 2021-04-05 09:04:04 -07:00