Skip to content

Enabling onnx session to run on GPU

Debottam Bakshi Gupta requested to merge dbakshig/athena:onnxUtilsForGPU into 22.0

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 ).

Merge request reports