onnxruntime
ffb94fd1 - Fix bug with delayed allocation of If and Scan outputs. (#2024)

Commit
6 years ago
Fix bug with delayed allocation of If and Scan outputs. (#2024) * Fix bug with delayed allocation of If and Scan outputs. If the subgraph is producing output on a non-CPU device the delayed allocation was incorrectly providing a CPU allocated tensor. Check for the required location, and update 'fetches' instead if a device copy is needed. The utils::ExecuteGraph logic will handle the device copy in this case.
Author
Parents
Loading