Skip to content

Enabling onnx session to run on GPU in master

Debottam Bakshi Gupta requested to merge dbakshig/athena:onnxUtilsForGPU_master into master

In this MR I am creating an additional function CreateORTSessionGPU which will create an onnx session compatible to run on GPU. I am adding @keli ( also @akraszna ). (closing !58460 (closed) as it was targeting 22.0 branch)

Merge request reports