Public
  1. Public

onnxruntime

Public

Commits

Sheil KumarGitHub
18133ddadb5
Sheil Kumar authored and GitHub committed fa16e2e0f39
Register CPU OptionalGetElement, OptionalHasElement on DirectML (#15926)

Register CPU OptionalGetElement, OptionalHasElement on DirectML

Graphs with OptionalGetElement and OptionalHasElement should work in a
DML graph without extra memcpy operation on and off the GPU.

CopyCpuTensor is swapped with DataTransferManager.CopyTensor() to make
the CPU operator usable by other providers.

---------

Co-authored-by: Dwayne Robinson <dwayner@microsoft.com>